From fc00373897c7f1de89cba4cca3c7bd3783f11d7d Mon Sep 17 00:00:00 2001
From: sachaaaaa <sacha@loki.network>
Date: Thu, 19 Sep 2019 16:14:06 +1000
Subject: [PATCH] Change LokiPublicChatAPI to LokiAppDotNetAPI, add annotations
 getter and setter

---
 js/modules/loki_public_chat_api.js | 159 ++++++++++++++++++++---------
 1 file changed, 108 insertions(+), 51 deletions(-)

diff --git a/js/modules/loki_public_chat_api.js b/js/modules/loki_public_chat_api.js
index 4823b0c1d..7d7dbfdd1 100644
--- a/js/modules/loki_public_chat_api.js
+++ b/js/modules/loki_public_chat_api.js
@@ -10,8 +10,7 @@ const PUBLICCHAT_CHAN_POLL_EVERY = 20 * 1000; // 20s
 const PUBLICCHAT_DELETION_POLL_EVERY = 5 * 1000; // 5s
 const PUBLICCHAT_MOD_POLL_EVERY = 30 * 1000; // 30s
 
-// singleton to relay events to libtextsecure/message_receiver
-class LokiPublicChatAPI extends EventEmitter {
+class LokiAppDotNetAPI extends EventEmitter {
   constructor(ourKey) {
     super();
     this.ourKey = ourKey;
@@ -24,8 +23,8 @@ class LokiPublicChatAPI extends EventEmitter {
       server => server.baseServerUrl === serverUrl
     );
     if (!thisServer) {
-      log.info(`LokiPublicChatAPI creating ${serverUrl}`);
-      thisServer = new LokiPublicServerAPI(this, serverUrl);
+      log.info(`LokiAppDotNetAPI creating ${serverUrl}`);
+      thisServer = new LokiAppDotNetServerAPI(this, serverUrl);
       this.servers.push(thisServer);
     }
     return thisServer;
@@ -57,7 +56,7 @@ class LokiPublicChatAPI extends EventEmitter {
   }
 }
 
-class LokiPublicServerAPI {
+class LokiAppDotNetServerAPI {
   constructor(chatAPI, url) {
     this.chatAPI = chatAPI;
     this.channels = [];
@@ -76,7 +75,7 @@ class LokiPublicServerAPI {
       channel => channel.channelId === channelId
     );
     if (!thisChannel) {
-      log.info(`LokiPublicChatAPI creating channel ${conversationId}`);
+      log.info(`LokiAppDotNetAPI creating channel ${conversationId}`);
       thisChannel = new LokiPublicChannelAPI(this, channelId, conversationId);
       this.channels.push(thisChannel);
     }
@@ -198,58 +197,19 @@ class LokiPublicServerAPI {
       return false;
     }
   }
-}
 
-class LokiPublicChannelAPI {
-  constructor(serverAPI, channelId, conversationId) {
-    // properties
-    this.serverAPI = serverAPI;
-    this.channelId = channelId;
-    this.baseChannelUrl = `channels/${this.channelId}`;
-    this.conversationId = conversationId;
-    this.conversation = ConversationController.get(conversationId);
-    this.lastGot = null;
-    this.modStatus = false;
-    this.deleteLastId = 1;
-    this.timers = {};
-    this.running = true;
-    // end properties
-
-    log.info(`registered LokiPublicChannel ${channelId}`);
-    // start polling
-    this.pollForMessages();
-    this.pollForDeletions();
-    this.pollForChannel();
-    this.pollForModerators();
-  }
-
-  stop() {
-    this.running = false;
-    if (this.timers.channel) {
-      clearTimeout(this.timers.channel);
-    }
-    if (this.timers.moderator) {
-      clearTimeout(this.timers.moderator);
-    }
-    if (this.timers.delete) {
-      clearTimeout(this.timers.delete);
-    }
-    if (this.timers.message) {
-      clearTimeout(this.timers.message);
-    }
-  }
 
   // make a request to the server
   async serverRequest(endpoint, options = {}) {
     const { params = {}, method, objBody, forceFreshToken = false } = options;
-    const url = new URL(`${this.serverAPI.baseServerUrl}/${endpoint}`);
+    const url = new URL(`${this.baseServerUrl}/${endpoint}`);
     if (params) {
       url.search = new URLSearchParams(params);
     }
     let result;
-    let { token } = this.serverAPI;
+    let { token } = this;
     if (!token) {
-      token = await this.serverAPI.getOrRefreshServerToken();
+      token = await this.getOrRefreshServerToken();
       if (!token) {
         log.error('NO TOKEN');
         return {
@@ -260,7 +220,7 @@ class LokiPublicChannelAPI {
     try {
       const fetchOptions = {};
       const headers = {
-        Authorization: `Bearer ${this.serverAPI.token}`,
+        Authorization: `Bearer ${this.token}`,
       };
       if (method) {
         fetchOptions.method = method;
@@ -281,7 +241,7 @@ class LokiPublicChannelAPI {
     try {
       response = await result.json();
     } catch (e) {
-      log.info(`serverRequest json arpse ${e}`);
+      log.warn(`serverRequest json arpse ${e}`);
       return {
         err: e,
         statusCode: result.status,
@@ -310,6 +270,103 @@ class LokiPublicChannelAPI {
     };
   }
 
+  async getUserAnnotations(pubKey) {
+    if (!pubKey){
+      log.warn('No pubkey provided to getUserAnnotations!');
+      return [];
+    }
+    const res = await this.serverRequest(`users/@${pubKey}`, {
+      method: 'GET',
+      params: {
+        include_user_annotations: 1,
+      },
+    });
+
+    if (res.err || !res.response || !res.response.data) {
+      if (res.err) {
+        log.error(`Error ${res.err}`);
+      }
+      return [];
+    }
+
+    return res.response.data.annotations || [];
+  }
+
+  // Only one annotation at a time
+  async setSelfAnnotation(type, value) {
+    let annotation;
+    const doDelete = !value;
+
+    if (doDelete) {
+      // to delete annotation, omit the "value" field
+      annotation = {
+        type,
+      };
+    } else {
+      annotation = {
+        type,
+        value,
+      };
+    }
+
+    const res = await this.serverRequest('users/me', {
+      method: 'PATCH',
+      objBody: {
+        annotations: [annotation],
+      },
+    });
+
+    if (!res.err && res.response) {
+      return res.response;
+    }
+
+    return false;
+  }
+}
+
+class LokiPublicChannelAPI {
+  constructor(serverAPI, channelId, conversationId) {
+    // properties
+    this.serverAPI = serverAPI;
+    this.channelId = channelId;
+    this.baseChannelUrl = `channels/${this.channelId}`;
+    this.conversationId = conversationId;
+    this.conversation = ConversationController.get(conversationId);
+    this.lastGot = null;
+    this.modStatus = false;
+    this.deleteLastId = 1;
+    this.timers = {};
+    this.running = true;
+    // end properties
+
+    log.info(`registered LokiPublicChannel ${channelId}`);
+    // start polling
+    this.pollForMessages();
+    this.pollForDeletions();
+    this.pollForChannel();
+    this.pollForModerators();
+  }
+
+  stop() {
+    this.running = false;
+    if (this.timers.channel) {
+      clearTimeout(this.timers.channel);
+    }
+    if (this.timers.moderator) {
+      clearTimeout(this.timers.moderator);
+    }
+    if (this.timers.delete) {
+      clearTimeout(this.timers.delete);
+    }
+    if (this.timers.message) {
+      clearTimeout(this.timers.message);
+    }
+  }
+
+  serverRequest(endpoint, options = {}) {
+    return this.serverAPI.serverRequest(endpoint, options);
+  }
+
   // get moderation actions
   async pollForModerators() {
     try {
@@ -669,4 +726,4 @@ class LokiPublicChannelAPI {
   }
 }
 
-module.exports = LokiPublicChatAPI;
+module.exports = LokiAppDotNetAPI;