{prng} = require 'crypto'
{expand_json,stub_json,hmac_obj} = require '../../lib/expand'
triplesec = require('triplesec')
pgp_utils = require('pgp-utils')
{katch,json_stringify_sorted} = pgp_utils.util
{KeyManager} = require('kbpgp').kb
{prng} = require 'crypto'
{constants,errors,alloc,Cryptocurrency} = require '../../'
{createHash} = require 'crypto'
{make_ids} = require '../../lib/base'
{make_esc} = require 'iced-error'
{errors} = require '../../lib/errors'

new_uid = () -> prng(15).toString('hex') + "19"
new_username = () -> "u_" + prng(5).toString('hex')

twiddle_hex = (s) ->
  b = Buffer.from(s, 'hex')
  b[0] ^= 1
  return b.toString('hex')

make_expansions = (v) ->
  ret = {}
  for obj in v
    key = prng(16)
    hmac = hmac_obj({obj, key}).toString('hex')
    ret[hmac] = { obj, key : key.toString('hex') }
  return ret

exports.happy = (T,cb) ->
  expansion_vals = [ ["a","b","c"], 300 ]
  expansions = make_expansions expansion_vals
  h = Object.keys(expansions)

  json = {
    a : 10
    b : "hi"
    c : d : e : h[0]
    f : [1,2,3]
    g : [
      { h : h[1] }
    ]
    i : h
  }
  res = expand_json { json, expansions }
  json.c.d.e = expansion_vals[0]
  json.g[0].h = expansion_vals[1]
  json.i[0] = expansion_vals[0]
  json.i[1] = expansion_vals[1]
  T.equal json, res, "worked"
  cb null

exports.sad1 = (T,cb) ->
  expansion_vals = [ ["a","b","c"], 300 ]
  expansions = make_expansions expansion_vals
  h = Object.keys(expansions)

  tests = [
    { expansions : { "a" : ["10"] }, json : 10, err : "bad hash" }
    { expansions : {}, json : 10, err : "hashcheck failure" }
    { expansions, json : 10, err : "Did not find expansion for" }
    { expansions : make_expansions(["💩"]), json : 10, err : "non-ASCII" }
    { expansions : {}, json : 10, err : "hashcheck failure" }
    { expansions : {}, json : 10, err : "bad hmac key" }
  ]
  expansion = expansions[h[0]]
  tests[1].expansions[h[0]] = { obj : ["a","b"], key : expansion.key }
  tests[4].expansions[h[0]] = { obj : expansion.obj, key : twiddle_hex(expansion.key) }
  tests[5].expansions[h[0]] = { obj : expansion.obj, key : expansion.key[0...12] }

  for t in tests
    err = null
    try
      expand_json t
    catch e
      err = e
    T.assert err?, "error for case #{JSON.stringify t}"
    T.assert err.toString().indexOf(t.err) >= 0, "found #{t.err}"

  cb null

exports.check_unstub_expand = (T,cb) ->
  json = {
    a : b : c : { z : 1, y : 2, w : 3}
    d : e : f : true
    g : h : false
  }
  orig = JSON.parse JSON.stringify json
  expansions = {}
  stub_json { json, expansions, path : "a.b.c" }
  stub_json { json, expansions, path : "d.e"}

  expanded = expand_json { expansions, json }
  T.equal orig, expanded, "got the right value out"

  cb null

exports.stubbed_chainlink = (T,cb) ->
  esc = make_esc cb, "@generate"
  await KeyManager.generate {}, esc defer km
  arg =
    user :
      local :
        username : new_username()
        uid : new_uid()
    host : "keybase.io"
    sig_eng : km.make_sig_eng()
    seq_type : constants.seq_types.PUBLIC
    cryptocurrency :
      address: "1BjgMvwVkpmmJ5HFGZ3L3H1G6fcKLNGT5h"
      type: "bitcoin"
    seqno : 1
    prev : null
    stub_paths : [ "body.cryptocurrency" ]
  btc = new Cryptocurrency arg
  await btc.generate_v2 esc defer out

  verifier = alloc out.inner.obj.body.type, arg
  varg = { armored : out.armored, skip_ids : true, make_ids : true, inner : out.inner.str, expansions : out.expansions}
  await verifier.verify_v2 varg, esc defer()
  for k,v of out.expansions
    v.obj.foo = "bar"
    out.expansions[k] = v
  await verifier.verify_v2 varg, defer err
  T.assert err?, "failed to destub"
  T.assert (err instanceof errors.ExpansionError), "right kind of error"
  T.assert (err.toString().indexOf("hashcheck failure in stub import") >= 0), "right error message"
  cb null