bashhub 3.0.2__py3-none-any.whl

bashhub/rest_client.py

@@ -0,0 +1,251 @@
+#!/usr/bin/python
+import json
+import sys
+import requests
+from requests import ConnectionError
+import os
+
+from .model import MinCommand
+from .model import StatusView
+from .model import Command
+from .model import LoginResponse
+from .model import System
+from .bashhub_globals import BH_URL, BH_AUTH
+from .version import __version__
+from requests import ConnectionError
+from requests import HTTPError
+
+# Build our user agent string
+user_agent = 'bashhub/%s' % __version__
+
+base_headers = {'User-Agent': user_agent, 'X-Bashhub-version': __version__}
+
+json_headers = dict(
+    {'content-type': 'application/json',
+     'Accept': 'application/json'}, **base_headers)
+
+
+def json_auth_headers():
+    return dict({'Authorization': 'Bearer {0}'.format(BH_AUTH())},
+                **json_headers)
+
+
+def base_auth_headers():
+    return dict({'Authorization': 'Bearer {0}'.format(BH_AUTH())},
+                **base_headers)
+
+
+def register_user(register_user):
+    url = BH_URL + "/api/v1/user"
+    try:
+        response = requests.post(url,
+                                 data=register_user.to_JSON(),
+                                 headers=json_headers)
+        response.raise_for_status()
+
+        # Return our username on a successful response
+        return register_user.username
+
+    except ConnectionError as error:
+        print("Looks like there's a connection error. Please try again later")
+    except HTTPError as error:
+        if response.status_code in (409, 422):
+            print(response.text)
+        else:
+            print(error)
+            print("Please try again...")
+    return None
+
+
+def login_user(login_form):
+    url = BH_URL + "/api/v1/login"
+    try:
+        response = requests.post(url,
+                                 data=login_form.to_JSON(),
+                                 headers=json_headers)
+
+        response.raise_for_status()
+        login_response_json = json.dumps(response.json())
+        return LoginResponse.from_JSON(login_response_json).access_token
+
+    except ConnectionError as error:
+        print("Looks like there's a connection error. Please try again later")
+        return None
+    except HTTPError as error:
+        if response.status_code in (409, 401):
+            print(response.text)
+        else:
+            print(error)
+            print("Please try again...")
+        return None
+
+
+def register_system(register_system):
+    url = BH_URL + "/api/v1/system"
+    headers = {'content-type': 'application/json'}
+    try:
+        response = requests.post(url,
+                                 data=register_system.to_JSON(),
+                                 headers=json_auth_headers())
+        response.raise_for_status()
+        return register_system.name
+
+    except ConnectionError as error:
+        print("Looks like there's a connection error. Please try again later")
+    except HTTPError as error:
+        if response.status_code == 409:
+            print(response.text)
+        else:
+            print(error)
+            print("Please try again...")
+    return None
+
+
+def get_system_information(mac):
+    url = BH_URL + '/api/v1/system'
+    payload = {'mac': mac}
+    try:
+        response = requests.get(url,
+                                params=payload,
+                                headers=json_auth_headers())
+        response.raise_for_status()
+        system_json = json.dumps(response.json())
+        return System.from_JSON(system_json)
+    except ConnectionError as error:
+        print("Looks like there's a connection error. Please try again later")
+    except HTTPError as error:
+        return None
+
+
+def get_command(uuid):
+    url = BH_URL + "/api/v1/command/{0}".format(uuid)
+    try:
+        response = requests.get(url, headers=json_auth_headers())
+        response.raise_for_status()
+        json_command = json.dumps(response.json())
+        return Command.from_JSON(json_command)
+
+    except ConnectionError as error:
+        print("Looks like there's a connection error. Please try again later")
+    except HTTPError as error:
+        print(error)
+        print("Please try again...")
+
+    return None
+
+
+def delete_command(uuid):
+    url = BH_URL + "/api/v1/command/{0}".format(uuid)
+    try:
+        response = requests.delete(url, headers=base_auth_headers())
+        response.raise_for_status()
+        return uuid
+
+    except ConnectionError as error:
+        pass
+    except HTTPError as error:
+        print(error)
+
+    return None
+
+
+def patch_system(system_patch, mac):
+
+    url = BH_URL + "/api/v1/system/{0}".format(mac)
+
+    r = None
+    try:
+        r = requests.patch(url,
+                           data=system_patch.to_JSON(),
+                           headers=json_auth_headers())
+        r.raise_for_status()
+        return r.status_code
+    except Exception as error:
+        if r is not None and r.status_code in (403, 401):
+            print("Permissions Issue. Run bashhub setup to re-login.")
+        return None
+
+
+def search(limit=None, path=None, query=None, system_name=None, unique=None, session_id=None):
+
+    payload = dict()
+
+    if limit:
+        payload["limit"] = limit
+
+    if path:
+        payload["path"] = path
+
+    if query:
+        payload["query"] = query
+
+    if system_name:
+        payload["systemName"] = system_name
+
+    if session_id:
+        payload["sessionUuid"] = session_id
+
+    payload["unique"] = str(unique).lower()
+    url = BH_URL + "/api/v1/command/search"
+
+    r = None
+    try:
+        r = requests.get(url, params=payload, headers=json_auth_headers())
+        return MinCommand.from_JSON_list(r.json())
+
+    except ConnectionError as error:
+        print("Sorry, looks like there's a connection error. Please try again later")
+    except Exception as error:
+        if r is not None:
+            if r.status_code in (403, 401):
+                print("Permissions Issue. Run bashhub setup to re-login.")
+            elif r.status_code in [400]:
+                print(
+                    "Sorry, an error occurred communicating with Bashhub. Response Code: "
+                    + str(r.status_code))
+                print(r.text)
+            else:
+                print(
+                    "Sorry, an error occurred communicating with Bashhub. Response Code: "
+                    + str(r.status_code))
+                print(error)
+    return []
+
+
+def save_command(command):
+    url = BH_URL + "/api/v1/command"
+
+    r = None
+    try:
+        r = requests.post(url,
+                          data=command.to_JSON(),
+                          headers=json_auth_headers())
+    except ConnectionError as error:
+        print("Sorry, looks like there's a connection error")
+        pass
+    except Exception as error:
+        if r is not None and r.status_code in (403, 401):
+            print("Permissions Issue. Run bashhub setup to re-login.")
+
+
+def get_status_view(process_id, start_time):
+    url = BH_URL + "/api/v1/client-view/status"
+
+    payload = {'processId': process_id, 'startTime': start_time}
+    r = None
+    try:
+        r = requests.get(url, params=payload, headers=json_auth_headers())
+        status_view_json = json.dumps(r.json())
+        return StatusView.from_JSON(status_view_json)
+    except ConnectionError as error:
+        print("Sorry, looks like there's a connection error")
+        return None
+    except Exception as error:
+        if r is not None:
+            if r.status_code in (403, 401):
+                print("Permissions Issue. Run bashhub setup to re-login.")
+            else:
+                print(
+                    "Sorry, an error occurred communicating with Bashhub. Response Code: "
+                    + str(r.status_code))
+        return None

