lbry-sdk/lbrynet/daemon/Publisher.py

import logging
import mimetypes
import os

from twisted.internet import defer

from lbrynet.core import file_utils
from lbrynet.file_manager.EncryptedFileCreator import create_lbry_file
from lbrynet.wallet.account import get_certificate_lookup

log = logging.getLogger(__name__)


class Publisher(object):
    def __init__(self, blob_manager, payment_rate_manager, storage, lbry_file_manager, wallet, certificate):
        self.blob_manager = blob_manager
        self.payment_rate_manager = payment_rate_manager
        self.storage = storage
        self.lbry_file_manager = lbry_file_manager
        self.wallet = wallet
        self.certificate = certificate
        self.lbry_file = None

    @defer.inlineCallbacks
    def create_and_publish_stream(self, name, bid, claim_dict, file_path, holding_address=None):
        """Create lbry file and make claim"""
        log.info('Starting publish for %s', name)
        if not os.path.isfile(file_path):
            raise Exception("File {} not found".format(file_path))
        if os.path.getsize(file_path) == 0:
            raise Exception("Cannot publish empty file {}".format(file_path))

        file_name = os.path.basename(file_path)
        with file_utils.get_read_handle(file_path) as read_handle:
            self.lbry_file = yield create_lbry_file(
                self.blob_manager, self.storage, self.payment_rate_manager, self.lbry_file_manager, file_name,
                read_handle
            )

        if 'source' not in claim_dict['stream']:
            claim_dict['stream']['source'] = {}
        claim_dict['stream']['source']['source'] = self.lbry_file.sd_hash
        claim_dict['stream']['source']['sourceType'] = 'lbry_sd_hash'
        claim_dict['stream']['source']['contentType'] = get_content_type(file_path)
        claim_dict['stream']['source']['version'] = "_0_0_1"  # need current version here
        tx = yield self.wallet.claim_name(
            name, bid, claim_dict, self.certificate, holding_address
        )

        # check if we have a file already for this claim (if this is a publish update with a new stream)
        old_stream_hashes = yield self.storage.get_old_stream_hashes_for_claim_id(
            tx.get_claim_id(0), self.lbry_file.stream_hash.decode()
        )
        if old_stream_hashes:
            for lbry_file in filter(lambda l: l.stream_hash in old_stream_hashes,
                                    list(self.lbry_file_manager.lbry_files)):
                yield self.lbry_file_manager.delete_lbry_file(lbry_file, delete_file=False)
                log.info("Removed old stream for claim update: %s", lbry_file.stream_hash)

        yield self.storage.save_content_claim(
            self.lbry_file.stream_hash.decode(), get_certificate_lookup(tx, 0)
        )
        defer.returnValue(tx)

    @defer.inlineCallbacks
    def publish_stream(self, name, bid, claim_dict, stream_hash, holding_address=None):
        """Make a claim without creating a lbry file"""
        tx = yield self.wallet.claim_name(
            name, bid, claim_dict, self.certificate, holding_address
        )
        if stream_hash:  # the stream_hash returned from the db will be None if this isn't a stream we have
            yield self.storage.save_content_claim(
                stream_hash.decode(), get_certificate_lookup(tx, 0)
            )
            self.lbry_file = [f for f in self.lbry_file_manager.lbry_files if f.stream_hash == stream_hash][0]
        defer.returnValue(tx)


def get_content_type(filename):
    return mimetypes.guess_type(filename)[0] or 'application/octet-stream'