/*
    Private Keyboard
    Copyright (C) 2021 Kevin Froman VoidNetwork LLC

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.

*/
let lastActive = null
const defaultHosts = "<all_urls>";

let appCode = function (){
  let keyBuffer = ""
  let elementToSendBuffer = null
  let bufferTimeout = setTimeout(function(){}, 1)
  let popupEnabled = false
  let popupGetter = browser.storage.sync.get("keyboardprivacyprompt")
  popupGetter.then(function(val){
    popupEnabled = val['keyboardprivacyprompt']
    console.debug(popupEnabled)
  }, function(){})

  let clearSelect = function()
  {
    if (window.getSelection) {window.getSelection().removeAllRanges()}
    else if
      (document.selection) {
        document.selection.empty()
      }
  }
  let minValue = 75
  let maxValue = 150
  let time = 0
  let last = null

  let doPopup = function(e) {
    let text = ""
    let inputType = ""

    if (e.target.tagName === "INPUT"){
      inputType = e.target.getAttribute("type")

      if (Date.now() - time < 1000 && e.target === last){
        return
      }
      if (! inputType){
        text = prompt("[PrivateKeyboard]\n\nEnter text for the text field:", e.target.value)
      }
      else if (! ["text", "search", "email", "number"].includes(inputType.toLowerCase())){
        return
      }
      else{
        text = prompt("[PrivateKeyboard]\n\nEnter text for the " + inputType + " field:", e.target.value)
      }
      if (text !== null) {
        e.target.value = text
      }
      last = e.target
      clearSelect()
      time = Date.now()
    }

  }

  let popupMode = function(){

    if (popupEnabled){
      document.addEventListener('focusin', doPopup, true)
    }

  }

  let mainKeyboardPrivacy = function(){
    popupMode()

    function getRandNum(){
      let buf = new Uint8Array(1)
      while (true){
        window.crypto.getRandomValues(buf);

        if (buf[0] <= maxValue && buf[0] >= minValue){
          break;
        }
        buf = new Uint8Array(1);
      }
      return buf[0];
    }
    let up = getRandNum()
    let down = getRandNum()

    console.debug('Protecting keyboard biometrics on ' + document.location.href)

    function pausecomp(millis)
    {
      // Yes i know this wastes cpu. i don't like it either, but it seems a blocking
      // approach is needed to prevent spying event listeners from reading key events in *real time*
      // Might use an off-page buffer solution in the future
        var date = new Date();
        var curDate = null;
        do { curDate = new Date(); }
        while(curDate-date < millis);
    }

    setTimeout(function(){

      document.addEventListener('keyup', function(e){
        if (e.key.startsWith('Arrow') || e.key.startsWith('Page')){
          return true;
        }
        pausecomp(up);
        return true;
      }, )

      document.addEventListener('keydown', function(e){
        if (e.key.startsWith('Arrow') || e.key.startsWith('Page') ||

        e.key == "Enter" || e.key == "Control" || e.key == "Tab"
        ){
          return true;
        }
        else if (e.key.length > 1 && e.key.startsWith("F")){
          // Ignore function keys (unicode can also be >1 length so check for f)
          return true;
        }
        else{
        }

        if (popupEnabled){
          setTimeout(
            function(){doPopup(e)}, 10)
        }

        pausecomp(down);
        return true;
      })


    }, 100)
  }
  function checkForLANThenRun(){

    let lan = browser.storage.sync.get("keyboardprivacylan");
    lan.then(function(val){

      if (! val['keyboardprivacylan']){
        mainKeyboardPrivacy()
      }
      else{
        let hostname = document.location.hostname
        if (/^(10)\.(.*)\.(.*)\.(.*)$/.test(hostname)){
          //10.x.x.x
        }else if (/^(172)\.(1[6-9]|2[0-9]|3[0-1])\.(.*)\.(.*)$/.test(hostname)){
          //172.16.x.x - 172.31.255.255
        }else if (/^(192)\.(168)\.(.*)\.(.*)$/.test(hostname)){
          //192.168.x.x
        }else if (/^(127)\.(.*)\.(.*)\.(.*)$/.test(hostname)){
        }
        else if (hostname == '[::1]'){
        }
        else {
          mainKeyboardPrivacy()
          return
        }

        console.debug("Not running private keyboard because lan/loopback hostname")
      }

    }, function(val){mainKeyboardPrivacy()})

  }
  function shouldRunKeyboardPrivacy(value){
    if (typeof value.keyboardprivacywhitelist === 'undefined'){
      mainKeyboardPrivacy()
      return
    }
    let vals = value.keyboardprivacywhitelist.split(',')
    for (i = 0; i < vals.length; i++){
      if (vals[i] === document.location.hostname || 'www.' + vals[i] === document.location.hostname){
        console.debug(document.location.hostname + ' whitelisted for no keyboard privacy')
        return
      }
    }
    checkForLANThenRun()

  }
  function noKeyboardPrivacySettings(value){
    mainKeyboardPrivacy()
  }

  let whitelist = browser.storage.sync.get("keyboardprivacywhitelist");
  whitelist.then(shouldRunKeyboardPrivacy, noKeyboardPrivacySettings)


  browser.runtime.onMessage.addListener(request => {

    console.log(request.keys);
    if (request.keys == "GetSet"){
      keyBuffer = elementToSendBuffer.value
      return Promise.resolve({response: elementToSendBuffer.value});
    }

    clearTimeout(bufferTimeout)

    if (request.keys == "Backspace"){
      keyBuffer = keyBuffer.slice(0, -1)
    }
    else if(request.keys.length > 4){
       // pass
    }
    else{
      keyBuffer += request.keys;
    }
    if (request.isEmpty){
      keyBuffer = ""
      sendBuffer()
    }
    bufferTimeout = setTimeout(function(){sendBuffer()}, 100)
    return Promise.resolve({response: "ack"});
  });

  function sendBuffer(){

    if (elementToSendBuffer == null){
      bufferTimeout = setTimeout(function(){sendBuffer()}, 100)
      return
    }
    console.debug("sending buffer to " + elementToSendBuffer)

    elementToSendBuffer.value = keyBuffer

  }

  function setSendEl(e){
    let activeEl = document.activeElement
    if (activeEl.tagName != "INPUT" && activeEl.tagName != "TEXTAREA"){
      keyBuffer = ""
      return
    }

    console.debug("active el is " + activeEl.tagName)
    elementToSendBuffer = activeEl
    keyBuffer = activeEl.value
  }


  document.addEventListener('focus', setSendEl)
  document.addEventListener('click', setSendEl)


}

const dummyStr = ''

let defaultCode = dummyStr + appCode;
defaultCode = defaultCode.replace('function (){', 'function keyboardPrivacy(){')

async function register(hosts, code) {

  return await browser.contentScripts.register({
    matches: [hosts],
    js: [{code}],
    runAt: "document_idle",
    "allFrames": true
  });

}

var registered = register(defaultHosts, defaultCode + ' keyboardPrivacy()');