bashhub/shell/bashhub.fish

@@ -0,0 +1,105 @@
+#
+# bashhub.fish
+# Main file that is sourced onto our path for fish.
+#
+
+#
+# Checks if an element is present in an array.
+#
+# @param The element to check if present
+# @param the array to check in
+# @return 0 if present 1 otherwise
+#
+function contains_element --argument-names element array
+    for e in $array
+        [ "$e" = "$element" ] && return 0
+    end
+
+    return 1
+end
+
+function __bh_path_add --argument-names item
+    if [ -d "$item" ] && not contains_element "$item" "$PATH"
+        set -x PATH "$item" "$PATH"
+    end
+end
+
+function __bh_interactive
+  fish -c "bh -i"
+end
+
+# Avoid duplicate inclusion
+if [ "$__bh_imported" = "defined" ]
+    __bh_path_add "$HOME/.bashhub/bin"
+else
+    set -Ux __bh_imported "defined"
+    set -Ux BH_HOME_DIRECTORY "$HOME/.bashhub/"
+
+    source "$BH_HOME_DIRECTORY/deps/fish/functions/__bh_check_bashhub_installation.fish"
+    bind \cb __bh_interactive
+end
+
+function __bh_preexec --on-event fish_preexec
+    set -g __BH_PWD "$PWD"
+    set -g __BH_SAVE_COMMAND "$argv[1]"
+end
+
+function __bh_precmd --on-event fish_prompt
+    set -x __BH_EXIT_STATUS $status
+
+    if [ -e "$BH_HOME_DIRECTORY/response.bh" ]
+        set -l cmd (head -n 1 "$BH_HOME_DIRECTORY/response.bh")
+        rm "$BH_HOME_DIRECTORY/response.bh"
+        echo $cmd
+    end
+
+    if [ -n "$BH_HOME_DIRECTORY" ]
+        set -g bashhub_dir "$BH_HOME_DIRECTORY"
+    else
+        set -g bashhub_dir "~/.bashhub"
+    end
+
+    set -x working_directory "$__BH_PWD"
+    set -x cmd "$__BH_SAVE_COMMAND"
+    set -x process_id $fish_pid
+
+    if [ -n "$__BH_SAVE_COMMAND" ]
+        set -e __BH_SAVE_COMMAND
+    else
+        return 0
+    end
+
+    if [ -e "$bashhub_dir" ]
+        fish -c '__bh_process_command "$cmd" "$working_directory" "$process_id" &' >> "$bashhub_dir"/log.txt 2>&1
+    end
+end
+
+#
+# Send our command to the server if everything
+# looks good.
+#
+function __bh_process_command --argument-names cmd dir pid
+    set -x bh_command (string trim $cmd)
+
+    # sanity check
+    if [ -z "$bh_command" ]
+        return 0
+    end
+
+    # ensure that bashhub is installed
+    if not type "bashhub" > /dev/null 2>&1
+        return 0
+    end
+
+    set -x working_directory "$dir"
+    set -x process_id "$pid"
+
+    # This is non-standard across systems. As GNU and BSD Date convert epochs
+    # differently, use python for cross-system compatibility.
+    set -l process_start_stamp (env LC_ALL=C ps -p $fish_pid -o lstart=)
+
+    set -x process_start (bashhub util parsedate "$process_start_stamp")
+    set -x exit_status "$__BH_EXIT_STATUS"
+
+    fish -c 'bashhub save "$bh_command" "$working_directory" "$process_id" "$process_start" "$exit_status" &'
+end

