Update sdk

coco/app.py

@@ -7,6 +7,8 @@ from .config import Config
 from .sshd import SSHServer
 from .ws import WSServer
 from .logging import create_logger
+from .sdk import AppService
+from .auth import AppAccessKey
 
 
 __version__ = '0.4.0'
@@ -46,6 +48,7 @@ class Coco:
         self.name = name
         self.lock = threading.Lock()
         self.stop_evt = threading.Event()
+        self.service = None
 
         if name is None:
             self.name = self.config['NAME']
@@ -62,6 +65,7 @@ class Coco:
     def prepare(self):
         self.sshd = SSHServer(self)
         self.ws = WSServer(self)
+        self.initial_service()
 
     def heartbeat(self):
         pass
@@ -117,6 +121,9 @@ class Coco:
             except:
                 pass
 
+    def initial_service(self):
+        self.service = AppService(self)
+
     def monitor_session(self):
         pass
 

coco/auth.py

@@ -3,14 +3,26 @@
 #
 
 import os
-import six
 import logging
-from io import IOBase
+import time
 
 from . import utils
 from .exception import LoadAccessKeyError
 
 
+def make_signature(access_key_secret, date=None):
+    if isinstance(date, bytes):
+        date = date.decode("utf-8")
+    if isinstance(date, int):
+        date_gmt = utils.http_date(date)
+    elif date is None:
+        date_gmt = utils.http_date(int(time.time()))
+    else:
+        date_gmt = date
+    data = str(access_key_secret) + "\n" + date_gmt
+    return utils.content_md5(data)
+
+
 class AccessKeyAuth(object):
     def __init__(self, access_key_id, access_key_secret):
         self.id = access_key_id
@@ -48,7 +60,8 @@ class SessionAuth(object):
 
 
 class Auth(object):
