🚧 SSH Key handling

fixes #22

Author: guyzmo

git_repo/repo.py

@@ -30,6 +30,12 @@
     {self} [--path=<path>] [-v...] <target> (gist|snippet) fetch <gist> [<gist_file>]
     {self} [--path=<path>] [-v...] <target> (gist|snippet) create [--secret] <description> [<gist_path> <gist_path>...]
     {self} [--path=<path>] [-v...] <target> (gist|snippet) delete <gist> [-f]
+    {self} [--path=<path>] [-v...] <target> key (list|ls)
+    {self} [--path=<path>] [-v...] <target> key fetch <key_id>
+    {self} [--path=<path>] [-v...] <target> key add [--title <key_name>] <key_path>
+    {self} [--path=<path>] [-v...] <target> key delete <gist> [-f]
+    {self} [--path=<path>] [-v...] <target> key <user> (list|ls)
+    {self} [--path=<path>] [-v...] <target> key <user> fetch <key_id>
     {self} [--path=<path>] [-v...] <target> config [--config=<gitconfig>]
     {self} [-v...] config [--config=<gitconfig>]
     {self} --help
@@ -497,6 +503,37 @@ def do_gist_delete(self):
         log.info('Successfully deleted gist!')
         return 0
 
+    @register_action('key', 'list')
+    def do_key_list(self):
+        service = self.get_service(lookup_repository=False)
+        print_iter(service.key_list(self.user))
+        return 0
+
+    @register_action('key', 'fetch')
+    def do_key_fetch(self):
+        service = self.get_service(lookup_repository=False)
+        print(service.key_fetch(self.key_id, self.user))
+        return 0
+
+    @register_action('key', 'add')
+    @register_action('key', 'create')
+    def do_key_create(self):
+        service = self.get_service(lookup_repository=False)
+        name, key = service.key_create(self.key_path, self.title)
+        log.info('Successfully added SSH key \'{}\' with id {} !'.format(name, key))
+        return 0
+
+    @register_action('key', 'delete')
+    def do_ssh_delete(self):
+        service = self.get_service(lookup_repository=False)
+        if not self.force: # pragma: no cover
+            if not confirm('ssh key', self.gist_ref):
+                return -1
+
+        service.key_delete(self.gist_ref)
+        log.info('Successfully deleted ssh key!')
+        return 0
+
     @register_action('config')
     def do_config(self):
         from getpass import getpass

git_repo/services/ext/github.py

@@ -330,6 +330,71 @@ def request_fetch(self, user, repo, request, pull=False, force=False):
                 raise ResourceNotFoundError('Could not find opened request #{}'.format(request)) from err
             raise err
 
+    def _get_key(self, key_id, user=None):
+        try:
+            id_num = int(key_id)
+        except ValueError:
+            raise ResourceError('Key id shall be an integer')
+        if not user:
+            key = self.get_key(key_id)
+            if key:
+                return key
+        user = self.gh.user(user)
+        if not user:
+            raise ResourceNotFoundError('Could not find user {}'.format(user))
+        for key in user.iter_keys():
+            if key.id == id_num:
+                return key
+        raise ResourceNotFoundError('Key {} not found.'.format(key_id))
+
+    def key_create(self, pubkey_file, name=None):
+        with open(os.path.join(pubkey_file, pubkey_file), 'r') as f:
+            key_content = f.read()
+            if self._check_ssh_key(key_content):
+                key = self.gh.create_key(name, key_content)
+                if key:
+                    return key.title, key.id
+            raise ResourceError('Error creating key {}'.format(name))
+
+    def key_list(self, user=None):
+        current_user = not user or user == self.user
+        user = self.gh.user(user)
+        if not user:
+            raise ResourceNotFoundError('Could not find user {}'.format(user))
+        if current_user:
+            yield "{:>9}  {:<12}  {}"
+            yield ('key id', 'Modified', 'Key title')
+            for key in user.iter_keys():
+                    key = self.gh.key(key.id)
+                    keydate = datetime.strptime(
+                            key.last_modified,
+                            '%a, %d %b %Y %H:%M:%S GMT'
+                    )
+                    if keydate.year < datetime.now().year:
+                        date_fmt = "%b %d %Y"
+                    else:
+                        date_fmt = "%b %d %H:%M"
+
+                    yield (key.id, keydate.strftime(date_fmt), key.title)
+        else:
+            yield "{:>8}  {:<60}"
+            yield ('key id', 'contents')
+            for key in user.iter_keys():
+                yield (key.id, key.key[:59]+'…')
+
+    def key_fetch(self, key_id, user=None):
+        return self._get_key(key_id, user).key
+
+    def key_delete(self, pubkey):
+        try:
+            key = self.gh.key(int(pubkey))
+        except ValueError:
+            raise ResourceError('Key id shall be an integer')
+        if key:
+            if key.delete():
+                return
+        raise ResourceNotFoundError('Could not find key {}'.format(pubkey))
+
     @classmethod
     def get_auth_token(cls, login, password, prompt=None):
         import platform

git_repo/services/service.py

@@ -580,6 +580,26 @@ def request_fetch(self, user, repo, request, pull=False, force=False): #pragma:
     def request_create(self, user, repo, from_branch, onto_branch, title, description=None, auto_slug=False):
         raise NotImplementedError
 
+    @staticmethod
+    def _check_ssh_key(key):
+        if not key.startswith('ssh-'):
+            raise ResourceError('Key is invalid: does not start with "ssh-"')
+        if not len(key.strip().splitlines()) == 1:
+            raise ResourceError('Key is invalid: has more than one line')
+        return True
+
+    def key_create(self, pubkey_file, name):
+        raise NotImplementedError
+
+    def key_delete(self, pubkey):
+        raise NotImplementedError
+
+    def key_list(self):
+        raise NotImplementedError
+
+    def key_fetch(self):
+        raise NotImplementedError
+
     @property
     def get_parent_project_url(self, user, repo, rw=True): #pragma: no cover
         raise NotImplementedError