lbry-sdk/lbrynet/extras/reflector/reupload.py

from twisted.internet import reactor, defer
from lbrynet.extras.reflector.client.client import EncryptedFileReflectorClientFactory
from lbrynet.extras.reflector.client.blob import BlobReflectorClientFactory


def _is_ip(host):
    try:
        if len(host.split(".")) == 4 and all([0 <= int(x) <= 255 for x in host.split(".")]):
            return True
        return False
    except ValueError:
        return False


@defer.inlineCallbacks
def resolve(host):
    if _is_ip(host):
        ip = host
    else:
        ip = yield reactor.resolve(host)
    defer.returnValue(ip)


@defer.inlineCallbacks
def _reflect_stream(blob_manager, stream_hash, sd_hash, reflector_server):
    reflector_address, reflector_port = reflector_server[0], reflector_server[1]
    factory = EncryptedFileReflectorClientFactory(blob_manager, stream_hash, sd_hash)
    ip = yield resolve(reflector_address)
    yield reactor.connectTCP(ip, reflector_port, factory)
    result = yield factory.finished_deferred
    defer.returnValue(result)


def _reflect_file(lbry_file, reflector_server):
    return _reflect_stream(lbry_file.blob_manager, lbry_file.stream_hash, lbry_file.sd_hash, reflector_server)


@defer.inlineCallbacks
def _reflect_blobs(blob_manager, blob_hashes, reflector_server):
    reflector_address, reflector_port = reflector_server[0], reflector_server[1]
    factory = BlobReflectorClientFactory(blob_manager, blob_hashes)
    ip = yield resolve(reflector_address)
    yield reactor.connectTCP(ip, reflector_port, factory)
    result = yield factory.finished_deferred
    return result


def reflect_file(lbry_file, reflector_server):
    if len(reflector_server.split(":")) == 2:
        host, port = tuple(reflector_server.split(":"))
        reflector_server = host, int(port)
    else:
        reflector_server = reflector_server, 5566
    return _reflect_file(lbry_file, reflector_server)


def reflect_blob_hashes(blob_hashes, blob_manager, reflector_server):
    if len(reflector_server.split(":")) == 2:
        host, port = tuple(reflector_server.split(":"))
        reflector_server = host, int(port)
    else:
        reflector_server = reflector_server, 5566
    return _reflect_blobs(blob_manager, blob_hashes, reflector_server)