source: Daodan/src/Patches/Input.c@ 1089

#include <stdlib.h>
#include <stdarg.h>

#include "../Daodan.h"
#include "Input.h"
#include "../Oni/Oni.h"
#include "../Daodan_Patch.h"
#include "Utility.h"

#define EVENT_KEYPRESS_SECURETIME 3

typedef struct {
        uint32_t key;
        const char* actionname;
        ActionEventType_t eventType;
        uint32_t keydownTimeoutTicks;
        int64_t lastevent;
        
        CustomActionCallback_t callback;        
        CustomActionCallbackArgument callbackArgument;
} DD_CustomAction_t;

static DD_CustomAction_t customActions[100];


void Input_RegisterCustomAction (const char* actionname, ActionEventType_t eventType, uint32_t keydownTimeoutTicks, CustomActionCallback_t callback, CustomActionCallbackArgument callbackArgument) {
        uint16_t i = 0;
        DD_CustomAction_t* cur = customActions;
        
        while ( (i < ARRAY_SIZE(customActions)) && (cur->callback != 0)) {
                cur++;
                i++;
        }
        
        if (i < ARRAY_SIZE(customActions)) {
                cur->actionname = actionname;
                cur->eventType = eventType;
                cur->keydownTimeoutTicks = keydownTimeoutTicks;
                cur->callback = callback;
                cur->callbackArgument = callbackArgument;
        } else {
                STARTUPMESSAGE("Registering action %s failed, maximum actions reached", actionname);
        }
}


_LIrBinding_Add Oni_LIrBinding_Add = (_LIrBinding_Add)0;
uint16_t ONICALL DD_LIrBinding_Add(uint32_t key, const char* name) {
        DD_CustomAction_t* cur;
        for (cur = customActions; cur->callback != 0; cur++) {
                if (!strcmp(name, cur->actionname)) {
                        cur->key = key;
                        return 0;
                }
        }

        return Oni_LIrBinding_Add(key, name);
}

_LIrActionBuffer_Add Oni_LIrActionBuffer_Add = (_LIrActionBuffer_Add)0;
void ONICALL DD_LIrActionBuffer_Add(void* unknown, LItDeviceInput* input) {
        DD_CustomAction_t* cur;
        for (cur = customActions; cur->callback != 0; cur++) {
                if (cur->key == input->input) {
                        int64_t curTime = UUrMachineTime_Sixtieths();
                        if (cur->eventType == EVENT_KEYPRESS) {
                                if (cur->lastevent + EVENT_KEYPRESS_SECURETIME < curTime) {
                                        cur->callback(cur->callbackArgument);
                                }
                                cur->lastevent = curTime;
                        } else if (cur->eventType == EVENT_KEYDOWN) {
                                if (cur->lastevent + cur->keydownTimeoutTicks < curTime) {
                                        cur->callback(cur->callbackArgument);
                                        cur->lastevent = curTime;
                                }
                        }
                        return;
                }
        }
        Oni_LIrActionBuffer_Add(unknown, input);
}

void Input_PatchCode () {
        Oni_LIrBinding_Add = DDrPatch_MakeDetour((void*)LIrBinding_Add, (void*)DD_LIrBinding_Add);
        Oni_LIrActionBuffer_Add = DDrPatch_MakeDetour((void*)LIrActionBuffer_Add, (void*)DD_LIrActionBuffer_Add);
}