package burp;

import burp.ui.ConfigPanel;
import burp.ui.ExtensionTab;

import java.io.PrintWriter;

public class BurpExtender implements IBurpExtender {
    public static String VERSION = "1.0";
    public static String NAME = "APIKit";
    public static String FULLNAME = NAME + " v" + VERSION;

    private static PrintWriter stdout;
    private static PrintWriter stderr;
    private static IBurpExtenderCallbacks callbacks;
    private static IExtensionHelpers helpers;
    private static ConfigPanel configPanel;
    private static CookieManager cookieManager;
    private static ExtensionTab extensionTab;
    private static PassiveScanner passiveScanner;

    public static PrintWriter getStdout() {
        return stdout;
    }

    public static PrintWriter getStderr() {
        return stderr;
    }

    public static IBurpExtenderCallbacks getCallbacks() {
        return callbacks;
    }

    public static IExtensionHelpers getHelpers() {
        return helpers;
    }

    public static ConfigPanel getConfigPanel() {
        return configPanel;
    }

    public static void setConfigPanel(ConfigPanel configPanel) {
        BurpExtender.configPanel = configPanel;
    }

    public static CookieManager getCookieManager() {
        return cookieManager;
    }

    public static ExtensionTab getExtensionTab() {
        return extensionTab;
    }

    public static PassiveScanner getPassiveScanner() {
        return passiveScanner;
    }

    @Override
    public void registerExtenderCallbacks(IBurpExtenderCallbacks callbacks) {
        BurpExtender.callbacks = callbacks;
        BurpExtender.helpers = callbacks.getHelpers();
        BurpExtender.stdout = new PrintWriter(callbacks.getStdout(), true);
        BurpExtender.stderr = new PrintWriter(callbacks.getStderr(), true);

        // , ExtensionTab  BurpExtender.callbacks, 
        BurpExtender.extensionTab = new ExtensionTab(NAME);
        BurpExtender.cookieManager = new CookieManager();
        BurpExtender.passiveScanner = new PassiveScanner();

        callbacks.registerScannerCheck(passiveScanner);
        callbacks.registerHttpListener(BurpExtender.cookieManager);
        callbacks.registerContextMenuFactory(new ContextMenu());

        callbacks.setExtensionName(FULLNAME);
        BurpExtender.stdout.println("===================================");
        BurpExtender.stdout.println(String.format("%s load success!", FULLNAME));
        BurpExtender.stdout.println("Author: yulige,rmb122");
        BurpExtender.stdout.println("===================================");
    }
}