var expect = require("chai").expect;
var allTests = typeof window === "undefined" ?
               !!process.env.ALL_TESTS :
               window.ALL_TESTS;

var bitmessage = require("./lib");
var Int64 = bitmessage.Int64;
var Address = bitmessage.Address;
var wif = bitmessage.wif;
var varint = require("./lib/varint");
var bmcrypto = require("./lib/crypto");

describe("var_int", function() {
  it("should decode", function() {
    var res;
    expect(varint.decode.bind(null, Buffer([]))).to.throw(Error);
    expect(varint.decode.bind(null, Buffer("fd00", "hex"))).to.throw(Error);

    res = varint.decode(Buffer([123]));
    expect(res.value).to.equal(123);
    expect(res.length).to.equal(1);
    expect(res.rest.toString("hex")).to.equal("")

    res = varint.decode(Buffer("fd123456", "hex"));
    expect(res.value).to.equal(0x1234);
    expect(res.length).to.equal(3);
    expect(res.rest.toString("hex")).to.equal("56");

    res = varint.decode(Buffer("fe1234567890", "hex"));
    expect(res.value).to.equal(0x12345678);
    expect(res.length).to.equal(5);
    expect(res.rest.toString("hex")).to.equal("90");

    res = varint.decode(Buffer("ff0000001234567890", "hex"));
    expect(res.value == 0x1234567890).to.be.true;
    expect(res.length).to.equal(9);
    expect(res.rest.length).to.equal(0);
  });

  it("should check for lowest length on decode", function() {
    expect(varint.decode.bind(null, Buffer("fd00fc", "hex"))).to.throw(Error);
    expect(varint.decode.bind(null, Buffer("fe0000ffff", "hex"))).to.throw(Error);
    expect(varint.decode.bind(null, Buffer("ff00000000ffffffff", "hex"))).to.throw(Error);
  });

  it("should encode", function() {
    expect(varint.encode(123).toString("hex")).to.equal("7b");
    expect(varint.encode(0x1234).toString("hex")).to.equal("fd1234");
    expect(varint.encode(0x12345678).toString("hex")).to.equal("fe12345678");
    expect(varint.encode(0x1234567890).toString("hex")).to.equal("ff0000001234567890");
    expect(varint.encode(new Int64("0xffffffffffffffff")).toString("hex")).to.equal("ffffffffffffffffff");
    expect(varint.encode(Buffer("1234567890", "hex")).toString("hex")).to.equal("ff0000001234567890");
    expect(varint.encode.bind(null, -123)).to.throw(Error);
    expect(varint.encode.bind(null, Buffer("123456789012345678", "hex"))).to.throw(Error);
    expect(varint.encode.bind(null, "test")).to.throw(Error);
  });
});

describe("WIF", function() {
  var wifSign = "5JgQ79vTBusc61xYPtUEHYQ38AXKdDZgQ5rFp7Cbb4ZjXUKFZEV";
  var wifEnc = "5K2aL8cnsEWHwHfHnUrPo8QdYyRfoYUBmhAnWY5GTpDLbeyusnE";
  var signPrivateKey = Buffer("71c95d26c716a5e85e9af9efe26fb5f744dc98005a13d05d23ee92c77e038d9f", "hex");
  var encPrivateKey = Buffer("9f9969c93c2d186787a7653f70e49be34c03c4a853e6ad0c867db0946bc433c6", "hex");

  it("should decode", function() {
    return wif.decode(wifSign)
    .then(function(key1) {
      expect(Buffer.isBuffer(key1)).to.be.true;
      expect(key1.length).to.equal(32);
      expect(key1.toString("hex")).to.equal(signPrivateKey.toString("hex"));
      return wif.decode(wifEnc).then(function(key2) {
        expect(Buffer.isBuffer(key2)).to.be.true;
        expect(key2.length).to.equal(32);
        expect(key2.toString("hex")).to.equal(encPrivateKey.toString("hex"));
        return Address({signPrivateKey: key1, encPrivateKey: key2}).encode();
      });
    }).then(function(str) {
      expect(str).to.equal("BM-2cTux3PGRqHTEH6wyUP2sWeT4LrsGgy63z");
    });
  });

  it("should encode", function() {
    return wif.encode(signPrivateKey).then(function(wif1) {
      expect(wif1).to.equal(wifSign);
      return wif.encode(encPrivateKey);
    }).then(function(wif2) {
      expect(wif2).to.equal(wifEnc);
    });
  });
});

