FrenchYeti/dexcalibur

const IFC = require("../../../src/InspectorFrontController.js");
var CONST = require("../../../src/CoreConst.js");
var AH = require("../../../src/AnalysisHelper.js");
const Disassembler = require("../../../src/Disassembler.js");
const Fs = require("fs");

var Controller =  new IFC.FrontController();

var DEBUG = false;


/*

*/
function getInvokedMethod(context){
    let meth = context.find.method("has."+AH.TAG.Invoked.Dynamically);
    return meth.toJsonObject();
}

function getExternalDex(context){
    let files = context.getInspector("DynamicLoader").getDB().getIndex("dex");

    return files.toJsonObject();
}

function getElementsDiscovered(context){
    let cls = context.find.class("tags:^"+AH.TAG.Discover.Dynamically+"$");
    return cls.toJsonObject();
}

function cleanupSavedDex(context){
    let files = context.getInspector("DynamicLoader").getDB().getIndex("dex");

    files.map(function(k,v){
        try{
            Fs.unlinkSync(v.getPath());
        }catch(err){}
    });  

    return true;
}

/**
 * Delegate front controller
 */
Controller.registerHandler(IFC.HANDLER.GET, function(ctx,req,res){

    var action = req.query.action;
    var act ={
        status: 404,
        data: { error: "Action not found. "}
    };

    switch(action){
        case 'refresh_reflect':
            act.status = 200;
            act.data.error = null;
            act.data.data = getInvokedMethod(ctx);
            break;
        case 'refresh_dex':
            act.status = 200;
            act.data.error = null;
            act.data.data = getExternalDex(ctx);
            break;
        case 'refresh_discover':
            act.status = 200;
            act.data.error = null;
            act.data.data = getElementsDiscovered(ctx);
            break;
        case 'cleanup':
            act.status = 200;
            act.data.error = null;
            act.data.data = cleanupSavedDex(ctx);
            break;
    }

    res.status(act.status).send(act.data);
});

/*
Controller.registerHandler(IFC.HANDLER.POST, function(ctx,req,res){
    console.log("POST", req.query);
    res.send({ msg:"ok" });
});*/


module.exports = Controller;