extension.js

const St = imports.gi.St;
const Main = imports.ui.main;
const PanelMenu = imports.ui.panelMenu;
const PopupMenu = imports.ui.popupMenu;
const Panel = imports.ui.panel;
const Lang = imports.lang;
const GLib = imports.gi.GLib;

const ExtensionUtils = imports.misc.extensionUtils;
const Me = ExtensionUtils.getCurrentExtension();
const AdbHelper = Me.imports.adbhelper;

let screenRecord = {recording: false, deviceId: 0};
let tcpConnection ={connected: false};

const AndroidMenuItem = new Lang.Class({
    Name: 'AndroidMenuItem',
    Extends: PopupMenu.PopupBaseMenuItem,

    _init: function(item) {

        this.parent();

        if(item.icon != null) {
            this._icon = new St.Icon({ icon_name: item.icon,
             icon_size: 16 });

            this.actor.add_child(this._icon);
        }

        this._label = new St.Label({ text: item.label });
        this.actor.add_child(this._label);

    },

    destroy: function() {
        this.parent();
    },

    activate: function(event) {
        this.parent(event);
    }

});

const AndroidMenu = new Lang.Class({
    Name: 'AndroidMenu.AndroidMenu',
    Extends: PanelMenu.Button,

    _init: function() {
        this.parent(0.0, "Android Menu", false);

        let hbox = new St.BoxLayout({style_class: 'panel-status-menu-box'});

        this._icon = new St.Icon({ icon_name: 'android_icon',
            style_class: 'system-status-icon'
        });

        //add panel menu button
        hbox.add_child(this._icon);
        hbox.add_child(PopupMenu.arrowIcon(St.Side.BOTTOM));
        this.actor.add_actor(hbox);
        this.actor.add_style_class_name('panel-status-button');


        this.actor.connect('button-press-event', Lang.bind(this,this._findDevices));

        this._addErrorItem("No devices found");

        //start adb daemon if not running
        AdbHelper.startDaemon();
        
    },

    _findDevices: function() {

        let result = AdbHelper.findDevices();

        if(result.error != null) {
            global.log(result.error);
            this._addErrorItem(result.error);
        } else {

            if(result.devices != null && result.devices.length!=0) {
                this._addMenuItems(result.devices);
            } else {
                this._addErrorItem("No devices found");
            }

        }

    },

    _screenshotClicked: function(device) {

        AdbHelper.takeScreenshot(device.deviceId);
        Main.notify("Screenshot saved in Desktop")
        
    },

    _recordScreen: function(device) {
        if(screenRecord.recording) {
            AdbHelper.stopScreenRecording(device.deviceId);
            screenRecord.recording = false;
            screenRecord.deviceId = 0;
            this._icon.icon_name = 'android_icon'
            Main.notify("Screen recording saved in Desktop")

        } else {
            AdbHelper.recordScreen(device.deviceId);
            screenRecord.recording = true;
            screenRecord.deviceId = device.deviceId;
            this._icon.icon_name = 'record_icon'
        }
    },

    _connectTCP: function(device) {

        if(tcpConnection.connected) {
            AdbHelper.useUsb();
            tcpConnection.connected = false;
        } else {
            let status = AdbHelper.establishTCPConnection(device.deviceId);

            if(status.indexOf("unable to connect") !== -1) {
                tcpConnection.connected = false;
            } else {
                tcpConnection.connected = true;
            }

            Main.notify(status);
        }
    },

    _captureBugReport: function(device) {
        AdbHelper.captureBugReport(device.deviceId);
        Main.notify("Capturing bug report, will be saved in Desktop");
    },

    _addMenuItems: function(devices) {

        this.menu.removeAll();

        for(var i=0; i < devices.length; i++) {

            let device = devices[i];

            if(device != null) {

                let section = new PopupMenu.PopupMenuSection();

                    //device name and id
                    let deviceItem = new AndroidMenuItem({label: device.name.trim() + " - " + device.deviceId.trim()});
                    section.addMenuItem(deviceItem);

                    //add screenshot item
                    let screenshotItem = new AndroidMenuItem({label: "Take screenshot", icon: 'screenshot_icon'});
                    screenshotItem.connect('activate', Lang.bind(this, function() {
                        this._screenshotClicked(device)
                    }));
                    section.addMenuItem(screenshotItem);

                    //add screen record item
                    let recordLabel = "Record screen"
                    let recordIcon = "record_icon"

                    if(screenRecord.recording && screenRecord.deviceId == device.deviceId) {
                        recordLabel = "Recording in progress. Click to stop recording";
                        recordIcon = 'record_off_icon'
                    }
                    let recordScreenItem = new AndroidMenuItem({label: recordLabel, icon: recordIcon});

                    recordScreenItem.connect('activate', Lang.bind(this, function() {
                        this._recordScreen(device)
                    }));
                    section.addMenuItem(recordScreenItem);

                    //add remote connection item
                    let remoteLabel = "Establish remote connection"
                    let remoteIcon = "remote_icon"

                    if(tcpConnection.connected) {
                        remoteLabel = "Connected over TCP. Click to use USB instead";
                        remoteIcon = 'remote_off_icon'
                    }

                    let remoteItem = new AndroidMenuItem({label: remoteLabel, icon: remoteIcon});

                    remoteItem.connect('activate', Lang.bind(this, function() {
                        this._connectTCP(device)
                    }));
                    section.addMenuItem(remoteItem);

                    //add bug report item
                    let bugReportItem = new AndroidMenuItem({label: "Capture bug report", icon: 'bug_report_icon'});
                    bugReportItem.connect('activate', Lang.bind(this, function() {
                        this._captureBugReport(device)
                    }));
                    section.addMenuItem(bugReportItem);

                    this.menu.addMenuItem(section)
                    this.menu.addMenuItem(new PopupMenu.PopupSeparatorMenuItem());

                }

                
            }

            
        },

        _addErrorItem: function(error) {

            this.menu.removeAll();
            let errorItem = new AndroidMenuItem({label: error});
            this.menu.addMenuItem(errorItem);

        },


        _toArray: function(str) {
            let arr = str.split(" ");
            return arr;
        }


    });


function init(extensionMeta) {
    //add icons to search path
    let theme = imports.gi.Gtk.IconTheme.get_default();
    theme.append_search_path(extensionMeta.path + "/icons");
}


let _indicator;


function enable() {
    _indicator = new AndroidMenu;
    
    Main.panel.addToStatusArea('android-menu', _indicator);

}

function disable() {
    _indicator.destroy();
}