From bc758cdc6282ec0facc9877ce1834cd8b4c916ba Mon Sep 17 00:00:00 2001
From: poka <poka@p2p.legal>
Date: Tue, 24 Nov 2020 10:40:13 +0100
Subject: [PATCH] Like,Unlike,Read like from a gchange profile

---
 gchange-profile.py |  54 ++++++++++
 lib/gchange.py     | 242 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 296 insertions(+)
 create mode 100755 gchange-profile.py
 create mode 100644 lib/gchange.py

diff --git a/gchange-profile.py b/gchange-profile.py
new file mode 100755
index 0000000..798d759
--- /dev/null
+++ b/gchange-profile.py
@@ -0,0 +1,54 @@
+#!/usr/bin/env python3
+
+import argparse, sys, os
+from os.path import join, dirname
+from shutil import copyfile
+from dotenv import load_dotenv
+from lib.gchange import ReadLikes, SendLikes, DeleteLikes
+
+# Get variables environment
+if not os.path.isfile('.env'):
+    copyfile(".env.template", ".env")
+dotenv_path = join(dirname(__file__), '.env')
+load_dotenv(dotenv_path)
+
+dunikey = os.getenv('DUNIKEY')
+pod = os.getenv('POD')
+if not dunikey or not pod:
+    sys.stderr.write("Please fill the path of your private key (PubSec), and a Cesium ES address in .env file\n")
+    sys.exit(1)
+
+
+# Parse arguments
+parser = argparse.ArgumentParser()
+
+subparsers = parser.add_subparsers()
+readLike_cmd = subparsers.add_parser('readlike', help="Lire les likes d'un profile")
+like_cmd = subparsers.add_parser('like', help="Liker un profile")
+deleteLike_cmd = subparsers.add_parser('deletelike', help="Supprimer un like")
+
+if len(sys.argv) <= 1 or not sys.argv[1] in ('readlike','like','deletelike'):
+    sys.stderr.write("Veuillez indiquer une commande valide:\n\n")
+    parser.print_help()
+    sys.exit(1)
+
+readLike_cmd.add_argument('-p', '--profile', help="Profile cible")
+
+like_cmd.add_argument('-p', '--profile', help="Profile cible")
+like_cmd.add_argument('-s', '--stars', type=int, help="Nombre d'étoile")
+
+deleteLike_cmd.add_argument('-i', '--id', help="ID du like à supprimer")
+
+args = parser.parse_args()
+
+# Build gchange class
+if sys.argv[1] == "readlike":
+    gchange = ReadLikes(dunikey, pod)
+    gchange.readLikes(args.profile)
+elif sys.argv[1] == "like":
+    gchange = SendLikes(dunikey, pod)
+    gchange.like(args.profile, args.stars)
+elif sys.argv[1] == "deletelike":
+    gchange = DeleteLikes(dunikey, pod)
+    gchange.deleteLike(args.id)
+
diff --git a/lib/gchange.py b/lib/gchange.py
new file mode 100644
index 0000000..dcc783e
--- /dev/null
+++ b/lib/gchange.py
@@ -0,0 +1,242 @@
+#!/usr/bin/env python3
+
+import os, sys, ast, requests, json, base58, base64, time, string, random, re
+from lib.natools import fmt, sign, get_privkey, box_decrypt, box_encrypt
+from hashlib import sha256
+from datetime import datetime
+from termcolor import colored
+
+PUBKEY_REGEX = "(?![OIl])[1-9A-Za-z]{42,45}"
+
+class ReadLikes:
+    def __init__(self, dunikey, pod):
+        # Get my pubkey from my private key
+        try:
+            self.dunikey = dunikey
+            if not dunikey:
+                raise ValueError("Dunikey is empty")
+        except:
+            sys.stderr.write("Please fill the path to your private key (PubSec)\n")
+            sys.exit(1)
+
+        self.issuer = get_privkey(dunikey, "pubsec").pubkey
+        self.pod = pod
+
+        if not re.match(PUBKEY_REGEX, self.issuer) or len(self.issuer) > 45:
+            sys.stderr.write("La clé publique n'est pas au bon format.\n")
+            sys.exit(1)
+
+    # Configure JSON document to send
+    def configDoc(self, profile):
+        timeSent = int(time.time())
+        if not profile: profile = self.issuer
+
+        data = {}
+        data['query'] = {}
+        data['query']['bool'] = {}
+        data['query']['bool']['filter'] = [
+            {'term': {'index': 'user'}},
+            {'term': {'type': 'profile'}},
+            {'term': {'id': profile}},
+            {'term': {'kind': 'STAR'}}
+        ]
+        data['query']['bool']['should'] = {'term':{'issuer': self.issuer}}
+        data['size'] = 1
+        data['_source'] = ['issuer','level']
+        data['aggs'] = {
+            'level_sum': {
+                'sum': {
+                    'field': 'level'
+                }
+            }
+        }
+
+        document = json.dumps(data)
+
+        # print(document)
+        # sys.exit(0)
+
+        return document
+
+    def sendDocument(self, document):
+
+        headers = {
+            'Content-type': 'application/json',
+        }
+
+        # Send JSON document and get JSON result
+        result = requests.post('{0}/like/record/_search'.format(self.pod), headers=headers, data=document)
+
+        if result.status_code == 200:
+            # print(result.text)
+            return result.text
+        else:
+            sys.stderr.write("Echec de l'envoi du document de lecture des messages...\n" + result.text + '\n')
+
+    def parseResult(self, result):
+        result = json.loads(result)
+        raw = result['hits']['hits']
+        for i in raw:
+            id = i['_id']
+            level = i['_source']['level']
+            print('ID:', id,'\nLevel:', level)
+
+    def readLikes(self, profile=False):
+        document = self.configDoc(profile)
+        result = self.sendDocument(document)
+        self.parseResult(result)
+
+
+
+
+#################### Sending class ####################
+
+
+
+
+class SendLikes:
+    def __init__(self, dunikey, pod):
+        # Get my pubkey from my private key
+        try:
+            self.dunikey = dunikey
+            if not dunikey:
+                raise ValueError("Dunikey is empty")
+        except:
+            sys.stderr.write("Please fill the path to your private key (PubSec)\n")
+            sys.exit(1)
+
+        self.issuer = get_privkey(dunikey, "pubsec").pubkey
+        self.pod = pod
+
+        if not re.match(PUBKEY_REGEX, self.issuer) or len(self.issuer) > 45:
+            sys.stderr.write("La clé publique n'est pas au bon format.\n")
+            sys.exit(1)
+
+    # Configure JSON document to send
+    def configDoc(self, profile, likes):
+        timeSent = int(time.time())
+
+        data = {}
+        data['version'] = 2
+        data['index'] = "user"
+        data['type'] = "profile"
+        data['id'] = profile
+        data['kind'] = "STAR"
+        data['level'] = likes
+        data['time'] = timeSent
+        data['issuer'] = self.issuer
+
+        document = json.dumps(data)
+
+        # Generate hash of document
+        hashDoc = sha256(document.encode()).hexdigest().upper()
+
+        # Generate signature of document
+        signature = fmt["64"](sign(hashDoc.encode(), get_privkey(self.dunikey, "pubsec"))[:-len(hashDoc.encode())]).decode()
+
+        # Build final document
+        data = {}
+        data['hash'] = hashDoc
+        data['signature'] = signature
+        signJSON = json.dumps(data)
+        finalJSON = {**json.loads(signJSON), **json.loads(document)}
+        finalDoc = json.dumps(finalJSON)
+
+        return finalDoc
+
+    def sendDocument(self, document):
+
+        headers = {
+            'Content-type': 'application/json',
+        }
+
+        # Send JSON document and get JSON result
+        result = requests.post('{0}/user/profile/:id/_like'.format(self.pod), headers=headers, data=document)
+
+        if result.status_code == 200:
+            print(colored("Profile liké avec succès !", 'green'))
+            return result.text
+        else:
+            sys.stderr.write("Echec de l'envoi du document de lecture des messages...\n" + result.text + '\n')
+
+
+    def like(self, profile, stars):
+        document = self.configDoc(profile, stars)
+        self.sendDocument(document)
+
+
+
+
+#################### Sending class ####################
+
+
+
+
+class DeleteLikes:
+    def __init__(self, dunikey, pod):
+        # Get my pubkey from my private key
+        try:
+            self.dunikey = dunikey
+            if not dunikey:
+                raise ValueError("Dunikey is empty")
+        except:
+            sys.stderr.write("Please fill the path to your private key (PubSec)\n")
+            sys.exit(1)
+
+        self.issuer = get_privkey(dunikey, "pubsec").pubkey
+        self.pod = pod
+
+        if not re.match(PUBKEY_REGEX, self.issuer) or len(self.issuer) > 45:
+            sys.stderr.write("La clé publique n'est pas au bon format.\n")
+            sys.exit(1)
+
+    # Configure JSON document to send
+    def configDoc(self, idLike):
+        timeSent = int(time.time())
+
+        data = {}
+        data['version'] = 2
+        data['index'] = "like"
+        data['type'] = "record"
+        data['id'] = idLike
+        data['issuer'] = self.issuer
+        data['time'] = timeSent
+
+        document = json.dumps(data)
+
+        # Generate hash of document
+        hashDoc = sha256(document.encode()).hexdigest().upper()
+
+        # Generate signature of document
+        signature = fmt["64"](sign(hashDoc.encode(), get_privkey(self.dunikey, "pubsec"))[:-len(hashDoc.encode())]).decode()
+
+        # Build final document
+        data = {}
+        data['hash'] = hashDoc
+        data['signature'] = signature
+        signJSON = json.dumps(data)
+        finalJSON = {**json.loads(signJSON), **json.loads(document)}
+        finalDoc = json.dumps(finalJSON)
+
+        return finalDoc
+
+    def sendDocument(self, document):
+
+        headers = {
+            'Content-type': 'application/json',
+        }
+
+        # Send JSON document and get JSON result
+        result = requests.post('{0}/history/delete'.format(self.pod), headers=headers, data=document)
+
+        if result.status_code == 200:
+            print(colored("Like supprimé avec succès !", 'green'))
+            return result.text
+        else:
+            sys.stderr.write("Echec de l'envoi du document de lecture des messages...\n" + result.text + '\n')
+
+
+    def deleteLike(self, idLike):
+        document = self.configDoc(idLike)
+        self.sendDocument(document)
+