describe("Crypto", function() {
  it("should implement SHA-512 hash", function() {
    return bmcrypto.sha512(Buffer("test")).then(function(res) {
      expect(res.toString("hex")).to.equal("ee26b0dd4af7e749aa1a8ee3c10ae9923f618980772e473f8819a5d4940e0db27ac185f8a0e1d5f84f88bc887fd67b143732c304cc5fa9ad8e6f57f50028a8ff");
    });
  });

  it("should implement SHA-256 hash", function() {
    return bmcrypto.sha256(Buffer("test")).then(function(res) {
      expect(res.toString("hex")).to.equal("9f86d081884c7d659a2feaa0c55ad015a3bf4f1b2b0b822cd15d6c15b0f00a08");
    });
  });

  it("should implement RIPEMD-160 hash", function() {
    return bmcrypto.ripemd160(Buffer("test")).then(function(res) {
      expect(res.toString("hex")).to.equal("5e52fee47e6b070565f74372468cdc699de89107");
    });
  });

  it("should implement cryptographically secure PRNG", function() {
    var size = 100;
    var rnd = bmcrypto.randomBytes(size);
    expect(Buffer.isBuffer(rnd)).to.be.true;
    expect(rnd.length).to.equal(size);
    // Very simple statistical test.
    var bytes = {};
    var sum = 0;
    var value;
    for (var i = 0; i < size; i++) {
      value = rnd[i];
      sum += value;
      if (!bytes[value]) {
        bytes[value] = 0;
      }
      bytes[value]++;
      expect(bytes[value]).to.be.below(7);
    }
    // Ideal sum = (255 / 2) * size = 12750
    expect(sum).to.be.above(10000);
    expect(sum).to.be.below(15000);
  });
});

describe("Address", function() {
  it("should decode Bitmessage address", function() {
    return Address.decode("BM-2cTux3PGRqHTEH6wyUP2sWeT4LrsGgy63z")
      .then(function(addr) {
        expect(addr.version).to.equal(4);
        expect(addr.stream).to.equal(1);
        expect(addr.ripe.toString("hex")).to.equal("003ab6655de4bd8c603eba9b00dd5970725fdd56");
      });
  });

  it("should decode Bitmessage address badly formatted", function() {
    return Address.decode("  2cTux3PGRqHTEH6wyUP2sWeT4LrsGgy63z ")
      .then(function(addr) {
        expect(addr.version).to.equal(4);
        expect(addr.stream).to.equal(1);
        expect(addr.ripe.toString("hex")).to.equal("003ab6655de4bd8c603eba9b00dd5970725fdd56");
      });
  });

  it("should allow to generate new Bitmessage address", function() {
    this.timeout(10000);
    return Address.fromRandom().then(function(addr) {
      expect(addr.version).to.equal(4);
      expect(addr.stream).to.equal(1);
      expect(addr.signPrivateKey.length).to.equal(32);
      expect(addr.encPrivateKey.length).to.equal(32);
      return addr.encode().then(function(str) {
        expect(str.slice(0, 3)).to.equal("BM-");
        return Address.decode(str).then(function(addr2) {
          expect(addr2.version).to.equal(4);
          expect(addr2.stream).to.equal(1);
          expect(addr2.ripe.length).to.equal(20);
          expect(addr2.ripe[0]).to.equal(0);
        });
      });
    });
  });

  if (allTests) {
    it("should allow to generate shorter address", function() {
      this.timeout(60000);
      return Address.fromRandom({ripelen: 18}).then(function(addr) {
        return addr.getRipe({short: true}).then(function(ripe) {
          expect(ripe.length).to.be.at.most(18);
        });
      });
    });
  }
});