FLO_Twitter/app/web/profile.js

var profileWebsocket, selfWebsocket,profileServerStatus;
var profiles;
var profileID, privKey, selfID, modSuperNode;

function viewProfile(){
  if( sessionStorage.profiles === undefined || sessionStorage.privKey === undefined || sessionStorage.selfID === undefined || sessionStorage.serverPass === undefined || sessionStorage.superNodeList === undefined ){
    alert("Login credentials failed! Returning to login page!");
    window.location.href = "index.html";
    return;
  }
  profiles = JSON.parse(sessionStorage.profiles);
  console.log(profiles);
  privKey = encrypt.retrieveShamirSecret(JSON.parse(sessionStorage.privKey));
  selfID = sessionStorage.selfID;
  var url = new URL(window.location.href);
  profileID = url.searchParams.get("floID");
  superNodeList = new Set(JSON.parse(sessionStorage.superNodeList));
  if(superNodeList.has(selfID))
    modSuperNode = true;
  kBucketObj.launchKBucket().then(result => {
    console.log(result)
    listProfiles();
    displayProfile(profileID);   
  }).catch(error => {
    console.log(error);
  });
}

function displayProfile(profileID){
  console.log(profileID);
  var errorMsg;
  if(!profileID)
    errorMsg = "Select a Profile to display";
  else if(!encrypt.validateAddr(profileID))
    errorMsg = "Invalid FLO ID";
  else if(!(profileID in profiles))
    errorMsg = "FLO ID not registered to FLO Tweet";

  if(errorMsg !== undefined){
    document.getElementById("profileBody").innerHTML = errorMsg;
    return;
  }

  console.log("displayProfile");
  document.getElementById("profileName").innerHTML=profiles[profileID].name;
  document.getElementById("profileFloID").innerHTML='@'+profileID;
  initselfWebSocket();
  displayTweetFromIDB(profileID).then(result => {
    connectToProfileServer(profileID).then(result => {
      console.log(result);
      profileServerStatus = true;
      pingProfileServerforNewTweets(profileID);
    }).catch(error => {
      console.log(error);
      pingSuperNodeforNewTweets(profileID);
      profileServerStatus = false;
    });
  }).catch(error => {
    console.log(error);
  }); 
}

function displayTweetFromIDB(floID){
  return new Promise((resolve,reject) => {
      var idb = indexedDB.open("FLO_Tweet");
      idb.onerror = (event) =>  { reject("Error in opening IndexedDB!") };
      idb.onsuccess = (event) =>  {
        var db = event.target.result;
        var obj = db.transaction("following", "readwrite").objectStore("following");
        obj.get(floID).onsuccess = (event) => {
          var followBtn = document.getElementById("follow-button");
          if(event.target.result === undefined){
            followBtn.innerHTML = "+ Follow";
            followBtn.value = "follow";
          }else{
            followBtn.innerHTML = "- Unfollow";
            followBtn.value = "unfollow";
          } 
          followBtn.disabled = false;
        }
        var obs = db.transaction("tweets", "readwrite").objectStore("tweets");
        obs.openCursor().onsuccess = (event) =>  {
          var cursor = event.target.result;
          if(cursor) {
            //console.log(cursor.value)
            if(cursor.value.floID == floID)
              createTweetElement(floID,cursor.value.time,cursor.value.data);
            cursor.continue();
          }else{
            resolve("Displayed Tweets from IDB!");
          }
        }
        db.close();
      };
  });
}

function listProfiles(){
  console.log("listProfiles");
  var profileList =  document.getElementById("profileList");
  profileList.innerHTML = "";
  for (p in profiles){
    var element =  document.createElement("div");
    element.setAttribute("class", "media");
    element.innerHTML = `<a href="profile.html?floID=${p}"><div class="media-body">
              <h5 class="media-heading">${profiles[p].name}</h5>
              <small>@${p}</small>
            </div></a>`
    profileList.appendChild(element);
  }
}

function createTweetElement(floID,time,tweet){
  var tweetDisplay = document.getElementById("profileBody");
  var element =  document.createElement("div");
  element.setAttribute("class", "media");
  element.innerHTML = `
            <div class="media-body">
              <h4 class="media-heading">${profiles[floID].name} <small>@${floID}</small></h4>
              <span>${tweet}</span>
              <ul class="nav nav-pills nav-pills-custom">             
                <li><a href="#"><span class="glyphicon glyphicon-share-alt"></span></a></li>
                <li><a href="#"><span class="glyphicon glyphicon-retweet"></span></a></li>
                <li><a href="#"><span class="glyphicon glyphicon-star"></span></a></li>
                <li><a href="#"><span class="glyphicon glyphicon-option-horizontal"></span></a></li>
                <li><span class="timestamp pull-right">${getTime(time)}</span></li>
              </ul>
            </div>`;
  tweetDisplay.insertBefore(element, tweetDisplay.firstChild);
}

function connectToProfileServer(floID){
  return new Promise((resolve,reject) => {
      profileWebsocket = new WebSocket("ws://"+profiles[floID].onionAddr+"/ws");
      profileWebsocket.onopen = (event) => { resolve("Connected to Profile Server!") };
      profileWebsocket.onerror = (event) =>  { reject("Profile Server is offline!") };
      profileWebsocket.onclose = (event) =>  { console.log("Disconnected from Profile Server!") };
      profileWebsocket.onmessage = (event) => { 
        console.log(event.data); 
        try{
          var data = JSON.parse(event.data);
          var id = data.id;
          data = data.data;
          if( floID!=data.floID || !encrypt.verify(data.tweet,data.sign,profiles[floID].pubKey))
            return
          storeTweet({floID:floID,time:data.time,data:data.tweet},id);
          createTweetElement(floID,data.time,data.tweet);
        }catch(error){
          console.log(error);
        }
      }
  });
}

