cli-droidscript/initialize-cmd.js

Command Line Interface for DroidScript.

const path = require("path");
const { prompt } = require('enquirer');
const ora = require('ora');
const ext = require("./extension");
const fs = require("fs");
const chalk = require("chalk");

let CONFIG = {};
let DSCONFIG = {};
const FOLDERS = "APKs,AABs,SPKs,PPKs,.edit,.node,Plugins,Extensions";

async function initialize() {

    // reload configurations
    CONFIG = ext.getLocalData();
    const dir = path.resolve( process.cwd() );
    let ndx = null;
    if(CONFIG.localProjects && CONFIG.localProjects.length) {
        ndx = CONFIG.localProjects.findIndex(m => m.path == dir);
        if(ndx >= 0) {
            return console.log( chalk.hex("#ff9800").bold(`Folder is already initialize as ${CONFIG.localProjects[ndx].PROJECT} app!`) );
        }
    }

    DSCONFIG = getLocalData();

    if(DSCONFIG && DSCONFIG.PROJECT) {
        return console.log( chalk.cyan(`Folder is already initialize as ${DSCONFIG.PROJECT} app!`) );
    }

    CONFIG.PROJECT = "";

    try {
        const serverInfo = await ext.getServerInfo();
        if(serverInfo.status != "ok") {
            return console.log(serverInfo);
        }
        if(serverInfo.usepass == true) {
            var res = await ext.login( CONFIG.password );
            if(res.data.status != "ok") {
                return console.log(res.data);
            }
        }
        return await loadProjects();
    }
    catch( error ) {
        console.log( error );
    }
}

async function loadProjects() {
    try {
        var res = await ext.listFolder("");
        if(res.status == "ok") {
            var response = await prompt({
                type: "autocomplete",
                name: 'project',
                message: 'Select a project',
                limit: 10,
                initial: 0,
                choices: res.list.filter(m => (!m.includes(".") && !m.includes("~") && !FOLDERS.includes(m)))
            });

            DSCONFIG.PROJECT = response.project;
            CONFIG.PROJECT = DSCONFIG.PROJECT;
            ext.setCONFIG(CONFIG);

            ext.saveLocaData();

            progress = ora("Downloading files...").start();

            await getAllFiles();

            // await saveLocaData();

            progress.succeed("Initialization complete!");
        }
    } catch( error ) {
        console.log( error );
    }
}

async function getAllFiles( folder ) {
    folder = folder || DSCONFIG.PROJECT;
    try {
        let data = await ext.listFolder( folder );
        if( data.status == "ok" && data.list.length ) {
            let fileName = "", response = {}, path = "", filePath = "";
            for(var i=0; i<data.list.length; i++) {
                fileName = data.list[i], path = folder+"/"+fileName;
                filePath = path.replace(CONFIG.PROJECT+"/", "");
                if(!fileName.startsWith("~") && fileName.includes(".")) {
                    try {

                        progress.color = getColor();
                        progress.text = `Downloading ${path}`;

                        response = await ext.loadFile(path);
                        if(response.status == "ok") {
                            writeFile(filePath, response.data);
                        }
                    } catch(err) {
                        console.log( err );
                    }
                }
                else if( !fileName.includes(".") ) {
                    createFolder( filePath );
                    await getAllFiles( path );
                }
            }
        }
    } catch( err ) {
        console.log( err );
    }
}

function getLocalData() {
    const dir = path.resolve( process.cwd() );
    var filePath = path.join(dir, ".droidscript", "dsconfig.json");
    if( fs.existsSync(filePath) ) {
        const fileData = fs.readFileSync(filePath, 'utf8');
        DSCONFIG = JSON.parse( fileData );
        return DSCONFIG;
    }
    return {};
}

function createFolder( folder ) {
    if( !fs.existsSync(folder) ) {
        fs.mkdirSync( folder );
    }
}

// Write the file to the workspace
function writeFile(fileName, content) {
    if(fileName.includes(".json") && typeof content == "object") {
        content = JSON.stringify(content, null, 4);
    }
    const dir = path.resolve( process.cwd() );
    const filePath = path.join(dir, fileName);
    fs.writeFileSync(filePath, content, { flag: 'w' });
}

// utilities
var __c = -1;
function getColor() {
    __c ++;
    const colors = ["red", "green", "yellow", "blue", "magenta", "cyan", "white"];
    if(__c == colors.length) __c = 0;
    return colors[__c];
}


module.exports = initialize;