-    def __init__(self, token=None, access_key_id=None, access_key_secret=None,
+    def __init__(self, token=None, access_key_id=None,
+                 access_key_secret=None,
                  session_id=None, csrf_token=None):
 
         if token is not None:
@@ -58,8 +71,8 @@ class Auth(object):
         elif session_id and csrf_token:
             self.instance = SessionAuth(session_id, csrf_token)
         else:
-            raise OSError('Need token or access_key_id, access_key_secret '
-                          'or session_id, csrf_token')
+            raise SyntaxError('Need token or access_key_id, access_key_secret '
+                              'or session_id, csrf_token')
 
     def sign_request(self, req):
         return self.instance.sign_request(req)
@@ -70,22 +83,27 @@ class AccessKey(object):
         self.id = id
         self.secret = secret
 
-    def clean(self, value, delimiter=':', silent=False):
+    @staticmethod
+    def clean(value, delimiter=':', silent=False):
         try:
-            self.id, self.secret = value.split(delimiter)
+            id, secret = value.split(delimiter)
         except (AttributeError, ValueError) as e:
             if not silent:
                 raise LoadAccessKeyError(e)
+            return '', ''
         else:
-            return ':'.join([self.id, self.secret])
+            return id, secret
 
-    def load_from_env(self, env, delimiter=':', silent=False):
+    @classmethod
+    def load_from_env(cls, env, **kwargs):
         value = os.environ.get(env)
-        return self.clean(value, delimiter, silent)
+        id, secret = cls.clean(value, **kwargs)
+        return cls(id, secret)
 
-    def load_from_f(self, f, delimiter=':', silent=False):
+    @classmethod
+    def load_from_f(cls, f, **kwargs):
         value = ''
-        if isinstance(f, six.string_types) and os.path.isfile(f):
+        if isinstance(f, str) and os.path.isfile(f):
             f = open(f)
         if hasattr(f, 'read'):
             for line in f:
@@ -93,10 +111,11 @@ class AccessKey(object):
                     value = line.strip()
                     break
             f.close()
-        return self.clean(value, delimiter, silent)
+        id, secret = cls.clean(value, **kwargs)
+        return cls(id, secret)
 
     def save_to_f(self, f, silent=False):
-        if isinstance(f, six.string_types):
+        if isinstance(f, str):
             f = open(f, 'w')
         try:
             f.write(str('{0}:{1}'.format(self.id, self.secret)))
@@ -113,7 +132,6 @@ class AccessKey(object):
 
     def __str__(self):
         return '{0}:{1}'.format(self.id, self.secret)
-
     __repr__ = __str__
 
 
@@ -133,7 +151,7 @@ class ServiceAccessKey(AccessKey):
     default_key_store = os.path.join(os.environ.get('HOME', ''), '.access_key')
 
     def __init__(self, id=None, secret=None, config=None):
-        super(ServiceAccessKey, self).__init__(id=id, secret=secret)
+        super().__init__(id=id, secret=secret)
         self.config = config or {}
         self._key_store = None
         self._key_env = None

coco/config.py

@@ -263,20 +263,4 @@ class Config(dict):
         return '<%s %s>' % (self.__class__.__name__, dict.__repr__(self))
 
 
-API_URL_MAPPING = {
-    'terminal-register': '/api/applications/v1/terminal/register/',
-    'terminal-heatbeat': '/api/applications/v1/terminal/heatbeat/',
-    'send-proxy-log': '/api/audits/v1/proxy-log/receive/',
-    'finish-proxy-log': '/api/audits/v1/proxy-log/%s/',
-    'send-command-log': '/api/audits/v1/command-log/',
-    'send-record-log': '/api/audits/v1/record-log/',
-    'user-auth': '/api/users/v1/auth/',
-    'user-assets': '/api/perms/v1/user/%s/assets/',
-    'user-asset-groups': '/api/perms/v1/user/%s/asset-groups/',
-    'user-asset-groups-assets': '/api/perms/v1/user/my/asset-groups-assets/',
-    'assets-of-group': '/api/perms/v1/user/my/asset-group/%s/assets/',
-    'my-profile': '/api/users/v1/profile/',
-    'system-user-auth-info': '/api/assets/v1/system-user/%s/auth-info/',
-    'validate-user-asset-permission':
-        '/api/perms/v1/asset-permission/user/validate/',
-}
+

coco/exception.py

@@ -12,3 +12,6 @@ class LoadAccessKeyError(Exception):
 class RequestError(Exception):
     pass
 
+
+class ResponseError(Exception):
+    pass

coco/forward.py

@@ -19,7 +19,7 @@ class ProxyServer:
     def __init__(self, app, client):
         self.app = app
         self.client = client
-        self.request = client.request
+        self.request = client.do
         self.server = None
         self.connecting = True
 

coco/interactive.py

@@ -18,7 +18,7 @@ class InteractiveServer:
     def __init__(self, app, client):
         self.app = app
         self.client = client
-        self.request = client.request
+        self.request = client.do
 
     def display_banner(self):
         self.client.send(char.CLEAR_CHAR)

coco/sshd.py

@@ -86,7 +86,7 @@ class SSHServer:
         self.dispatch(client)
 
     def dispatch(self, client):
-        request_type = client.request.type
+        request_type = client.do.type
         if request_type == 'pty':
             InteractiveServer(self.app, client).activate()
         elif request_type == 'exec':

coco/utils.py

@@ -4,6 +4,7 @@ from __future__ import unicode_literals
 
 import hashlib
 import re
+import os
 import threading
 import base64
 import calendar
@@ -11,12 +12,11 @@ import time
 import datetime
 from io import StringIO
 
+import paramiko
 import pyte
 import pytz
 from email.utils import formatdate
 
-import paramiko
-from dotmap import DotMap
 
 
 try:
@@ -24,9 +24,6 @@ try:
 except ImportError:
     from queue import Queue, Empty
 
-from .compat import to_string, to_bytes
-
-
 
 def ssh_key_string_to_obj(text):
     key_f = StringIO(text)
@@ -258,21 +255,6 @@ def b64encode_as_string(data):
     return to_string(base64.b64encode(data))
 
 
-
-def make_signature(access_key_secret, date=None):
-    if isinstance(date, bytes):
-        date = date.decode("utf-8")
-    if isinstance(date, int):
-        date_gmt = http_date(date)
-    elif date is None:
-        date_gmt = http_date(int(time.time()))
-    else:
-        date_gmt = date
-
-    data = str(access_key_secret) + "\n" + date_gmt
-    return content_md5(data)
-
-
 def split_string_int(s):
     """Split string or int
 
@@ -335,17 +317,6 @@ def timestamp_to_datetime_str(ts):
     return dt.strftime(datetime_format)
 
 
-def to_dotmap(data):
-    """将接受dict转换为DotMap"""
-    if isinstance(data, dict):
-        data = DotMap(data)
-    elif isinstance(data, list):
-        data = [DotMap(d) for d in data]
-    else:
-        raise ValueError('Dict or list type required...')
-    return data
-
-
 class MultiQueue(Queue):
     def mget(self, size=1, block=True, timeout=5):
         items = []