function pingSuperNodeforNewTweets(floID){
  getLastTweetCount(floID).then(result => {
    var data = JSON.stringify({reqNewTweets:true,floID:floID,tid:result,requestor:selfID})
    sendDataToSuperNode(floID,data);
  }).catch(error => {
    console.log(error);
  });       
}

function pingProfileServerforNewTweets(floID){
  getLastTweetCount(floID).then(result => {
    console.log(profileWebsocket);
    profileWebsocket.send(`>${result}`);
    console.log("sent");
    resolve('Sent New tweet request to user server!');
  }).catch(error => {
    console.log(error);
  });               
}

function getLastTweetCount(floid){
  return new Promise((resolve,reject) => {
      var idb = indexedDB.open("FLO_Tweet");
      idb.onerror = (event) =>  { reject("Error in opening IndexedDB!") };
      idb.onsuccess = (event) =>  {
        var db = event.target.result;
        var lastTweet = db.transaction('lastTweet', "readwrite").objectStore('lastTweet');
        var lastTweetReq = lastTweet.get(floid);
        lastTweetReq.onsuccess = (event) => {
          var result = event.target.result;
          if(result === undefined)
            result = 0;
          resolve(result);
        }
        db.close();         
      };
  });
}

function initselfWebSocket(){
  selfWebsocket = new WebSocket("ws://"+location.host+"/ws");
  selfWebsocket.onopen = (event) => { 
    console.log("Connecting");
    var serverPass = encrypt.retrieveShamirSecret(JSON.parse(sessionStorage.serverPass));
    selfWebsocket.send("$"+serverPass);
  };
  selfWebsocket.onclose = (event) => { console.log("DISCONNECTED") };
  selfWebsocket.onmessage = (event) => {
    console.log(event.data); 
    if(event.data[0] == '$')
      return;
    try{
      data = JSON.parse(event.data);
        if(data.follow && encrypt.verify(selfID, data.sign, profiles[data.floID].pubKey)){
          var idb = indexedDB.open("FLO_Tweet");
          idb.onsuccess = (event) =>  {
            var db = event.target.result;
            var obs = db.transaction("followers", "readwrite").objectStore("followers");
            obs.add(data.sign,data.floID);
            db.close();
          };
          selfWebsocket.send(`F${data.floID}-${data.sign}`);
        }else if(data.unfollow && encrypt.verify(selfID, data.sign, profiles[data.floID].pubKey)){
          var idb = indexedDB.open("FLO_Tweet");
          idb.onsuccess = (event) =>  {
            var db = event.target.result;
            var obs = db.transaction("followers", "readwrite").objectStore("followers");
            obs.delete(data.floID);
            db.close();
          };
          selfWebsocket.send(`U${data.floID}`);
        }else if(data.message && data.to == selfID){
          var msg = encrypt.decryptMessage(data.secret,data.pubVal)
          if(encrypt.verify(msg,data.sign,profiles[data.from].pubKey))
            storeMsg({time:data.time,floID:data.from,text:msg,type:'R'});
        }else if(data.fromSuperNode && data.floID == profileID){
          var tid = data.tid;
          data = JSON.parse(data.data);
          if(encrypt.verify(data.tweet,data.sign,profiles[data.floID].pubKey)){
            storeTweet({floID:data.floID,time:data.time,data:data.tweet},tid);
            createTweetElement(data.floID,data.time,data.tweet);
          }
        }else if(modSuperNode){
          superNodeMode(data);
        }
    }catch(error){
      console.log(error)
    }
  };
  selfWebsocket.onerror = (event) => { console.log(event) };
}

function follow(){
  var followBtn = document.getElementById("follow-button");
  if(followBtn.value == 'follow'){
    var sign = encrypt.sign(profileID,privKey);
    var data = JSON.stringify({follow:true, floID:selfID, sign:sign});
    if(profileServerStatus)
      profileWebsocket.send(data);
    else{
      var SNdata = JSON.stringify({viaSuperNodeMsg:true,from:selfID,to:profileID,data:data});
      sendDataToSuperNode(profileID,SNdata);
    }
    selfWebsocket.send(`f${profileID}-${sign}`)
    var idb = indexedDB.open("FLO_Tweet");
    idb.onsuccess = (event) =>  {
      var db = event.target.result;
      var obs = db.transaction("following", "readwrite").objectStore("following");
      obs.add(sign,profileID);
      db.close();
    };
    followBtn.value = 'unfollow';
    followBtn.innerHTML = "- Unfollow";
  }
  else if(followBtn.value == 'unfollow'){
    var sign = encrypt.sign(profileID,privKey);
    var data = JSON.stringify({unfollow:true, floID:selfID, sign:sign});
    if(profileServerStatus)
      profileWebsocket.send(data);
    else{
      var SNdata = JSON.stringify({viaSuperNodeMsg:true,from:selfID,to:profileID,data:data});
      sendDataToSuperNode(profileID,SNdata);
    }
    selfWebsocket.send(`u${profileID}`)
    var idb = indexedDB.open("FLO_Tweet");
    idb.onsuccess = (event) =>  {
      var db = event.target.result;
      var obs = db.transaction("following", "readwrite").objectStore("following");
      obs.delete(profileID);
      db.close();
    };
    followBtn.value = 'follow';
    followBtn.innerHTML = "+ Follow";
  }
}