bashhub/shell/bashhub.sh

@@ -0,0 +1,73 @@
+#
+# bashhub.sh
+# Main file that is sourced onto our path for Bash.
+#
+
+# Avoid duplicate inclusion
+if [[ "$__bh_imported" == "defined" ]]; then
+    __bh_path_add "$HOME/.bashhub/bin"
+    return 0
+fi
+
+__bh_imported="defined"
+
+export BH_HOME_DIRECTORY="$HOME/.bashhub/"
+
+BH_DEPS_DIRECTORY=${BH_DEPS_DIRECTORY:=$BH_HOME_DIRECTORY/deps}
+
+__bh_setup_bashhub() {
+
+    # check that we're using bash and that all our
+    # dependencies are satisfied.
+    if [[ -n $BASH_VERSION ]] && \
+       [[ -f $BH_DEPS_DIRECTORY/lib-bashhub.sh ]] && \
+       [[ -f $BH_DEPS_DIRECTORY/bash-preexec.sh ]]; then
+
+        # Pull in our libs
+        source "$BH_DEPS_DIRECTORY/lib-bashhub.sh"
+        source "$BH_DEPS_DIRECTORY/bash-preexec.sh"
+
+        # Hook bashhub into preexec and precmd.
+        __bh_hook_bashhub
+
+        # Install our tab completion (requires bash 4.0+ for compopt)
+        if [[ "${BASH_VERSINFO[0]}" -ge 4 ]]; then
+            eval "$(_BASHHUB_COMPLETE=bash_source bashhub)"
+        fi
+    fi
+}
+
+__bh_hook_bashhub() {
+
+    if [ -t 1 ]; then
+        # Alias to bind Ctrl + B
+        bind '"\C-b":"\C-ubh -i\n"'
+    fi
+
+    # Hook into preexec and precmd functions
+    if ! contains_element __bh_preexec "${preexec_functions[@]}"; then
+        preexec_functions+=(__bh_preexec)
+    fi
+
+    if ! contains_element __bh_precmd "${precmd_functions[@]}"; then
+        # Order seems to matter here due to the fork at the end of __bh_precmd
+        precmd_functions+=(__bh_bash_precmd)
+        precmd_functions+=(__bh_precmd)
+    fi
+}
+
+__bh_bash_precmd() {
+    if [[ -e $BH_HOME_DIRECTORY/response.bh ]]; then
+        local command=$(head -n 1 "$BH_HOME_DIRECTORY/response.bh")
+        rm "$BH_HOME_DIRECTORY/response.bh"
+        history -s "$command"
+        # Save that we're executing this command again by calling bashhub's
+        # preexec and precmd functions
+        __bh_preexec "$command"
+        echo "$command"
+        eval "$command"
+        __bh_precmd
+     fi;
+}
+
+__bh_setup_bashhub

