forked from LBRYCommunity/lbry-sdk
67 lines
No EOL
2.3 KiB
Python
67 lines
No EOL
2.3 KiB
Python
import logging
|
|
import os
|
|
import sys
|
|
|
|
from twisted.web import server, guard
|
|
from twisted.internet import defer, reactor, error
|
|
from twisted.cred import portal
|
|
|
|
from lbrynet.conf import settings
|
|
from lbrynet.lbrynet_daemon.Daemon import Daemon
|
|
from lbrynet.lbrynet_daemon.Resources import LBRYindex, HostedEncryptedFile, EncryptedFileUpload
|
|
from lbrynet.lbrynet_daemon.auth.auth import PasswordChecker, HttpPasswordRealm
|
|
from lbrynet.lbrynet_daemon.auth.util import initialize_api_key_file
|
|
from lbrynet.lbrynet_daemon.DaemonRequest import DaemonRequest
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class DaemonServer(object):
|
|
def __init__(self, analytics_manager=None):
|
|
self.root = None
|
|
self.analytics_manager = analytics_manager
|
|
|
|
def _setup_server(self, use_auth):
|
|
ui_path = os.path.join(settings.ensure_data_dir(), "lbry-ui", "active")
|
|
self.root = LBRYindex(ui_path)
|
|
self._api = Daemon(self.root, self.analytics_manager)
|
|
self.root.putChild("view", HostedEncryptedFile(self._api))
|
|
self.root.putChild("upload", EncryptedFileUpload(self._api))
|
|
self.root.putChild(settings.API_ADDRESS, self._api)
|
|
|
|
lbrynet_server = server.Site(get_site_base(use_auth, self.root))
|
|
lbrynet_server.requestFactory = DaemonRequest
|
|
|
|
try:
|
|
reactor.listenTCP(settings.api_port, lbrynet_server, interface=settings.API_INTERFACE)
|
|
except error.CannotListenError:
|
|
log.info('Daemon already running, exiting app')
|
|
sys.exit(1)
|
|
|
|
return defer.succeed(True)
|
|
|
|
def start(self, use_auth):
|
|
d = self._setup_server(use_auth)
|
|
d.addCallback(lambda _: self._api.setup())
|
|
return d
|
|
|
|
|
|
def get_site_base(use_auth, root):
|
|
if use_auth:
|
|
log.info("Using authenticated API")
|
|
return create_auth_session(root)
|
|
else:
|
|
log.info("Using non-authenticated API")
|
|
return server.Site(root)
|
|
|
|
|
|
def create_auth_session(root):
|
|
pw_path = os.path.join(settings.data_dir, ".api_keys")
|
|
initialize_api_key_file(pw_path)
|
|
checker = PasswordChecker.load_file(pw_path)
|
|
realm = HttpPasswordRealm(root)
|
|
portal_to_realm = portal.Portal(realm, [checker, ])
|
|
factory = guard.BasicCredentialFactory('Login to lbrynet api')
|
|
_lbrynet_server = guard.HTTPAuthSessionWrapper(portal_to_realm, [factory, ])
|
|
return _lbrynet_server |