jaklis/lib/gchange.py

#!/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)