import os,sys; sys.path.append(os.path.abspath(os.path.join(os.path.abspath(os.path.join(os.path.dirname(__file__),'..')),'..'))) from komrade import * from komrade.backend import * from komrade.backend.keymaker import * class Persona(Caller): def __init__(self, name=None, passphrase=DEBUG_DEFAULT_PASSPHRASE): super().__init__(name=name,passphrase=passphrase) if SHOW_STATUS: from komrade.cli import CLI self.cli = CLI(name=name, persona=self) # self.boot(create=False) # def boot(self,create=False): # # Do I already have my keys? # # yes? -- login # keys = self.keychain() # if keys.get('pubkey') and keys.get('privkey'): # self.log('booted!') # return True # # If not, forge them -- only once! # if not have_keys and create: # self.get_new_keys() def exists_locally_as_contact(self): return self.pubkey and not self.privkey def exists_locally_as_persona(self): return self.pubkey and self.privkey def exists_on_server(self): answer = self.phone.ring_ring({ '_route':'does_username_exist', 'name':self.name }) self.log('answer??',answer) return answer # login? # def login(self): # if keys.get('pubkey') and keys.get('privkey') def register(self, name = None, passphrase = None, is_group=None, show_intro=0,show_body=True): ## Defaults if name and not self.name: self.name=name if not name and self.name: name=self.name if not name and not self.name: name='' ## 1) Have name? if SHOW_STATUS and show_intro: name = self.cli.status_keymaker_part1(name) elif not name: name = input('@Keymaker: What is the name for this new account?\n@?: ') ## 2) Make pub public/private keys keypair = KomradeAsymmetricKey() pubkey,privkey = keypair.pubkey_obj,keypair.privkey_obj self.log(f'Keymaker has cut private and public keys:\n\n(1) {pubkey}\n\n(2) {privkey}') ## 3) Have passphrase? if SHOW_STATUS and not passphrase: passphrase = self.cli.status_keymaker_part2(name,passphrase,pubkey,privkey,hasher,self) else: while not passphrase: passphrase=getpass('Enter a memorable password to encrypt your private key with: ') self.passphrase=passphrase ## 4) Get hashed password passhash = hasher(passphrase) # self.log(f'''Keymaker has created a symmetric encryption cell using the disguised password:\n\n\t(2A) [Symmetric Encryption Key]\n\t({make_key_discreet_str(passhash)})''') ## 5) Encrypt private key privkey_decr = KomradeSymmetricKeyWithPassphrase(passphrase) privkey_encr = privkey_decr.encrypt(privkey.data) privkey_encr_obj = KomradeEncryptedAsymmetricPrivateKey(privkey_encr) self.log(f"For my private key, I will store it only on my device as it was encrypted by my password-generated key:\n\n[Encrypted Private Key]\n({make_key_discreet_str(privkey_encr_obj.data_b64)})") ## 6) Test keychain works privkey_decr2 = KomradeSymmetricKeyWithPassphrase(passphrase) assert privkey_decr2.decrypt(privkey_encr) == privkey.data self._keychain['pubkey']=pubkey self._keychain['privkey_encr']=privkey_encr_obj # self._keychain['privkey_decr']=privkey_decr # we should be able to reassemble privkey now? # self.log('this is my keychain now:') assert 'privkey' in self.keychain() # self.log('My keychain now looks like:',dict_format(self.keychain())) ## 6) More narration? if SHOW_STATUS: self.cli.status_keymaker_part3(privkey,privkey_decr,privkey_encr,passphrase) ## 7) Save data to server data = { 'name':name, 'pubkey': pubkey.data, } self.log('For my public key, I will be sending it to @Operator on the remote server:',dict_format(data,tab=2)) # ring operator # call from phone since I don't have pubkey on record on Op yet # self.log('my keychain:',self._keychain,pubkey,self.op._keychain) resp_msg_obj = self.ring_ring( { 'name':name, 'pubkey': pubkey.data, }, route='register_new_user' ) self.log('register got back from op:',dict_format(resp_msg_obj,tab=2)) def ring_ring(self,msg,route=None,**y): if type(msg)==dict and not ROUTE_KEYNAME in msg: msg[ROUTE_KEYNAME]=route return super().ring_ring(msg,caller=self,**y) def send_msg_to(self,msg,to_whom): msg = self.compose_msg_to(msg,to_whom) msg.encrypt() {'_route':'deliver_to', 'msg':msg} return self.ring_ring(msg) def test_register(): import random num = random.choice(list(range(0,1000))) botname=f'marx{str(num).zfill(3)}' marxbot = Persona(botname) # marxbot=Persona() marxbot.register() if __name__=='__main__': test_register() # marx = Persona('marx') # elon = Persona('elon') # marx.register() # # elon.register() # # person.register() # # print(person.pubkey) # # elon.send_msg_to('youre dumb',marx) # #Caller('elon').ring_ring({'_route':'say_hello','msg':'my dumb message to operator'}) # # print(marx.exists_on_server())