[pkg] initial migration of bitmask.{core,cli}

1 files changed, 76 insertions, 0 deletions

diff --git a/src/leap/bitmask/cli/user.py b/src/leap/bitmask/cli/user.py
new file mode 100644
index 00000000..dccfc7d5
--- /dev/null
+++ b/src/leap/bitmask/cli/user.py
@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+# user
+# Copyright (C) 2016 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/>.
+"""
+Bitmask Command Line interface: user
+"""
+import argparse
+import getpass
+import sys
+
+from leap.bitmask.cli import command
+
+
+class User(command.Command):
+    service = 'user'
+    usage = '''{name} user <subcommand>
+
+Bitmask account service
+
+SUBCOMMANDS:
+
+   create     Registers new user, if possible
+   auth       Logs in against the provider
+   logout     Ends any active session with the provider
+   active     Shows the active user, if any
+
+'''.format(name=command.appname)
+
+    commands = ['active']
+
+    def create(self, raw_args):
+        username = self.username(raw_args)
+        passwd = getpass.getpass()
+        self.data += ['signup', username, passwd]
+        return self._send(printer=command.default_dict_printer)
+
+    def auth(self, raw_args):
+        username = self.username(raw_args)
+        passwd = getpass.getpass()
+        self.data += ['authenticate', username, passwd]
+        return self._send(printer=command.default_dict_printer)
+
+    def logout(self, raw_args):
+        username = self.username(raw_args)
+        self.data += ['logout', username]
+        return self._send(printer=command.default_dict_printer)
+
+    def username(self, raw_args):
+        args = tuple([command.appname] + sys.argv[1:3])
+        parser = argparse.ArgumentParser(
+            description='Bitmask user',
+            prog='%s %s %s' % args)
+        parser.add_argument('username', nargs=1,
+                            help='username ID, in the form <user@example.org>')
+        subargs = parser.parse_args(raw_args)
+
+        username = subargs.username[0]
+        if not username:
+            self._error("Missing username ID but needed for this command")
+        if '@' not in username:
+            self._error("Username ID must be in the form <user@example.org>")
+
+        return username