107 lines
2.4 KiB
Python
107 lines
2.4 KiB
Python
|
# Copyright (C) 2018-2019 Garmin Ltd.
|
||
|
|
#
|
||
|
|
# SPDX-License-Identifier: GPL-2.0-only
|
||
|
|
#
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
from contextlib import closing
|
||
|
|
import itertools
|
||
|
|
import json
|
||
|
|
from collections import namedtuple
|
||
|
|
from urllib.parse import urlparse
|
||
|
|
from bb.asyncrpc.client import parse_address, ADDR_TYPE_UNIX, ADDR_TYPE_WS
|
||
|
|
|
||
|
|
User = namedtuple("User", ("username", "permissions"))
|
||
|
|
|
||
|
|
def create_server(
|
||
|
|
addr,
|
||
|
|
dbname,
|
||
|
|
*,
|
||
|
|
sync=True,
|
||
|
|
upstream=None,
|
||
|
|
read_only=False,
|
||
|
|
db_username=None,
|
||
|
|
db_password=None,
|
||
|
|
anon_perms=None,
|
||
|
|
admin_username=None,
|
||
|
|
admin_password=None,
|
||
|
|
):
|
||
|
|
def sqlite_engine():
|
||
|
|
from .sqlite import DatabaseEngine
|
||
|
|
|
||
|
|
return DatabaseEngine(dbname, sync)
|
||
|
|
|
||
|
|
def sqlalchemy_engine():
|
||
|
|
from .sqlalchemy import DatabaseEngine
|
||
|
|
|
||
|
|
return DatabaseEngine(dbname, db_username, db_password)
|
||
|
|
|
||
|
|
from . import server
|
||
|
|
|
||
|
|
if "://" in dbname:
|
||
|
|
db_engine = sqlalchemy_engine()
|
||
|
|
else:
|
||
|
|
db_engine = sqlite_engine()
|
||
|
|
|
||
|
|
if anon_perms is None:
|
||
|
|
anon_perms = server.DEFAULT_ANON_PERMS
|
||
|
|
|
||
|
|
s = server.Server(
|
||
|
|
db_engine,
|
||
|
|
upstream=upstream,
|
||
|
|
read_only=read_only,
|
||
|
|
anon_perms=anon_perms,
|
||
|
|
admin_username=admin_username,
|
||
|
|
admin_password=admin_password,
|
||
|
|
)
|
||
|
|
|
||
|
|
(typ, a) = parse_address(addr)
|
||
|
|
if typ == ADDR_TYPE_UNIX:
|
||
|
|
s.start_unix_server(*a)
|
||
|
|
elif typ == ADDR_TYPE_WS:
|
||
|
|
url = urlparse(a[0])
|
||
|
|
s.start_websocket_server(url.hostname, url.port)
|
||
|
|
else:
|
||
|
|
s.start_tcp_server(*a)
|
||
|
|
|
||
|
|
return s
|
||
|
|
|
||
|
|
|
||
|
|
def create_client(addr, username=None, password=None):
|
||
|
|
from . import client
|
||
|
|
|
||
|
|
c = client.Client(username, password)
|
||
|
|
|
||
|
|
try:
|
||
|
|
(typ, a) = parse_address(addr)
|
||
|
|
if typ == ADDR_TYPE_UNIX:
|
||
|
|
c.connect_unix(*a)
|
||
|
|
elif typ == ADDR_TYPE_WS:
|
||
|
|
c.connect_websocket(*a)
|
||
|
|
else:
|
||
|
|
c.connect_tcp(*a)
|
||
|
|
return c
|
||
|
|
except Exception as e:
|
||
|
|
c.close()
|
||
|
|
raise e
|
||
|
|
|
||
|
|
|
||
|
|
async def create_async_client(addr, username=None, password=None):
|
||
|
|
from . import client
|
||
|
|
|
||
|
|
c = client.AsyncClient(username, password)
|
||
|
|
|
||
|
|
try:
|
||
|
|
(typ, a) = parse_address(addr)
|
||
|
|
if typ == ADDR_TYPE_UNIX:
|
||
|
|
await c.connect_unix(*a)
|
||
|
|
elif typ == ADDR_TYPE_WS:
|
||
|
|
await c.connect_websocket(*a)
|
||
|
|
else:
|
||
|
|
await c.connect_tcp(*a)
|
||
|
|
|
||
|
|
return c
|
||
|
|
except Exception as e:
|
||
|
|
await c.close()
|
||
|
|
raise e
|