node-fftcg/inc/db.coffee

# libraries
bcrypt = (require 'bcrypt')
sqlite3 = (require 'sqlite3').verbose()
FFTCGLOG = new (require './console')('FFTCGDB')

# bruteforce countermeasure
saltRounds = 13

FFTCGDB = (filename, truncate) ->
  that = @
  @filename = filename

  @db = new sqlite3.Database @filename, (err) ->
    if err
      FFTCGLOG.error err.message

    else
      FFTCGLOG.log "Connected to '#{that.filename}'"

      that.db.run 'PRAGMA foreign_keys = ON;', (err) ->
        FFTCGLOG.error err.message if err

        if truncate == true
          that.db.run 'DROP TABLE IF EXISTS users;', (err) ->
            FFTCGLOG.error err.message if err
            that.db.run '''
              CREATE TABLE users (
                user integer PRIMARY KEY,
                login text NOT NULL COLLATE NOCASE,
                pwdhash text NOT NULL,
                UNIQUE(login)
              );
              ''', (err) ->
                FFTCGLOG.error err.message if err

          that.db.run 'DROP TABLE IF EXISTS decks;', (err) ->
            FFTCGLOG.error err.message if err
            that.db.run '''
              CREATE TABLE decks (
                deck integer PRIMARY KEY,
                user integer NOT NULL,
                json text,
                FOREIGN KEY (user) REFERENCES users (user)
                  ON DELETE CASCADE
              );
              ''', (err) ->
                FFTCGLOG.error err.message if err

          FFTCGLOG.log 'recreated DB'

  return

FFTCGDB::close = ->
  new Promise (resolve, reject) ->
    @db.close (err) ->
      if err
        FFTCGLOG.log "Error closing: '#{err.message}'"
        resolve 'ok'
      else
        FFTCGLOG.error  "Closed '#{@filename}'"
        reject 'db'

FFTCGDB::register = (login, password) ->
  that = @

  new Promise (resolve, reject) ->
    # validate user input
    if login == '' or password == ''
      # no user name or password given
      FFTCGLOG.log "reg: user name '#{login}' or password empty"
      reject 'invalid'

    # hash password
    bcrypt.hash password, saltRounds, (err, hash) ->
      if err
        FFTCGLOG.log "reg: hash fail for name '#{login}'"
        reject 'hash'

      # try creating row in users table
      stmt = that.db.prepare 'INSERT INTO users (login, pwdhash) VALUES (?, ?)'
      stmt.run [login, hash], (err) ->
        if err
          FFTCGLOG.log "reg: DB fail '#{err.code}' for name '#{login}'"
          stmt.finalize()
          # reduce attack surface, don't disclose user names
          reject 'db' # user already exists

        else
          FFTCGLOG.log "reg: OK '#{login}'"
          stmt.finalize()
          # registration successful
          resolve
            user: @lastID
            login: login

FFTCGDB::login = (login, password) ->
  that = @

  new Promise (resolve, reject) ->
    # get users table row
    stmt = that.db.prepare 'SELECT user, login, pwdhash FROM users WHERE login = ?'
    stmt.get [login], (err, row) ->
      if err
        FFTCGLOG.log "login: DB fail '#{err.code}' for name '#{login}'"
        stmt.finalize()
        reject 'db'

      else if not row
        # hash the password for timing attack reasons
        bcrypt.hash password, saltRounds, (err, hash) ->
          FFTCGLOG.log "login: nonexistent '#{login}'"
          stmt.finalize()
          # reduce attack surface, don't disclose user names
          reject 'login' # user doesnt exist

      else
        bcrypt.compare password, row.pwdhash, (err, res) ->
          if err
            FFTCGLOG.log "login: hash fail for name '#{login}'"
            reject 'hash'

          if res == true
            FFTCGLOG.log "login: OK '#{row.login}'"
            stmt.finalize()
            # login successful
            resolve
              user: row.user
              login: row.login

          else
            FFTCGLOG.log "login: wrong password for '#{login}'"
            stmt.finalize()
            # login failed
            reject 'login'

FFTCGDB::addDeck = (user, deckCards) ->
  that = @

  new Promise (resolve, reject) ->
    # try creating row in decks table
    stmt = that.db.prepare 'INSERT INTO decks (user, json) VALUES (?, ?)'
    stmt.run [user, JSON.stringify deckCards], (err) ->
      if err
        FFTCGLOG.log "addDeck: DB fail '#{err.code}' for id '#{user}'"
        stmt.finalize()
        reject 'db'

      else
        stmt.finalize()
        # deck added successfully, now add cards
        that.modDeck(@lastID, deckCards)
          .then (deckID) ->
            resolve deckID
          .catch (error) ->
            reject error

FFTCGDB::modDeck = (deckID, deckCards) ->
  that = @

  new Promise (resolve, reject) ->
    # delete old deck cards
    stmt = that.db.prepare 'DELETE FROM decks_cards WHERE deck = ?'
    stmt.run [deckID], (err) ->
      stmt.finalize()
      if err
        FFTCGLOG.log "modDeck: DB fail '#{err.code}' for deck '#{deckID}'"
        reject 'db'
      else
        stmt = that.db.prepare 'INSERT INTO decks_cards (deck, card, quant) VALUES (?, ?, ?)'
        # add new cards
        that.db.parallelize ->
          # needs to be done in several queries
          promiseCount = deckCards.length
          deckCards.forEach (card) ->
            stmt.run [deckID, card.id, card.quant], (err) ->
              if err
                FFTCGLOG.log "modDeck: DB fail '#{err.code}' for card '#{deckID}', '#{card.id}', '#{card.quant}'"
                stmt.finalize()
                reject 'db'
              else
                # check if all queries are done
                promiseCount -= 1
                if promiseCount == 0
                  FFTCGLOG.log "modDeck: OK '#{deckID}'"
                  stmt.finalize()
                  resolve deckID

FFTCGDB::getDecks = (user) ->
  that = @

  new Promise (resolve, reject) ->
    # try deleting correct row in decks table
    decks = {}
    stmt = that.db.prepare 'SELECT decks.deck, decks.json FROM decks INNER JOIN users ON decks.user = users.user WHERE users.user = ?'
    stmt.all [user], (err, rows) ->
      stmt.finalize()
      if err
        FFTCGLOG.log "getDeck: DB fail '#{err.code}' for deck '#{deckID}'"
        reject 'db'
      else
        FFTCGLOG.log "getDeck: OK '#{deckID}'"
        for row in rows
          decks[row.deck] = JSON.parse row.json
        resolve decks

FFTCGDB::delDeck = (deckID) ->
  that = @

  new Promise (resolve, reject) ->
    # try deleting correct row in decks table
    stmt = that.db.prepare 'DELETE FROM decks WHERE deck = ?'
    stmt.run [deckID], (err) ->
      stmt.finalize()
      if err
        FFTCGLOG.log "delDeck: DB fail '#{err.code}' for deck '#{deckID}'"
        reject 'db'
      else
        FFTCGLOG.log "delDeck: OK '#{deckID}'"
        resolve deckID


module.exports = FFTCGDB