docs

#!/usr/bin/env node
var argv = require("optimist").argv,
    path = require("path"),
    docs = require("../lib/docs");
require("colors");

var task = argv._[0],
    dir = (argv.length - 1) ? (argv._[argv._.length - 1]) : ".";

switch (task) {
  case "generate":
    console.log("GENERATING YOUR FILES".yellow);
    docs.generate(dir+"/pages", dir+"/public");
    break;

  case "serve":

    console.log("SERVING YOUR FILES".yellow);

    var HTTPServer = require('http-server').HTTPServer;
    var httpServer = new HTTPServer({root: dir+'/public/'});

    httpServer.start();

    process.on('SIGINT', function() {
      httpServer.log('http-server stopped.'.red);
      return process.exit();
    });

    break;
  default:
    console.log("Invalid instructions!".red);
    console.log();
    console.log("You may either:");
    console.log();
    console.log("* blacksmith generate <folder>");
    console.log("* blacksmith serve <folder>");
    console.log();
}