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

import random

from twisted.internet import reactor, defer
from lbrynet import conf
from . import ClientFactory, BlobClientFactory


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 = ClientFactory(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 = BlobClientFactory(blob_manager, blob_hashes)
    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=None):
    if 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
    else:
        reflector_server = random.choice(conf.settings['reflector_servers'])
    return _reflect_file(lbry_file, reflector_server)


@defer.inlineCallbacks
def reflect_stream(blob_manager, stream_hash, reflector_server=None):
    if 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
    else:
        reflector_server = random.choice(conf.settings['reflector_servers'])
    sd_hash = yield blob_manager.storage.get_sd_blob_hash_for_stream(stream_hash)
    result = yield _reflect_stream(blob_manager, stream_hash, sd_hash, reflector_server)
    defer.returnValue(result)


def reflect_blob_hashes(blob_hashes, blob_manager, reflector_server=None):
    if 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
    else:
        reflector_server = random.choice(conf.settings['reflector_servers'])
    return _reflect_blobs(blob_manager, blob_hashes, reflector_server)