107 lines
3.7 KiB
Python
Executable File
107 lines
3.7 KiB
Python
Executable File
"""Onionr - Private P2P Communication.
|
|
|
|
This file handles the commands for adding
|
|
and getting files from the Onionr network
|
|
"""
|
|
import sys
|
|
import os
|
|
import logger
|
|
from onionrblocks.onionrblockapi import Block
|
|
import onionrexceptions
|
|
from onionrutils import stringvalidators
|
|
from etc import onionrvalues
|
|
from onionrblocks import insert
|
|
"""
|
|
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/>.
|
|
"""
|
|
|
|
_ORIG_DIR = onionrvalues.ORIG_RUN_DIR_ENV_VAR
|
|
|
|
|
|
def _get_dir(path: str) -> str:
|
|
if not os.getenv(_ORIG_DIR) is None:
|
|
return os.getenv(_ORIG_DIR) + '/' + path # type: ignore
|
|
else:
|
|
return path
|
|
|
|
|
|
def add_html(singleBlock=True, blockType='html'):
|
|
"""Create one-off web page from HTML file, no ext resources."""
|
|
add_file(blockType=blockType)
|
|
|
|
|
|
add_html.onionr_help = "Adds an HTML file into Onionr. Does " # type: ignore
|
|
add_html.onionr_help += "not include dependant resources" # type: ignore
|
|
|
|
|
|
def add_file(blockType='bin'):
|
|
"""Add a file to the onionr network."""
|
|
if len(sys.argv) >= 3:
|
|
filename = sys.argv[2]
|
|
|
|
if not os.path.exists(_get_dir(filename)):
|
|
logger.error(
|
|
'That file does not exist. Improper path (specify full path)?',
|
|
terminal=True)
|
|
return
|
|
logger.info('Adding file, this might take a long time.',
|
|
terminal=True)
|
|
try:
|
|
with open(_get_dir(filename), 'rb') as single_file:
|
|
blockhash = insert(single_file.read(), header=blockType)
|
|
if len(blockhash) > 0:
|
|
logger.info('File %s saved in block %s' %
|
|
(filename, blockhash), terminal=True)
|
|
except Exception as err: # pylint: disable=W0703
|
|
logger.error('Failed to save file in block ' +
|
|
str(err), timestamp=False, terminal=True)
|
|
else:
|
|
logger.error('%s add-file <filename>' %
|
|
sys.argv[0], timestamp=False, terminal=True)
|
|
|
|
|
|
add_file.onionr_help = "<file path> Add a file into " # type: ignore
|
|
add_file.onionr_help += "the Onionr network" # type: ignore
|
|
|
|
|
|
def get_file():
|
|
"""Get a file from onionr blocks."""
|
|
try:
|
|
file_name = _get_dir(sys.argv[2])
|
|
bHash = sys.argv[3]
|
|
except IndexError:
|
|
logger.error("Syntax %s %s" % (
|
|
sys.argv[0], '/path/to/filename <blockhash>'), terminal=True)
|
|
else:
|
|
logger.info(file_name, terminal=True)
|
|
|
|
if os.path.exists(file_name):
|
|
logger.error("File already exists", terminal=True)
|
|
return
|
|
if not stringvalidators.validate_hash(bHash):
|
|
logger.error('Block hash is invalid', terminal=True)
|
|
return
|
|
|
|
try:
|
|
with open(file_name, 'wb') as my_file:
|
|
my_file.write(Block(bHash).bcontent)
|
|
except onionrexceptions.NoDataAvailable:
|
|
logger.error(
|
|
'That block is not available. Trying again later may work.',
|
|
terminal=True)
|
|
|
|
|
|
get_file.onionr_help = "<file path> <block hash>: Download " # type: ignore
|
|
get_file.onionr_help += "a file from the onionr network." # type: ignore
|