summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKali Kaneko <kali@leap.se>2015-09-09 02:00:21 -0400
committerKali Kaneko <kali@leap.se>2015-09-09 02:00:21 -0400
commit6371dd9282b04a5496dae1a6f7b25a403989c167 (patch)
tree7ca8c8216d18aa4125bf189682f23f53184efe47
parent085e5ffa868aee3c2852a951336046e6d107e3f3 (diff)
add simple bonafide_cli for tests
-rw-r--r--bonafide/src/leap/bonafide/bonafide_cli.py97
1 files changed, 97 insertions, 0 deletions
diff --git a/bonafide/src/leap/bonafide/bonafide_cli.py b/bonafide/src/leap/bonafide/bonafide_cli.py
new file mode 100644
index 0000000..43c8ced
--- /dev/null
+++ b/bonafide/src/leap/bonafide/bonafide_cli.py
@@ -0,0 +1,97 @@
+# -*- coding: utf-8 -*-
+# bonafide_cli.py
+# Copyright (C) 2015 LEAP
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+"""
+Command Line interface for bonafide cli.
+"""
+
+# XXX Warning! Work in progress------------------------------------------------
+# This is just a demo, real cli will work against a persistent bonafide daemon.
+# XXX -------------------------------------------------------------------------
+
+import argparse
+import os
+import sys
+from getpass import getpass
+
+from colorama import init as color_init
+from colorama import Fore
+
+from twisted.cred.credentials import UsernamePassword
+from twisted.internet import reactor
+
+from leap.bonafide import provider
+from leap.bonafide import session
+
+COMMANDS = ('signup', 'authenticate')
+
+
+def _cbShutDown(ignored):
+ reactor.stop()
+
+
+def _authEb(failure):
+ print(Fore.RED + "[error] " + Fore.YELLOW +
+ failure.getErrorMessage() + Fore.RESET)
+
+
+def _display_token(result, _session):
+ if result == session.OK:
+ print('[ok] token--> ' + Fore.GREEN +
+ _session.token + Fore.RESET)
+
+def _display_registered(result, _session, _provider):
+ ok, user = result
+ if ok == session.OK:
+ print('[ok] registered username--> ' + Fore.GREEN +
+ '%s@%s' % (user, _provider))
+
+
+def run_command(command, _provider, username, password):
+ api = provider.Api('https://api.%s:4430' % _provider)
+ credentials = UsernamePassword(username, password)
+ cdev_pem = os.path.expanduser(
+ '~/.config/leap/providers/%s/keys/ca/cacert.pem' % _provider)
+ _session = session.Session(credentials, api, cdev_pem)
+
+ if command == 'authenticate':
+ d = _session.authenticate()
+ d.addCallback(_display_token, _session)
+ elif command == 'signup':
+ d = _session.signup(username, password)
+ d.addCallback(_display_registered, _session, _provider)
+ else:
+ print(Fore.YELLOW + "Command not implemented" + Fore.RESET)
+ sys.exit()
+
+ d.addErrback(_authEb)
+ d.addCallback(lambda _: _session.logout())
+ d.addBoth(_cbShutDown)
+ reactor.run()
+
+if __name__ == '__main__':
+ color_init()
+ description = (Fore.YELLOW + 'Manage and configure a LEAP Account '
+ 'using the bonafide protocol.' + Fore.RESET)
+ parser = argparse.ArgumentParser(description=description)
+ parser.add_argument('command', type=str, choices=COMMANDS)
+ parser.add_argument('--provider', dest='provider', required=True)
+ parser.add_argument('--username', dest='username', required=True)
+
+ ns = parser.parse_args()
+ password = getpass(
+ Fore.BLUE + '%s@%s password:' % (ns.username, ns.provider) + Fore.RESET)
+ run_command(ns.command, ns.provider, ns.username, password)