#!/usr/bin/env node

var swig = require('../index')
var optimist = require('optimist')
var fs = require('fs')
var path = require('path')
var filters = require('../lib/filters')
var utils = require('../lib/utils')
var uglify = require('uglify-js')

var command,
  wrapstart = 'var tpl = ',
  argv = optimist
    .usage('\n Usage:\n' +
      '    $0 compile [files] [options]\n' +
      '    $0 run [files] [options]\n' +
      '    $0 render [files] [options]\n'
      )
    .describe({
      v: 'Show the Swig version number.',
      o: 'Output location.',
      h: 'Show this help screen.',
      j: 'Variable context as a JSON file.',
      c: 'Variable context as a CommonJS-style file. Used only if option `j` is not provided.',
      m: 'Minify compiled functions with uglify-js',
      'filters': 'Custom filters as a CommonJS-style file',
      'tags': 'Custom tags as a CommonJS-style file',
      'options': 'Customize Swig\'s Options from a CommonJS-style file',
      'wrap-start': 'Template wrapper beginning for "compile".',
      'wrap-end': 'Template wrapper end for "compile".',
      'method-name': 'Method name to set template to and run from.'
    })
    .alias('v', 'version')
    .alias('o', 'output')
    .default('o', 'stdout')
    .alias('h', 'help')
    .alias('j', 'json')
    .alias('c', 'context')
    .alias('m', 'minify')
    .default('wrap-start', wrapstart)
    .default('wrap-end', ';')
    .default('method-name', 'tpl')
    .check(function (argv) {
      if (argv.v) {
        return
      }

      if (!argv._.length) {
        throw new Error('')
      }

      command = argv._.shift()
      if (command !== 'compile' && command !== 'render' && command !== 'run') {
        throw new Error('Unrecognized command "' + command + '". Use -h for help.')
      }

      if (argv['method-name'] !== 'tpl' && argv['wrap-start'] !== wrapstart) {
        throw new Error('Cannot use arguments "--method-name" and "--wrap-start" together.')
      }

      if (argv['method-name'] !== 'tpl') {
        argv['wrap-start'] = 'var ' + argv['method-name'] + ' = '
      }
    })
    .argv,
  ctx = {},
  out = function (file, str) {
    console.log(str)
  },
  efn = function () { },
  anonymous,
  files,
  fn

// What version?
if (argv.v) {
  console.log(require('../package').version)
  process.exit(0)
}

// Pull in any context data provided
if (argv.j) {
  ctx = JSON.parse(fs.readFileSync(argv.j, 'utf8'))
} else if (argv.c) {
  ctx = require(argv.c)
}

if (argv.o !== 'stdout') {
  argv.o += '/'
  argv.o = path.normalize(argv.o)

  try {
    fs.mkdirSync(argv.o)
  } catch (e) {
    if (e.code !== 'EEXIST') {
      throw e
    }
  }

  out = function (file, str) {
    file = path.basename(file)
    fs.writeFileSync(argv.o + file, str, { flags: 'w' })
    console.log('Wrote', argv.o + file)
  }
}

// Set any custom filters
if (argv.filters) {
  utils.each(require(path.resolve(argv.filters)), function (filter, name) {
    swig.setFilter(name, filter)
  })
}

// Set any custom tags
if (argv.tags) {
  utils.each(require(path.resolve(argv.tags)), function (tag, name) {
    swig.setTag(name, tag.parse, tag.compile, tag.ends, tag.block)
  })
}

// Specify swig default options
if (argv.options) {
  swig.setDefaults(require(argv.options))
}

switch (command) {
  case 'compile':
    fn = function (file, str) {
      var r = swig.precompile(str, { filename: file, locals: ctx }).tpl.toString().replace('anonymous', '')

      r = argv['wrap-start'] + r + argv['wrap-end']

      if (argv.m) {
        r = uglify.minify(r, { fromString: true }).code
      }

      out(file, r)
    }
    break

  case 'run':
    fn = function (file, str) {
      (function () {
        eval(str)
        var __tpl = eval(argv['method-name'])
        out(file, __tpl(swig, ctx, filters, utils, efn))
      }())
    }
    break

  case 'render':
    fn = function (file, str) {
      out(file, swig.render(str, { filename: file, locals: ctx }))
    }
    break
}

argv._.forEach(function (file) {
  var str = fs.readFileSync(file, 'utf8')
  fn(file, str)
})