#!/usr/bin/env node
'use strict'
const os = require('os')
const fs = require('fs')
const path = require('path')

try {
  require(path.join(os.homedir(), '.fxrc')) // Should be required before config.js usage.
} catch (err) {
  if (err.code !== 'MODULE_NOT_FOUND') {
    throw err
  }
}

const print = require('./print')
const reduce = require('./reduce')

const usage = `
  Usage
    $ fx [code ...]

  Examples
    $ echo '{"key": "value"}' | fx 'x => x.key'
    value

    $ echo '{"key": "value"}' | fx .key
    value    

    $ echo '[1,2,3]' | fx 'this.map(x => x * 2)'
    [2, 4, 6]

    $ echo '{"items": ["one", "two"]}' | fx 'this.items' 'this[1]'
    two

    $ echo '{"count": 0}' | fx '{...this, count: 1}'
    {"count": 1}

    $ echo '{"foo": 1, "bar": 2}' | fx ?
    ["foo", "bar"]

`

const {stdin, stdout, stderr} = process
const args = process.argv.slice(2)
const skip = Symbol('skip')
global.select = select


void function main() {
  stdin.setEncoding('utf8')

  if (stdin.isTTY) {
    handle('')
    return
  }

  const reader = stream()

  stdin.on('readable', reader.read)
  stdin.on('end', () => {
    handle(reader.value())
  })
}()


function handle(input) {
  let filename = 'fx'

  if (input === '') {
    if (args.length === 0 || (args.length === 1 && (args[0] === '-h' || args[0] === '--help'))) {
      stderr.write(usage)
      process.exit(2)
    }
    if (args.length === 1 && (args[0] === '-v' || args[0] === '--version')) {
      stderr.write(require('./package.json').version + '\n')
      process.exit(2)
    }
    if (args.length === 1 && args[0] === '--life') {
      require('./bang')
      return
    }

    input = fs.readFileSync(args[0])
    filename = path.basename(args[0])
    args.shift()
  }

  const json = JSON.parse(input)

  if (args.length === 0 && stdout.isTTY) {
    require('./fx')(filename, json)
    return
  }

  apply(json)
}


function apply(json) {
  let output

  try {
    output = args.reduce(reduce, json)
  } catch (e) {
    if (e !== skip) {
      throw e
    } else {
      return
    }
  }

  if (typeof output === 'undefined') {
    stderr.write('undefined\n')
  } else if (typeof output === 'string') {
    console.log(output)
  } else {
    const [text] = print(output)
    console.log(text)
  }
}


function select(cb) {
  return json => {
    if (!cb(json)) {
      throw skip
    }
    return json
  }
}


function stream() {
  let buff = ''
  let len = 0
  let depth = 0
  let isString = false
  let isStream = false
  let head = ''

  const check = (i) => {
    if (depth <= 0) {
      const input = buff.substring(0, len + i + 1)

      if (isStream) {
        if (head !== '') {
          const json = JSON.parse(head)
          apply(json)
          head = ''
        }

        const json = JSON.parse(input)
        apply(json)
      } else {
        isStream = true
        head = input
      }

      buff = buff.substring(len + i + 1)
      len = buff.length
    }
  }

  return {
    value() {
      return head + buff
    },
    read() {
      let chunk

      while ((chunk = stdin.read())) {
        len = buff.length
        buff += chunk

        for (let i = 0; i < chunk.length; i++) {
          if (isString) {
            if (chunk[i] === '"') {
              if ((i > 1 && chunk[i - 1] !== '\\') || buff[buff.length] !== '\\') {
                isString = false
                check(i)
              }
            }
            continue
          }

          if (chunk[i] === '{' || chunk[i] === '[') {
            depth++
          } else if (chunk[i] === '}' || chunk[i] === ']') {
            depth--
            check(i)
          } else if (chunk[i] === '"') {
            isString = true
          }
        }
      }
    }
  }
}