# -*- coding: utf-8 -*-
"""
Created on Wed Jun  8 22:21:45 2022

@author: Abinash.m
"""

import sys, time
import base64
import streamlit as st
def printError(msg=''):
    if msg != "":
        print("Error: " + msg)
    print('Usage: ')
    print(sys.argv[0] + " [private key file] [consumer ID] [Key version]")
    print('')
    print('Example: ')
    print(sys.argv[0] + " private_key.pem 44444444-23f9-3333-2222-111111111111 1")
    print('')
    print('')
    sys.exit()

def sign_data(private_key, data):
    from Crypto.PublicKey import RSA 
    from Crypto.Signature import PKCS1_v1_5 
    from Crypto.Hash import SHA256 
    from base64 import b64encode, b64decode 
    key = open(private_key, "r") 
    rsakey = RSA.importKey(key.read()) 
    signer = PKCS1_v1_5.new(rsakey) 
    digest = SHA256.new()
    digest.update(data.encode('utf-8')) 
    sign = signer.sign(digest) 
    return b64encode(sign)

def main(argv):
    global logData
    global hostname
    try:

        
        privateKey ='untitled1.txt'


    except:
        printError()
    epoch_time = int(time.time()) * 1000
    data = 'd23d5ad5-b016-4a76-b6bb-c6bae0372374' + '\n' + str(epoch_time) + '\n' + '1' + '\n'
    #print('Timestamp:', epoch_time)
    #print('Signature:' , sign_data(privateKey, data).decode())
    sig = sign_data(privateKey, data).decode()
    my_dict = {'Authorization Token':sig,'Timestamp':epoch_time}
    st.json(my_dict)

if __name__ == "__main__":
    main(sys.argv)