Onionr/onionr/onionrcommands/filecommands.py

'''
    Onionr - Private P2P Communication

    This file handles the commands for adding and getting files from the Onionr network
'''
'''
    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <https://www.gnu.org/licenses/>.
'''

import base64, sys, os
import logger
from onionrblockapi import Block
def add_file(o_inst, singleBlock=False, blockType='bin'):
    '''
        Adds a file to the onionr network
    '''

    if len(sys.argv) >= 3:
        filename = sys.argv[2]
        contents = None

        if not os.path.exists(filename):
            logger.error('That file does not exist. Improper path (specify full path)?')
            return
        logger.info('Adding file... this might take a long time.')
        try:
            with open(filename, 'rb') as singleFile:
                blockhash = o_inst.onionrCore.insertBlock(base64.b64encode(singleFile.read()), header=blockType)
            if len(blockhash) > 0:
                logger.info('File %s saved in block %s' % (filename, blockhash))
        except:
            logger.error('Failed to save file in block.', timestamp = False)
    else:
        logger.error('%s add-file <filename>' % sys.argv[0], timestamp = False)

def getFile(o_inst):
    '''
        Get a file from onionr blocks
    '''
    try:
        fileName = sys.argv[2]
        bHash = sys.argv[3]
    except IndexError:
        logger.error("Syntax %s %s" % (sys.argv[0], '/path/to/filename <blockhash>'))
    else:
        logger.info(fileName)

        contents = None
        if os.path.exists(fileName):
            logger.error("File already exists")
            return
        if not o_inst.onionrUtils.validateHash(bHash):
            logger.error('Block hash is invalid')
            return

        with open(fileName, 'wb') as myFile:
            myFile.write(base64.b64decode(Block(bHash, core=o_inst.onionrCore).bcontent))
    return