fs = require 'fs'
_ = require 'underscore'
vm = require 'vm'
md = require('markdown').markdown
coffee = require 'coffee-script'
CodeMirror = require 'codemirror/addon/runmode/runmode.node'
PDFDocument = require 'pdfkit'

Array::first ?= -> @[0]

Array::last ?= -> @[@length - 1]

process.chdir(__dirname)

# setup code mirror coffeescript mode
filename = require.resolve('codemirror/mode/coffeescript/coffeescript')
coffeeMode = fs.readFileSync filename, 'utf8'
vm.runInNewContext coffeeMode, CodeMirror: CodeMirror

# style definitions for markdown
styles =
  default:
    font: 'Times-Roman'
    fontSize: 12
    lineGap: 24
    align: 'left'
  meta:
    indent: 0
  title:
    align: 'center'
  para:
    indent: 72/2
  blockquote:
    indent: 0
    color: 'red'
    font: 'Times-Italic'
    marginLeft: 72
  em:
    font: 'Times-Italic'
  strong:
    font: 'Times-Bold'

# syntax highlighting colors
# based on Github's theme
colors =
  keyword: '#cb4b16'
  atom: '#d33682'
  number: '#009999'
  def: '#2aa198'
  variable: '#108888'
  'variable-2': '#b58900'
  'variable-3': '#6c71c4'
  property: '#2aa198'
  operator: '#6c71c4'
  comment: '#999988'
  string: '#dd1144'
  'string-2': '#009926'
  meta: '#768E04'
  qualifier: '#b58900'
  builtin: '#d33682'
  bracket: '#cb4b16'
  tag: '#93a1a1'
  attribute: '#2aa198'
  header: '#586e75'
  quote: '#93a1a1'
  link: '#93a1a1'
  special: '#6c71c4'
  default: '#002b36'

codeBlocks = []
lastType = null

# This class represents a node in the markdown tree, and can render it to pdf
class Node
  constructor: (tree) ->
    # special case for text nodes
    if typeof tree is 'string'
      @type = 'text'
      @text = tree
      return

    # console.dir tree
    @type = tree.shift()
    @attrs = {}

    if typeof tree[0] is 'object' and not Array.isArray tree[0]
      @attrs = tree.shift()

    # parse sub nodes
    @content = while tree.length
      new Node tree.shift()

    # console.log "type =", @type

    switch @type
      when 'header'
        @type = 'h' + @attrs.level

      # when 'code_block'
      #   # use code mirror to syntax highlight the code block
      #   code = @content[0].text
      #   @content = []
      #   CodeMirror.runMode code, 'coffeescript', (text, style) =>
      #     color = colors[style] or colors.default
      #     opts =
      #       color: color
      #       continued: text isnt '\n'

      #     @content.push new Node ['code', opts, text]

      #   @content[@content.length - 1]?.attrs.continued = false
      #   codeBlocks.push code

      when 'img'
        # images are used to generate inline example output
        # compiles the coffeescript to JS so it can be run
        # in the render method
        @type = 'example'
        code = codeBlocks[@attrs.alt]
        @code = coffee.compile code if code
        @height = +@attrs.title or 0

    @style = _.extend({}, styles.default, styles[@type])
    # @style.continued = @attrs.continued if @attrs.continued?
    # console.log @style

  # sets the styles on the document for this node
  setStyle: (doc) ->
    if @style.font
      doc.font @style.font

    if @style.fontSize
      doc.fontSize @style.fontSize

    if @style.color
      doc.fillColor @style.color
    else
      doc.fillColor 'black'

    # options = _.extend({}, @style)
    # # options.lineGap   = @style.lineGap
    # # options.align     = @style.align
    # # options.indent    = @style.indent
    # # options.link    = @attrs.href or false # override continued link
    # options.continued = @attrs.continued if @attrs.continued?
    # console.log "options =", options
    # return options
    null

  # renders this node and its subnodes to the document
  render: (doc, continued = false) ->
    console.log "rendering node: ", @
    switch @type
      when 'hr'
        doc.addPage()
      else
        # loop through subnodes and render them
        for fragment, index in @content
          if fragment.type is 'text'
            # add a new page for each heading, unless it follows another heading
            # if @type in ['h1', 'h2'] and lastType? and lastType isnt 'h1'
            #   doc.addPage()

            # set styles and whether this fragment is continued (for rich text wrapping)
            @setStyle doc
            # @style.continued ?= continued or index < @content.length - 1
            # @style.continued = continued

            # remove newlines unless this is code
            # unless @type is 'code'
            #   fragment.text = fragment.text.replace(/[\r\n]\s*/g, ' ')

            # console.log "rendering text. continued =", continued, 'attrs.continued =', @attrs.continued
            doc.text fragment.text, _.extend({}, @style, {continued: continued or index < @content.length - 1})
          else
            console.log "rendering fragment #{fragment.type}"
            fragment.render doc, index < @content.length - 1 and @type isnt 'bulletlist'
            # fragment.render doc, fragment.type == "para"

          lastType = @type

    if @style.marginTop
      doc.y += @style.marginTop
    # if @style.marginLeft
    #   doc.x += @style.marginLeft

# reads and renders a markdown/literate coffeescript file to the document
render = (doc, filename) ->
  doc.font 'Times-Roman'
  doc.fontSize 12

  codeBlocks = []
  content = fs.readFileSync(filename, 'utf8')

  body = ""

  metadata_pattern = /// ^
    ([\w.-]+) # key
    \:\       # colon
    \s*       # optional whitespace
    (.+)      # value
  $///

  metadata = {}

  for line in content.split("\n")
    if meta = line.match(metadata_pattern)
      key = meta[1]
      value = meta[2]
      metadata[key] = value
    else
      body += line + "\n"

  metadata.lastName ||= metadata.author?.split(" ").last()
  # console.log metadata

  # add header
  doc.text(metadata.author,     _.extend({}, styles.default, styles.meta))
  doc.text(metadata.instructor, _.extend({}, styles.default, styles.meta))
  doc.text(metadata.course,     _.extend({}, styles.default, styles.meta))
  doc.text(metadata.date,       _.extend({}, styles.default, styles.meta))
  doc.text(metadata.title,      _.extend({}, styles.default, styles.title))

  tree = md.parse body
  console.log tree
  tree.shift() # ignore 'markdown' first element

  while tree.length
    node = new Node tree.shift()
    node.render(doc)

  # add page numbers
  range = doc.bufferedPageRange() # => { start: 0, count: 2 }

  for i in [range.start...range.start + range.count]
    doc.switchToPage(i)
    doc.y = 72/2
    doc.x = 72
    doc.text "#{metadata.lastName} #{i + 1}",
      align: 'right'

  doc.flushPages()
  doc

# renders the title page of the guide
# renderTitlePage = (doc) ->
#   title = 'PDFKit Guide'
#   author = 'By Devon Govett'
#   version = 'Version ' + require('./package.json').version

#   doc.font 'fonts/AlegreyaSans-Light.ttf', 60
#   doc.y = doc.page.height / 2 - doc.currentLineHeight()
#   doc.text title, align: 'center'
#   w = doc.widthOfString(title)

#   doc.fontSize 20
#   doc.y -= 10
#   doc.text author,
#     align: 'center'
#     indent: w - doc.widthOfString(author)

#   doc.font styles.para.font, 10
#   doc.text version,
#     align: 'center'
#     indent: w - doc.widthOfString(version)

#   doc.addPage()

do ->
  doc = new PDFDocument
    bufferPages: true

  doc.pipe fs.createWriteStream('guide.pdf')
  render doc, 'test.coffee.md'

  doc.end()