import JsConfuser from "../../../src/index";

it("should extract properties", async () => {
  var code = `
    var TEST_OBJECT = {
      TEST_1: "Hello World",
      'TEST_2': 64
    }
    
    // ensures the original object is no longer defined
    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);

    input(TEST_OBJECT.TEST_1, TEST_OBJECT['TEST_2'], check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  // console.log(output);

  function input(a, b, c) {
    expect(a).toStrictEqual("Hello World");
    expect(b).toStrictEqual(64);
    expect(c).toStrictEqual(true);
  }

  eval(output);
});

it("should extract function properties correctly", async () => {
  var code = `
    var TEST_OBJECT = {
      isBoolean: x=>typeof x === "boolean",
      isString: function(x){return typeof x === "string"}
    }
    
    // ensures the original object is no longer defined
    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);

    input(TEST_OBJECT.isBoolean(true), TEST_OBJECT.isString(false), check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  // console.log(output);

  function input(a, b, c) {
    expect(a).toStrictEqual(true);
    expect(b).toStrictEqual(false);
    expect(c).toStrictEqual(true);
  }

  eval(output);
});

it("should not extract properties on with dynamically added keys", async () => {
  var code = `
    var TEST_OBJECT = {
      first_key: 1
    };

    TEST_OBJECT['DYNAMIC_PROPERTY'] = 1;
    TEST_OBJECT.DYNAMIC_PROPERTY = 1;


    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});

it("should not extract properties on with dynamically added keys even when in nested contexts", async () => {
  var code = `
    var TEST_OBJECT = {
      first_key: 1
    };

    (function(){
      TEST_OBJECT['DYNAMIC_PROPERTY'] = 1;
      TEST_OBJECT.DYNAMIC_PROPERTY = 1;
    })()
  


    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});

it("should not extract properties on objects with computed properties", async () => {
  var code = `
    
    var key = "111"
    var TEST_OBJECT = {
      [key]: 111,
    };


    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});

it("should not extract properties on objects with computed properties (string)", async () => {
  var code = `
    
    var v = "key";
    var TEST_OBJECT = {
      [v + ""]: 111,
    };


    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});

it("should not extract properties on objects when the object is referenced independently", async () => {
  var code = `
    
    var TEST_OBJECT = {
      isString: x=>typeof x === "string"
    };

    TEST_OBJECT;


    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});

it("should not extract properties on objects when the variable gets redefined", async () => {
  var code = `
    
    var TEST_OBJECT = {
      key: "value"
    };
    var TEST_OBJECT = {x: 1};


    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});

it("should not extract properties on objects when the variable gets reassigned", async () => {
  var code = `
    
    var TEST_OBJECT = {
      key: "value"
    };

    TEST_OBJECT = {key: "value_2"}


    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});

it("should not extract properties on objects with methods referencing 'this'", async () => {
  var code = `
    
    var TEST_OBJECT = {
      key: "value",
      getKey: function(){
        return this.key;
      }
    };


    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});

it("should not extract properties on objects when properties are dynamically deleted", async () => {
  var code = `
    
    var TEST_OBJECT = {
      key: "value"
    };

    delete TEST_OBJECT.key;

    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});

it("should not extract properties on objects with computed accessors", async () => {
  var code = `
    
    var TEST_OBJECT = {
      key: "value"
    };

    TEST_OBJECT["k" + "e" + "y"];


    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: true,
  });

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});

it("should properly use custom callback to exclude certain names from being changed", async () => {
  var code = `
    
    var TEST_OBJECT = {
      key: "value",
    };

    var check = false;
    eval(\`
      try {TEST_OBJECT} catch(e) {
        check = true;
      }
    \`);
    
    input(check);
  `;

  var seen = new Set();
  var output = await JsConfuser(code, {
    target: "browser",
    objectExtraction: (name) => {
      seen.add(name);

      return name !== "TEST_OBJECT";
    },
  });

  expect(seen.has("TEST_OBJECT")).toStrictEqual(true);

  function input(x) {
    expect(x).toStrictEqual(false);
  }

  eval(output);
});