bashhub/shell/bashhub.zsh

@@ -0,0 +1,65 @@
+#
+# bashhub.zsh
+# Main file that is sourced onto our path for Zsh.
+#
+
+# Avoid duplicate inclusion
+if [[ "$__bh_imported" == "defined" ]]; then
+    __bh_path_add "$HOME/.bashhub/bin"
+    return 0
+fi
+
+__bh_imported="defined"
+
+export BH_HOME_DIRECTORY="$HOME/.bashhub/"
+
+BH_DEPS_DIRECTORY=${BH_DEPS_DIRECTORY:=$BH_HOME_DIRECTORY/deps}
+
+__bh_setup_bashhub() {
+
+    # check that we're using zsh and that all our
+    # dependencies are satisfied.
+    if [[ -n $ZSH_VERSION ]] && [[ -f $BH_DEPS_DIRECTORY/lib-bashhub.sh ]]; then
+
+        # Pull in our library.
+        source $BH_DEPS_DIRECTORY/lib-bashhub.sh
+
+        # Hook bashhub into preexec and precmd.
+        __bh_hook_bashhub
+
+        # Install our tab completion.
+        autoload compinit && compinit
+        eval "$(_BASHHUB_COMPLETE=zsh_source bashhub)"
+
+        # Turn on Bash style comments. Otherwise zsh tries to execute #some-comment.
+        setopt interactivecomments
+
+    fi
+}
+
+__bh_hook_bashhub() {
+
+    # Bind ctrl + b to bh -i
+    bindkey -s '^b' "bh -i\n"
+
+    # Hook into preexec and precmd functions if they're not already
+    # present there.
+    if ! contains_element __bh_preexec $preexec_functions; then
+        preexec_functions+=(__bh_preexec)
+    fi
+
+    if ! contains_element __bh_precmd  $precmd_functions; then
+        precmd_functions+=(__bh_zsh_precmd)
+        precmd_functions+=(__bh_precmd)
+    fi
+}
+
+__bh_zsh_precmd() {
+    if [[ -e $BH_HOME_DIRECTORY/response.bh ]]; then
+        local COMMAND="`head -n 1 $BH_HOME_DIRECTORY/response.bh`"
+        rm $BH_HOME_DIRECTORY/response.bh
+        print -z $COMMAND
+    fi;
+}
+
+__bh_setup_bashhub