index.js

#!/usr/bin/env node
var path = require('path');
var fs = require('fs');

var command = process.argv[2];
var param = process.argv[3];

if (command == 'generate') {
    if (!param) {
        console.error('Filename is required when calling generate. Example: "usite generate blog"');
        return;
    }

    var generator = require(path.resolve(process.cwd(), param));
    if (!generator.generate) {
        console.error("Provided generator is missing generate function");
        return;
    }

    var uSite = require("./dist/usite");
    generator.generate(uSite.default);
}

if (command == 'init') {
    var fse = require('fs-extra');
    var dirPath = __dirname;

    var demoPath = path.resolve(dirPath, './node_modules/usiteDemo/');
    if (!fse.existsSync(demoPath)) {
        demoPath = path.resolve(dirPath, '../usiteDemo/')
        if (!fse.existsSync(demoPath)) {
            console.error("usiteDemo dependency not found");
            return;
        }
    }

    var workingDirectory = process.cwd();

    fse.copySync(path.resolve(demoPath, "./content"), path.resolve(workingDirectory, "./content"));
    fse.copySync(path.resolve(demoPath, "./template"), path.resolve(workingDirectory, "./template"));
    fse.copySync(path.resolve(demoPath, "./blog.js"), path.resolve(workingDirectory, "./blog.js"));
    fse.copySync(path.resolve(demoPath, "./website.json"), path.resolve(workingDirectory, "./website.json"));
}