Onionr/src/onionrutils/localcommand.py

"""Onionr - Private P2P Communication.

send a command to the local API server
"""
import urllib
import time

import requests
import deadsimplekv

import logger
import config

from . import getclientapiserver
import filepaths
"""
    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/>.
"""
config.reload()

cache = deadsimplekv.DeadSimpleKV(filepaths.cached_storage,
                                  refresh_seconds=1000)


def get_hostname():
    hostname = ''
    waited = 0
    max_wait = 3
    while True:
        if cache.get('client_api') is None:
            try:
                hostname = getclientapiserver.get_client_API_server()
            except FileNotFoundError:
                hostname = False
            else:
                cache.put('hostname', hostname)
                cache.flush()
        else:
            hostname = cache.get('hostname')
        if hostname == '' or hostname is None:
            time.sleep(1)
            if waited == max_wait:
                return False
        else:
            return hostname


def local_command(command, data='', silent=True, post=False,
                  post_data={}, max_wait=20,
                  is_json=False
                  ):
    """Send a command to the local http API server, securely.
    Intended for local clients, DO NOT USE for remote peers."""
    hostname = get_hostname()
    # if the api host cannot be reached, return False
    if not hostname:
        return False

    if data != '':
        data = '&data=' + urllib.parse.quote_plus(data)
    payload = 'http://%s/%s%s' % (hostname, command, data)
    if not config.get('client.webpassword'):
        config.reload()

    try:
        if post:
            if is_json:
                ret_data = requests.post(
                    payload,
                    json=post_data,
                    headers={'token': config.get('client.webpassword'),
                             'Connection': 'close'},
                    timeout=(max_wait, max_wait)).text
            else:
                ret_data = requests.post(
                    payload,
                    data=post_data,
                    headers={'token': config.get('client.webpassword'),
                             'Connection': 'close'},
                    timeout=(max_wait, max_wait)).text
        else:
            ret_data = requests.get(payload,
                                    headers={'token':
                                             config.get('client.webpassword'),
                                             'Connection': 'close'},
                                    timeout=(max_wait, max_wait)).text
    except Exception as error:
        if not silent:
            logger.error('Failed to make local request (command: %s):%s' %
                         (command, error), terminal=True)
        ret_data = False
    return ret_data