import { readFileSync } from "fs";
import path from "path";
import { Buffer } from "buffer";
import * as fs from "fs";
import type { DataItemCreateOptions } from "../../index";
import { SolanaSigner, bundleAndSignData, DataItem, createData, ArweaveSigner } from "../../index";
import base64url from "base64url";
const wallet0 = JSON.parse(readFileSync(path.join(__dirname, "test_key0.json")).toString());

describe("Creating and indexing a data item", function () {
  it("should create with all and get", async function () {
    // await arweave.wallets.ownerToAddress(wallet0.n)
    //   .then(async (r) => {
    //     await arweave.wallets.getBalance(r)
    //       .then(w => console.log(arweave.ar.winstonToAr(w)))
    //   });

    const _d: DataItemCreateOptions = {
      anchor: "Math.apt'#]gng(36).substring(30)",
      target: "OXcT1sVRSA5eGwt2k6Yuz8-3e3g9WJi5uSE99CWqsBs",
      tags: [{ name: "Content-Type", value: "image/png" }],
    };

    const signer = new ArweaveSigner(wallet0);

    const d = createData("hello", signer, _d);
    await d.sign(signer);

    expect(d.rawData.toString()).toEqual("hello");
    expect(d.owner).toBe(wallet0.n);
    expect(d.target).toBe("OXcT1sVRSA5eGwt2k6Yuz8-3e3g9WJi5uSE99CWqsBs");
    expect(d.anchor).toEqual(base64url.encode("Math.apt'#]gng(36).substring(30)"));
    expect(d.tags).toEqual([{ name: "Content-Type", value: "image/png" }]);
    expect(await DataItem.verify(d.getRaw())).toEqual(true);
  }, 5000000);

  it("should create with no target and get", async function () {
    const _d: DataItemCreateOptions = {
      anchor: "Math.apt'#]gng(36).substring(30)",
      tags: [
        {
          name: "testname",
          value: "testvalue",
        },
      ],
    };

    const signer = new ArweaveSigner(wallet0);

    const d = await createData("tasty", signer, _d);
    await d.sign(signer);
    expect(Buffer.from(d.rawData).toString()).toBe("tasty");
    expect(d.owner).toBe(wallet0.n);
    expect(d.target).toBe("");
    expect(d.anchor).toEqual(base64url.encode("Math.apt'#]gng(36).substring(30)"));
    expect(d.tags).toEqual([
      {
        name: "testname",
        value: "testvalue",
      },
    ]);
    expect(await DataItem.verify(d.getRaw())).toEqual(true);
  });

  it("should create with no anchor and get", async function () {
    const _d: DataItemCreateOptions = {
      target: "pFwvlpz1x_nebBPxkK35NZm522XPnvUSveGf4Pz8y4A",
      tags: [
        {
          name: "testname",
          value: "testvalue",
        },
      ],
    };

    const signer = new ArweaveSigner(wallet0);

    const d = await createData("tasty", signer, _d);
    await d.sign(signer);
    expect(Buffer.from(d.rawData).toString()).toBe("tasty");
    expect(d.owner).toBe(wallet0.n);
    expect(d.target).toBe("pFwvlpz1x_nebBPxkK35NZm522XPnvUSveGf4Pz8y4A");
    expect(d.anchor).toEqual("");
    expect(d.tags).toEqual([
      {
        name: "testname",
        value: "testvalue",
      },
    ]);
    expect(await DataItem.verify(d.getRaw())).toEqual(true);
  });

  it("should create with no target or anchor and get", async function () {
    const _d: DataItemCreateOptions = {
      tags: [
        {
          name: "testname",
          value: "testvalue",
        },
      ],
    };

    const signer = new ArweaveSigner(wallet0);

    const d = await createData("tasty", signer, _d);
    await d.sign(signer);
    expect(Buffer.from(d.rawData).toString()).toBe("tasty");
    expect(d.owner).toBe(wallet0.n);
    expect(d.target).toBe("");
    expect(d.anchor).toEqual("");
    expect(d.tags).toEqual([
      {
        name: "testname",
        value: "testvalue",
      },
    ]);
    expect(await DataItem.verify(d.getRaw())).toEqual(true);
  });

  it("Test Bundle", async function () {
    const signer = new ArweaveSigner(wallet0);
    const solSigner = new SolanaSigner("rUC3u5oz8W1Y2b8b2tq1K5AUWnXMiVV5o9Fx29yTJepFqFPfYPdwjainQhUvxfNuuhMJAGoawA3qYWzo8QhC5pj");
    const _dataItems = [
      createData("tasty", signer, {
        target: "pFwvlpz1x_nebBPxkK35NZm522XPnvUSveGf4Pz8y4A",
        anchor: "Math.randomgng(36).substring(30)",
        tags: [{ name: "x", value: "y" }],
      }),
      createData("tasty", solSigner, {
        target: "pFwvlpz1x_nebBPxkK35NZm522XPnvUSveGf4Pz8y4A",
        anchor: "Math.randomgng(36).substring(30)",
        tags: [{ name: "x", value: "y" }],
      }),
    ];

    const bundle = await bundleAndSignData([_dataItems[0], _dataItems[0]], signer);
    const dataItems = bundle.items;

    expect(bundle.length).toEqual(2);
    expect(dataItems.length).toEqual(2);
    expect(Buffer.from(dataItems[0].rawData).toString()).toBe("tasty");
    expect(dataItems[0].owner).toBe(wallet0.n);
    expect(Buffer.from(dataItems[0].target).toString()).toBe("pFwvlpz1x_nebBPxkK35NZm522XPnvUSveGf4Pz8y4A");
    expect(dataItems[0].anchor).toEqual(base64url.encode("Math.randomgng(36).substring(30)"));
    expect(dataItems[0].tags).toEqual([{ name: "x", value: "y" }]);
    expect(await DataItem.verify(dataItems[0].getRaw())).toEqual(true);
  });

  it("Test bugs", async function () {
    const signer = new ArweaveSigner(wallet0);
    const bundle = await bundleAndSignData([createData("1894", signer), createData("4242", signer)], signer);

    expect(bundle.get(1).rawData).toEqual(Buffer.from("4242"));
  });

  it("Test failed file verification", async function () {
    fs.writeFileSync("test", Buffer.from("hi"));
    const fileBinary = fs.readFileSync("test");
    expect(await DataItem.verify(fileBinary)).toEqual(false);
  });

  it("should verify", async function () {
    const signer = new ArweaveSigner(wallet0);

    const tags = [{ name: "gdf", value: "gfgdf" }];

    const items = [createData("1984", signer, { tags }), createData("4242", signer)];
    const bundle = await bundleAndSignData(items, signer);

    expect(await bundle.verify()).toEqual(true);
  });

  it("should get all correct data", async function () {
    const signer = new ArweaveSigner(wallet0);
    const d = { tags: [{ name: "", value: "" }] };

    const data = await createData("hi", signer, d);
    await data.sign(signer);
    expect(data.signatureType).toEqual(1);
    expect(data.owner).toEqual(wallet0.n);
    expect(data.anchor).toEqual("");
    expect(data.tags).toEqual([{ name: "", value: "" }]);
    expect(data.target).toEqual("");
    expect(data.rawData.toString()).toEqual("hi");
    expect(await DataItem.verify(data.getRaw())).toEqual(true);
  });

  // it("should verify large bundle", async function () {
  //   const signer = new ArweaveSigner(wallet0);
  //
  //   const tags = [{ name: "gdf", value: "gfgdf" }];
  //
  //   const items = [];
  //
  //   const img = fs.readFileSync("/media/josh/Extra/large/large.jpg");
  //
  //   for (let i = 0; i < 2; i++) {
  //     const item = createData(img, signer, { tags });
  //     items.push(item);
  //   }
  //
  //   const bundle = await bundleAndSignData(items, signer);
  //
  //   expect(await bundle.verify()).toEqual(true);
  // });

  // it("should bundle in loop", async function () {
  //   const signer = new ArweaveSigner(wallet0);
  //   const tags = [
  //     {
  //       name: "Content-Type",
  //       value: "image/png",
  //     },
  //   ];
  //   const data = {
  //     tags,
  //   };
  //   const items = new Array(25_000).fill(data);
  //   let now = performance.now();
  //   const ids = [];
  //   for (let i = 0; i < 3000; i++) {
  //     if (i % 1000 === 0) {
  //       const now2 = performance.now();
  //       console.log(`${i} - ${now2 - now}ms`);
  //       now = now2;
  //     }
  //     const item = await createData(
  //       await fs.promises
  //         .readFile("large_llama.png")
  //         .then((r) => Buffer.from(r.buffer)),
  //       signer,
  //       data
  //     );
  //
  //     const id = base64url(await item.sign(signer));
  //     ids.push(id);
  //     items[i] = item;
  //   }
  //   console.log(sizeof(items));
  //
  //   const bundle = await bundleAndSignData(items, signer);
  //
  //   const tx = await bundle.toTransaction(arweave, wallet0);
  //
  //   await arweave.transactions.sign(tx, wallet0);
  //
  //   const uploader = await arweave.transactions.getUploader(tx);
  //
  //   while (!uploader.isComplete) {
  //     await uploader.uploadChunk();
  //     console.log(
  //       `${uploader.pctComplete}% complete, ${uploader.uploadedChunks}/${uploader.totalChunks}`
  //     );
  //   }
  //
  //   console.log(ids.slice(0, 10));
  // }, 1000000000);
  //
  // it("Test unbundle", async function () {
  //   const signer = new ArweaveSigner(wallet0);
  //   const tags = [
  //     {
  //       name: "Content-Type",
  //       value: "text/html",
  //     },
  //   ];
  //   const data = { tags };
  //
  //   const num = 1;
  //   const items = new Array(num);
  //
  //   for (let i = 0; i < num; i++) {
  //     items[i] = await createData("hello", signer, data);
  //   }
  //   const bundle = await bundleAndSignData(items, signer);
  //
  //   console.log(bundle.verify());
  //
  //   const tx = await bundle.toTransaction(arweave, wallet0);
  //
  //   await arweave.transactions.sign(tx, wallet0);
  //
  //   console.log(tx.id);
  //   console.log(bundle.getIds());
  //
  //   const uploader = await arweave.transactions.getUploader(tx);
  //
  //   while (!uploader.isComplete) {
  //     await uploader.uploadChunk();
  //     console.log(
  //       `${uploader.pctComplete}% complete, ${uploader.uploadedChunks}/${uploader.totalChunks}`
  //     );
  //   }
  // }, 1000000);
  //
  // it("should not leak", function() {
  //
  //   const myTags = [
  //     { name: 'App-Name', value: 'myApp' },
  //     { name: 'App-Version', value: '1.0.0' }
  //   ];
  //
  //   const signer = new ArweaveSigner(wallet0);
  //
  //   for (let i = 0; i < 300000000; i++) {
  //     const opts = { tags: myTags };
  //     const data = new Uint8Array(1_000_000_000).fill(10);
  //     const item = createData(data, signer, opts);
  //     const used = process.memoryUsage();
  //     console.log(used);
  //     console.log(item);
  //   }
  // });
  //
  // it("should send loads", async function() {
  //   const signer = new ArweaveSigner(wallet0);
  //   const tags = [{ name: "Content-Type", value: "image/png" }];
  //   const data = createData(fs.readFileSync("large_llama.png"), signer, { tags });
  //
  //   await data.sign(signer);
  //   for (let i = 0; i < 2; i++) {
  //     console.log(data.id);
  //     await data.sendToBundler("http://bundler.arweave.net:10000");
  //   }
  // }, 50000);
  //
  // it('should parse', function() {
  //   const output = tagsParser.fromBuffer(Buffer.from("0342104170702d4e616d650a6d79417070164170702d56657273696f6e0a312e302e3000", "hex"));
  //   console.log(output);
  // });
  //
  // it("should fetch and index", async function() {
  //   const ardb = new ArDB(arweave);
  //
  //   const txs = await ardb.search("transactions")
  //     .min(770000)
  //     .max(770010)
  //     .from("OXcT1sVRSA5eGwt2k6Yuz8-3e3g9WJi5uSE99CWqsBs")
  //     .tags([
  //       { name: "Bundle-Format", values: "binary" },
  //       { name: "Bundle-Version", values: "2.0.0" }
  //     ])
  //     .findAll();
  //
  //   console.log(txs.length);
  //
  //   for (const tx of txs) {
  //     const data = await arweave.transactions.getData(tx.id).then(r => base64url.toBuffer(r as string));
  //     console.log(typeof data);
  //     const bundle = new Bundle(Buffer.from(data));
  //     try {
  //       console.log(await bundle.verify())
  //     } catch (e) {
  //       console.log("Error", false);
  //     }
  //   }
  // }, 1000000);
  //
  // it("should index", async function() {
  //   const bundleStr = fs.readFileSync(path.join(__dirname, "./IbLRUSkcIzpAn93o6KWC946p7XIjdeZiBeTW2JlbZL0"));
  //
  //   const bundle = new Bundle(bundleStr);
  //   console.log(bundle.getRaw().toString().indexOf("As a member of the Alura racing faction"));
  //   console.log(bundle.getRaw().slice(3168, 3168+2601).toString());
  //   console.log(bundle.getSizes());
  //   console.log(bundle.getIds());
  //   console.log(bundle.items[0].rawData.slice(1076).toString());
  //
  //
  //
  //   console.log(bundle.items[0].rawData.toString());
  //   const allIds = bundle.items[0];
  //   console.log(allIds.getRaw().toString())
  // });
  //
  // it("should not cause out of memory", async function () {
  //   const bundleStr = await axios.get(
  //     "https://arweave.net/YX4M5Hbeg_67bDGA0HapkPlMEwNf5vBAaiPIOk_Rcqg",
  //     { responseType: "arraybuffer" },
  //   );
  //
  //   const bundle = new Bundle(bundleStr.data);
  //   console.log("Length", bundle.length);
  //   console.log("Verify", await bundle.verify());
  //   console.log("ids", bundle.getIds());
  //   console.log("mem", process.memoryUsage());
  // }, 1000000);
});