-import asyncio
+import io
import json
import logging
+import os
import random
+import shlex
import ssl
import string
+import subprocess
import websockets
+import yaml
+
+from juju.errors import JujuAPIError
log = logging.getLogger("websocket")
+
class Connection:
"""
- Usage:
- client = await Connection.connect(info)
+ Usage::
+
+ # Connect to an arbitrary api server
+ client = await Connection.connect(
+ api_endpoint, model_uuid, username, password, cacert)
+
+ # Connect using a controller/model name
+ client = await Connection.connect_model('local.local:default')
+
+ # Connect to the currently active model
+ client = await Connection.connect_current()
+
"""
- def __init__(self):
- self.__requestId__ = 0
+ def __init__(self, endpoint, uuid, username, password, cacert=None):
+ self.endpoint = endpoint
+ self.uuid = uuid
+ self.username = username
+ self.password = password
+ self.cacert = cacert
+
+ self.__request_id__ = 0
self.addr = None
self.ws = None
+ self.facades = {}
+
+ @property
+ def is_open(self):
+ if self.ws:
+ return self.ws.open
+ return False
def _get_ssl(self, cert):
return ssl.create_default_context(
result = json.loads(result)
return result
- async def rpc(self, msg):
- self.__requestID += 1
- msg['RequestId'] = self.__requestID
- if'Params' not in msg:
- msg['Params'] = {}
- if "Version" not in msg:
- msg['Version'] = self.facades[msg['Type']]
- outgoing = json.dumps(msg, indent=2)
+ async def rpc(self, msg, encoder=None):
+ self.__request_id__ += 1
+ msg['request-id'] = self.__request_id__
+ if'params' not in msg:
+ msg['params'] = {}
+ if "version" not in msg:
+ msg['version'] = self.facades[msg['type']]
+ outgoing = json.dumps(msg, indent=2, cls=encoder)
await self.ws.send(outgoing)
result = await self.recv()
- log.debug("send %s got %s", msg, result)
- if result and 'Error' in result:
- raise RuntimeError(result)
+ #log.debug("Send: %s", outgoing)
+ #log.debug("Recv: %s", result)
+ if result and 'error' in result:
+ raise JujuAPIError(result)
return result
+ async def clone(self):
+ """Return a new Connection, connected to the same websocket endpoint
+ as this one.
+
+ """
+ return await Connection.connect(
+ self.endpoint,
+ self.uuid,
+ self.username,
+ self.password,
+ self.cacert,
+ )
+
+ async def controller(self):
+ """Return a Connection to the controller at self.endpoint
+
+ """
+ return await Connection.connect(
+ self.endpoint,
+ None,
+ self.username,
+ self.password,
+ self.cacert,
+ )
+
@classmethod
- async def connect(cls, info):
- uuid = info.model_uuid
- controller = info.controller
- details = controller['details']
- endpoint = "wss://{}/model/{}/api".format(
- details['api-endpoints'][0],
- entity.model_uuid)
- client = cls()
- await client.dial(endpoint, details['ca-cert'])
- server_info = await client.login(info)
+ async def connect(cls, endpoint, uuid, username, password, cacert=None):
+ """Connect to the websocket.
+
+ If uuid is None, the connection will be to the controller. Otherwise it
+ will be to the model.
+
+ """
+ if uuid:
+ url = "wss://{}/model/{}/api".format(endpoint, uuid)
+ else:
+ url = "wss://{}/api".format(endpoint)
+ client = cls(endpoint, uuid, username, password, cacert)
+ await client.open(url, cacert)
+ server_info = await client.login(username, password)
client.build_facades(server_info['facades'])
- log.info("Driver connected to juju %s", endpoint)
+ log.info("Driver connected to juju %s", url)
+
return client
+ @classmethod
+ async def connect_current(cls):
+ """Connect to the currently active model.
+
+ """
+ jujudata = JujuData()
+ controller_name = jujudata.current_controller()
+ controller = jujudata.controllers()[controller_name]
+ endpoint = controller['api-endpoints'][0]
+ cacert = controller.get('ca-cert')
+ accounts = jujudata.accounts()[controller_name]
+ username = accounts['user']
+ password = accounts['password']
+ models = jujudata.models()[controller_name]
+ model_name = models['current-model']
+ model_uuid = models['models'][model_name]['uuid']
+
+ return await cls.connect(
+ endpoint, model_uuid, username, password, cacert)
+
+ @classmethod
+ async def connect_model(cls, model):
+ """Connect to a model by name.
+
+ :param str model: <controller>:<model>
+
+ """
+ controller_name, model_name = model.split(':')
+
+ jujudata = JujuData()
+ controller = jujudata.controllers()[controller_name]
+ endpoint = controller['api-endpoints'][0]
+ cacert = controller.get('ca-cert')
+ accounts = jujudata.accounts()[controller_name]
+ username = accounts['user']
+ password = accounts['password']
+ models = jujudata.models()[controller_name]
+ model_uuid = models['models'][model_name]['uuid']
+
+ return await cls.connect(
+ endpoint, model_uuid, username, password, cacert)
+
def build_facades(self, info):
self.facades.clear()
for facade in info:
- self.facades[facade['Name']] = facade['Versions'][-1]
-
- async def login(self, info):
- account = info.account
- result = await client.rpc({
- "Type": "Admin",
- "Request": "Login",
- "Version": 3,
- "Params": {
- "auth-tag": "user-{}".format(account['user']),
- "credentials": account['password'],
- "Nonce": "".join(random.sample(string.printable, 12)),
+ self.facades[facade['name']] = facade['versions'][-1]
+
+ async def login(self, username, password):
+ if not username.startswith('user-'):
+ username = 'user-{}'.format(username)
+
+ result = await self.rpc({
+ "type": "Admin",
+ "request": "Login",
+ "version": 3,
+ "params": {
+ "auth-tag": username,
+ "credentials": password,
+ "nonce": "".join(random.sample(string.printable, 12)),
}})
- return result['Response']
+ return result['response']
+
+
+class JujuData:
+ def __init__(self):
+ self.path = os.environ.get('JUJU_DATA') or '~/.local/share/juju'
+ self.path = os.path.abspath(os.path.expanduser(self.path))
+
+ def current_controller(self):
+ cmd = shlex.split('juju show-controller --format yaml')
+ output = subprocess.check_output(cmd)
+ output = yaml.safe_load(output)
+ return list(output.keys())[0]
+
+ def controllers(self):
+ return self._load_yaml('controllers.yaml', 'controllers')
+
+ def models(self):
+ return self._load_yaml('models.yaml', 'controllers')
+ def accounts(self):
+ return self._load_yaml('accounts.yaml', 'controllers')
+ def _load_yaml(self, filename, key):
+ filepath = os.path.join(self.path, filename)
+ with io.open(filepath, 'rt') as f:
+ return yaml.safe_load(f)[key]