Compare commits
31 Commits
DEV
...
2025.0519.
Author | SHA1 | Date | |
---|---|---|---|
d8236a58df | |||
625de98890 | |||
7c1b51a93c | |||
1be7af571f | |||
1439ec2f79 | |||
6c3c35bf89 | |||
548ffea6f9 | |||
434a2bc6da | |||
cf8738aee9 | |||
d80820db15 | |||
fb6974b51a | |||
263edd9b50 | |||
e45afe460b | |||
630a9fd19a | |||
f89d90c12b | |||
5883c62c54 | |||
b278e81533 | |||
6c99b429b9 | |||
2b446f80a3 | |||
314a5fe96a | |||
828171c977 | |||
668cef5a05 | |||
e7c6d38273 | |||
d0152c3ec7 | |||
ebb101e381 | |||
dc2f694ebe | |||
038d08e638 | |||
27f86e95e7 | |||
891f0d023f | |||
91f706ffcd | |||
0e1ac9ddd8 |
1
.gitignore
vendored
1
.gitignore
vendored
@ -7,6 +7,7 @@ out/
|
|||||||
bin/
|
bin/
|
||||||
lib/
|
lib/
|
||||||
output/
|
output/
|
||||||
|
autogen/
|
||||||
|
|
||||||
# Compiled Object files
|
# Compiled Object files
|
||||||
*.o
|
*.o
|
||||||
|
@ -8,6 +8,8 @@ A system management tool for server operations, written in C++.
|
|||||||
curl -fsSL https://gitea.jde.nz/public/dropshell/releases/download/latest/install.sh | sudo bash
|
curl -fsSL https://gitea.jde.nz/public/dropshell/releases/download/latest/install.sh | sudo bash
|
||||||
```
|
```
|
||||||
|
|
||||||
|
This installs as dropshell, with a symlink ds if the ds command does not already exist.
|
||||||
|
|
||||||
## Installation of Agent
|
## Installation of Agent
|
||||||
|
|
||||||
Install the Agent on each server you wish to manage. Supports amd64 (x86_64) and arm64 (aarch64) architectures.
|
Install the Agent on each server you wish to manage. Supports amd64 (x86_64) and arm64 (aarch64) architectures.
|
||||||
@ -24,3 +26,7 @@ Manual steps:
|
|||||||
1. Test ssh'ing into the server.
|
1. Test ssh'ing into the server.
|
||||||
|
|
||||||
|
|
||||||
|
## Install Services
|
||||||
|
|
||||||
|
Set up a server and install a service:
|
||||||
|
1. `ds create-server SERVERNAME`
|
||||||
|
@ -48,10 +48,23 @@ add_custom_target(run_createagent ALL
|
|||||||
add_executable(dropshell ${SOURCES})
|
add_executable(dropshell ${SOURCES})
|
||||||
add_dependencies(dropshell run_createagent)
|
add_dependencies(dropshell run_createagent)
|
||||||
|
|
||||||
|
# Mark the generated files as GENERATED so CMake knows they'll be created during build
|
||||||
|
set_source_files_properties(
|
||||||
|
${CMAKE_CURRENT_SOURCE_DIR}/src/autogen/_agent-remote.cpp
|
||||||
|
${CMAKE_CURRENT_SOURCE_DIR}/src/autogen/_agent-local.cpp
|
||||||
|
PROPERTIES GENERATED TRUE
|
||||||
|
)
|
||||||
|
|
||||||
|
# Explicitly add the generated agent files, as they might not be in the source directory when globbed at the start.
|
||||||
|
target_sources(dropshell PRIVATE
|
||||||
|
${CMAKE_CURRENT_SOURCE_DIR}/src/autogen/_agent-remote.cpp
|
||||||
|
${CMAKE_CURRENT_SOURCE_DIR}/src/autogen/_agent-local.cpp
|
||||||
|
)
|
||||||
|
|
||||||
# Set include directories
|
# Set include directories
|
||||||
# build dir goes first so that we can use the generated version.hpp
|
# build dir goes first so that we can use the generated version.hpp
|
||||||
target_include_directories(dropshell PRIVATE
|
target_include_directories(dropshell PRIVATE
|
||||||
$<BUILD_INTERFACE:${CMAKE_CURRENT_BINARY_DIR}/src>
|
$<BUILD_INTERFACE:${CMAKE_CURRENT_BINARY_DIR}/src/autogen>
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/src
|
${CMAKE_CURRENT_SOURCE_DIR}/src
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/src/utils
|
${CMAKE_CURRENT_SOURCE_DIR}/src/utils
|
||||||
${CMAKE_CURRENT_SOURCE_DIR}/src/contrib
|
${CMAKE_CURRENT_SOURCE_DIR}/src/contrib
|
||||||
|
83
source/agent-local/agent-install.sh
Executable file
83
source/agent-local/agent-install.sh
Executable file
@ -0,0 +1,83 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
SCRIPT_DIR=$(dirname "$0")
|
||||||
|
|
||||||
|
echo "Installing dropshell host agent on this computer..."
|
||||||
|
|
||||||
|
|
||||||
|
# Prints an error message in red and exits with status code 1.
|
||||||
|
_die() {
|
||||||
|
echo -e "Error: $1"
|
||||||
|
exit 1
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# Checks if listed environment variables are set; calls _die() if any are missing.
|
||||||
|
_check_required_env_vars() {
|
||||||
|
local required_vars=("$@")
|
||||||
|
for var in "${required_vars[@]}"; do
|
||||||
|
if [ -z "${!var}" ]; then
|
||||||
|
_die "Required environment variable $var is not set"
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
# Checks if Docker is installed, running, and user has permission. Returns 1 on failure.
|
||||||
|
_check_docker_installed() {
|
||||||
|
if ! command -v docker &> /dev/null; then
|
||||||
|
echo "Docker is not installed"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# check if docker daemon is running
|
||||||
|
if ! docker info &> /dev/null; then
|
||||||
|
echo "Docker daemon is not running"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# check if user has permission to run docker
|
||||||
|
if ! docker run --rm hello-world &> /dev/null; then
|
||||||
|
echo "User does not have permission to run docker"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
function install_bb64() {
|
||||||
|
# check curl installed
|
||||||
|
if ! command -v curl &> /dev/null; then
|
||||||
|
_die "Curl is not installed. Curl is required for agent installation."
|
||||||
|
fi
|
||||||
|
|
||||||
|
curl -fsSL "https://gitea.jde.nz/public/bb64/releases/download/latest/install.sh" | bash -s -- "$AGENT_PATH" "$(id -u $USER):$(id -g $USER)"
|
||||||
|
|
||||||
|
# test result code from curl
|
||||||
|
if [ $? -ne 0 ]; then
|
||||||
|
_die "Failed to install bb64. Curl returned non-zero exit code."
|
||||||
|
fi
|
||||||
|
|
||||||
|
# test if bb64 is installed
|
||||||
|
"$AGENT_PATH/bb64" -v
|
||||||
|
if [ $? -ne 0 ]; then
|
||||||
|
_die "bb64 did not install correctly."
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "bb64 installed successfully."
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
#-------------------------------------------------------------------------
|
||||||
|
|
||||||
|
set -a
|
||||||
|
AGENT_PATH="$SCRIPT_DIR"
|
||||||
|
set +a
|
||||||
|
|
||||||
|
_check_required_env_vars "AGENT_PATH"
|
||||||
|
echo "Installing host agent into $AGENT_PATH"
|
||||||
|
|
||||||
|
_check_docker_installed || _die "Docker is required."
|
||||||
|
|
||||||
|
install_bb64
|
||||||
|
|
@ -73,6 +73,7 @@ function run_command() {
|
|||||||
# update the main variables.
|
# update the main variables.
|
||||||
CONFIG_PATH="${service_path}/config"
|
CONFIG_PATH="${service_path}/config"
|
||||||
SERVICE="${SERVICE_NAME}"
|
SERVICE="${SERVICE_NAME}"
|
||||||
|
DOCKER_CLI_HINTS=false
|
||||||
|
|
||||||
set +a
|
set +a
|
||||||
|
|
||||||
@ -99,6 +100,10 @@ function command_exists() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
if [ ! -d "${SERVICES_PATH}" ]; then
|
||||||
|
echo "Services path does not exist: ${SERVICES_PATH}"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
# Get all service names
|
# Get all service names
|
||||||
SERVICE_NAMES=$(ls "${SERVICES_PATH}")
|
SERVICE_NAMES=$(ls "${SERVICES_PATH}")
|
87
source/agent-remote/agent-install.sh
Executable file
87
source/agent-remote/agent-install.sh
Executable file
@ -0,0 +1,87 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# This script is used to install the dropshell agent on a remote server.
|
||||||
|
|
||||||
|
SCRIPT_DIR=$(dirname "$0")
|
||||||
|
|
||||||
|
set -a
|
||||||
|
|
||||||
|
AGENT_PATH="$SCRIPT_DIR"
|
||||||
|
|
||||||
|
set +a
|
||||||
|
|
||||||
|
|
||||||
|
if [ -f "$SCRIPT_DIR/common.sh" ]; then
|
||||||
|
source "$SCRIPT_DIR/common.sh"
|
||||||
|
else
|
||||||
|
echo "Error: common.sh not found in $SCRIPT_DIR"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
_check_required_env_vars "AGENT_PATH"
|
||||||
|
|
||||||
|
function install_bb64() {
|
||||||
|
curl -fsSL "https://gitea.jde.nz/public/bb64/releases/download/latest/install.sh" | bash -s -- "$AGENT_PATH" "$(id -u $USER):$(id -g $USER)"
|
||||||
|
|
||||||
|
# test result code from curl
|
||||||
|
if [ $? -ne 0 ]; then
|
||||||
|
_die "Failed to install bb64. Curl returned non-zero exit code."
|
||||||
|
fi
|
||||||
|
|
||||||
|
# test if bb64 is installed
|
||||||
|
VER=$("$AGENT_PATH/bb64" -v)
|
||||||
|
if [ $? -ne 0 ]; then
|
||||||
|
_die "bb64 did not install correctly."
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "bb64 v$VER installed."
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
#-------------------------------------------------------------------------
|
||||||
|
|
||||||
|
# Check pre-requisites
|
||||||
|
|
||||||
|
# Check if curl is installed
|
||||||
|
if ! command -v curl &> /dev/null; then
|
||||||
|
_die "Curl is not installed. Curl is required for agent installation."
|
||||||
|
fi
|
||||||
|
|
||||||
|
# check docker installation
|
||||||
|
if ! command -v docker &> /dev/null; then
|
||||||
|
echo "Docker is not installed. Docker is required for agent installation."
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
|
||||||
|
#-------------------------------------------------------------------------
|
||||||
|
|
||||||
|
echo "Installing dropshell agent..."
|
||||||
|
|
||||||
|
install_bb64
|
||||||
|
|
||||||
|
#-------------------------------------------------------------------------
|
||||||
|
|
||||||
|
# confirm we're in a good state.
|
||||||
|
|
||||||
|
required_files=(
|
||||||
|
"$AGENT_PATH/bb64"
|
||||||
|
"$AGENT_PATH/_allservicesstatus.sh"
|
||||||
|
"$AGENT_PATH/common.sh"
|
||||||
|
"$AGENT_PATH/datacommands.sh"
|
||||||
|
)
|
||||||
|
|
||||||
|
# check if all files exist
|
||||||
|
for file in "${required_files[@]}"; do
|
||||||
|
if [ ! -f "$file" ]; then
|
||||||
|
_die "$file does not exist. Please check the installation."
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
#-------------------------------------------------------------------------
|
||||||
|
|
||||||
|
echo "Completed dropshell agent installation."
|
||||||
|
|
||||||
|
#-------------------------------------------------------------------------
|
||||||
|
|
||||||
|
exit 0
|
@ -28,20 +28,10 @@
|
|||||||
|
|
||||||
# Prints an error message in red and exits with status code 1.
|
# Prints an error message in red and exits with status code 1.
|
||||||
_die() {
|
_die() {
|
||||||
echo -e "\033[91mError: $1\033[0m"
|
echo -e "Error: $1"
|
||||||
exit 1
|
exit 1
|
||||||
}
|
}
|
||||||
|
|
||||||
# Switches terminal output color to grey.
|
|
||||||
_grey_start() {
|
|
||||||
echo -e -n "\033[90m"
|
|
||||||
}
|
|
||||||
|
|
||||||
# Resets terminal output color from grey.
|
|
||||||
_grey_end() {
|
|
||||||
echo -e -n "\033[0m"
|
|
||||||
}
|
|
||||||
|
|
||||||
# Creates/starts a container, verifying it runs.
|
# Creates/starts a container, verifying it runs.
|
||||||
_create_and_start_container() {
|
_create_and_start_container() {
|
||||||
if [ -z "$1" ] || [ -z "$2" ]; then
|
if [ -z "$1" ] || [ -z "$2" ]; then
|
||||||
@ -55,9 +45,7 @@ _create_and_start_container() {
|
|||||||
_is_container_running $container_name && return 0
|
_is_container_running $container_name && return 0
|
||||||
_start_container $container_name
|
_start_container $container_name
|
||||||
else
|
else
|
||||||
_grey_start
|
|
||||||
$run_cmd
|
$run_cmd
|
||||||
_grey_end
|
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if ! _is_container_running $container_name; then
|
if ! _is_container_running $container_name; then
|
@ -47,7 +47,7 @@ _autocommandrun_path() {
|
|||||||
local path_parent=$(dirname ${path})
|
local path_parent=$(dirname ${path})
|
||||||
local path_child=$(basename ${path})
|
local path_child=$(basename ${path})
|
||||||
if [ -d "${path_parent}/${path_child}" ]; then
|
if [ -d "${path_parent}/${path_child}" ]; then
|
||||||
docker run --rm -v ${path_parent}:/volume debian bash -c "rm -rf /volume/${path_child}" || echo "Failed to nuke path ${path}"
|
docker run --rm -v ${path_parent}:/volume debian bash -c "rm -rfv /volume/${path_child}" || echo "Failed to nuke path ${path}"
|
||||||
else
|
else
|
||||||
echo "Path ${path} does not exist - nothing to nuke"
|
echo "Path ${path} does not exist - nothing to nuke"
|
||||||
fi
|
fi
|
||||||
@ -61,8 +61,14 @@ _autocommandrun_path() {
|
|||||||
fi
|
fi
|
||||||
;;
|
;;
|
||||||
restore)
|
restore)
|
||||||
echo "Restoring path ${path}"
|
if [ ! -f "${backup_folder}/backup.tgz" ]; then
|
||||||
|
echo "Backup file ${backup_folder}/backup.tgz does not exist - nothing to restore"
|
||||||
|
else
|
||||||
|
echo "Clearing existing data in path ${path}"
|
||||||
|
docker run --rm -v ${path}:/path debian bash -c "rm -rfv /path/{*,.*}"
|
||||||
|
echo "Restoring path ${path} from backup file ${backup_folder}/backup.tgz"
|
||||||
tar -xzvf ${backup_folder}/backup.tgz -C ${path} --strip-components=1
|
tar -xzvf ${backup_folder}/backup.tgz -C ${path} --strip-components=1
|
||||||
|
fi
|
||||||
;;
|
;;
|
||||||
esac
|
esac
|
||||||
}
|
}
|
||||||
@ -91,7 +97,8 @@ _autocommandrun_file() {
|
|||||||
restore)
|
restore)
|
||||||
echo "Restoring file ${filepath}"
|
echo "Restoring file ${filepath}"
|
||||||
local file_name=$(basename ${filepath})
|
local file_name=$(basename ${filepath})
|
||||||
cp ${backup_folder}/${file_name} ${filepath}
|
rm -f ${filepath} || die "Unable to remove existing file ${filepath}, restore failed."
|
||||||
|
cp ${backup_folder}/${file_name} ${filepath} || die "Unable to copy file ${backup_folder}/${file_name} to ${filepath}, restore failed."
|
||||||
;;
|
;;
|
||||||
esac
|
esac
|
||||||
}
|
}
|
@ -142,8 +142,12 @@ if [ -n "$SUDO_USER" ] && [ "$SUDO_USER" != "root" ]; then
|
|||||||
fi
|
fi
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# check if dehydrate command is installed
|
||||||
|
if ! command -v dehydrate &> /dev/null; then
|
||||||
|
curl -fsSL https://gitea.jde.nz/public/dehydrate/releases/download/latest/install.sh | bash
|
||||||
|
fi
|
||||||
|
|
||||||
|
dehydrate -u
|
||||||
|
|
||||||
|
|
||||||
print_status "All dependencies installed successfully!"
|
print_status "All dependencies installed successfully!"
|
||||||
|
@ -14,6 +14,7 @@ if ! command -v dehydrate &> /dev/null; then
|
|||||||
curl -fsSL https://gitea.jde.nz/public/dehydrate/releases/download/latest/install.sh | bash
|
curl -fsSL https://gitea.jde.nz/public/dehydrate/releases/download/latest/install.sh | bash
|
||||||
fi
|
fi
|
||||||
|
|
||||||
SCRIPT_DIR=$(dirname "$0")
|
mkdir -p "${SCRIPT_DIR}/src/autogen"
|
||||||
|
|
||||||
dehydrate "${SCRIPT_DIR}/agent" "${SCRIPT_DIR}/src/autogen"
|
dehydrate "${SCRIPT_DIR}/agent-remote" "${SCRIPT_DIR}/src/autogen"
|
||||||
|
dehydrate "${SCRIPT_DIR}/agent-local" "${SCRIPT_DIR}/src/autogen"
|
||||||
|
@ -6,7 +6,9 @@
|
|||||||
|
|
||||||
set -e
|
set -e
|
||||||
|
|
||||||
rm -f build_amd64/dropshell build_arm64/dropshell build/dropshell.amd64 build/dropshell.arm64
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
|
||||||
|
rm -f $SCRIPT_DIR/build_amd64/dropshell $SCRIPT_DIR/build_arm64/dropshell $SCRIPT_DIR/output/dropshell.amd64 $SCRIPT_DIR/output/dropshell.arm64
|
||||||
|
|
||||||
# Determine number of CPU cores for parallel build
|
# Determine number of CPU cores for parallel build
|
||||||
if command -v nproc >/dev/null 2>&1; then
|
if command -v nproc >/dev/null 2>&1; then
|
||||||
@ -15,6 +17,9 @@ else
|
|||||||
JOBS=4 # fallback default
|
JOBS=4 # fallback default
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
PREV_PWD=$PWD
|
||||||
|
cd $SCRIPT_DIR
|
||||||
|
|
||||||
# Build for amd64 (musl)
|
# Build for amd64 (musl)
|
||||||
echo "Building for amd64 (musl)..."
|
echo "Building for amd64 (musl)..."
|
||||||
cmake -B build_amd64 -DCMAKE_BUILD_TYPE=Release \
|
cmake -B build_amd64 -DCMAKE_BUILD_TYPE=Release \
|
||||||
@ -23,8 +28,8 @@ cmake -B build_amd64 -DCMAKE_BUILD_TYPE=Release \
|
|||||||
-DCMAKE_EXE_LINKER_FLAGS="-static" \
|
-DCMAKE_EXE_LINKER_FLAGS="-static" \
|
||||||
-DCMAKE_CXX_FLAGS="-march=x86-64" .
|
-DCMAKE_CXX_FLAGS="-march=x86-64" .
|
||||||
cmake --build build_amd64 --target dropshell --config Release -j"$JOBS"
|
cmake --build build_amd64 --target dropshell --config Release -j"$JOBS"
|
||||||
mkdir -p build
|
mkdir -p output
|
||||||
cp build_amd64/dropshell build/dropshell.amd64
|
cp build_amd64/dropshell output/dropshell.amd64
|
||||||
|
|
||||||
# Build for arm64 (musl)
|
# Build for arm64 (musl)
|
||||||
echo "Building for arm64 (musl)..."
|
echo "Building for arm64 (musl)..."
|
||||||
@ -35,18 +40,20 @@ cmake -B build_arm64 -DCMAKE_BUILD_TYPE=Release \
|
|||||||
-DCMAKE_CXX_FLAGS="-march=armv8-a" \
|
-DCMAKE_CXX_FLAGS="-march=armv8-a" \
|
||||||
-DCMAKE_SYSTEM_PROCESSOR=aarch64 .
|
-DCMAKE_SYSTEM_PROCESSOR=aarch64 .
|
||||||
cmake --build build_arm64 --target dropshell --config Release -j"$JOBS"
|
cmake --build build_arm64 --target dropshell --config Release -j"$JOBS"
|
||||||
mkdir -p build
|
mkdir -p output
|
||||||
cp build_arm64/dropshell build/dropshell.arm64
|
cp build_arm64/dropshell output/dropshell.arm64
|
||||||
|
|
||||||
if [ ! -f build/dropshell.amd64 ]; then
|
if [ ! -f output/dropshell.amd64 ]; then
|
||||||
echo "build/dropshell.amd64 not found!" >&2
|
echo "output/dropshell.amd64 not found!" >&2
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [ ! -f build/dropshell.arm64 ]; then
|
if [ ! -f output/dropshell.arm64 ]; then
|
||||||
echo "build/dropshell.arm64 not found!" >&2
|
echo "output/dropshell.arm64 not found!" >&2
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
echo "Builds complete:"
|
echo "Builds complete:"
|
||||||
ls -lh build/dropshell.*
|
ls -lh output/dropshell.*
|
||||||
|
|
||||||
|
cd $PREV_PWD
|
@ -1,6 +1,10 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
set -e
|
set -e
|
||||||
|
|
||||||
|
# directory of this script
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
echo "Script directory: $SCRIPT_DIR"
|
||||||
|
|
||||||
# Check for GITEA_TOKEN_DEPLOY or GITEA_TOKEN
|
# Check for GITEA_TOKEN_DEPLOY or GITEA_TOKEN
|
||||||
if [ -n "$GITEA_TOKEN_DEPLOY" ]; then
|
if [ -n "$GITEA_TOKEN_DEPLOY" ]; then
|
||||||
TOKEN="$GITEA_TOKEN_DEPLOY"
|
TOKEN="$GITEA_TOKEN_DEPLOY"
|
||||||
@ -11,27 +15,32 @@ else
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
$SCRIPT_DIR/multibuild.sh
|
||||||
|
BUILD_DIR=$SCRIPT_DIR/build
|
||||||
|
|
||||||
./multibuild.sh
|
OLD_PWD=$PWD
|
||||||
|
cd $SCRIPT_DIR
|
||||||
|
|
||||||
if [ ! -f "build/dropshell.amd64" ]; then
|
|
||||||
echo "build/dropshell.amd64 not found!" >&2
|
if [ ! -f "output/dropshell.amd64" ]; then
|
||||||
|
echo "output/dropshell.amd64 not found!" >&2
|
||||||
echo "Please run multibuild.sh first." >&2
|
echo "Please run multibuild.sh first." >&2
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [ ! -f "build/dropshell.arm64" ]; then
|
if [ ! -f "output/dropshell.arm64" ]; then
|
||||||
echo "build/dropshell.arm64 not found!" >&2
|
echo "output/dropshell.arm64 not found!" >&2
|
||||||
echo "Please run multibuild.sh first." >&2
|
echo "Please run multibuild.sh first." >&2
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
TAG=$(./build/dropshell.amd64 --version)
|
TAG=$("$SCRIPT_DIR/output/dropshell.amd64" --version)
|
||||||
|
[ -z "$TAG" ] && echo "Failed to get version from dropshell.amd64" >&2 && exit 1
|
||||||
|
|
||||||
echo "Publishing dropshell version $TAG"
|
echo "Publishing dropshell version $TAG"
|
||||||
|
|
||||||
# make sure we've commited.
|
# make sure we've commited.
|
||||||
git add . && git commit -m "dropshell release $TAG" && git push
|
git add "$SCRIPT_DIR/../" && git commit -m "dropshell release $TAG" && git push
|
||||||
|
|
||||||
|
|
||||||
# Find repo info from .git/config
|
# Find repo info from .git/config
|
||||||
@ -73,8 +82,10 @@ fi
|
|||||||
|
|
||||||
# Upload binaries and install.sh
|
# Upload binaries and install.sh
|
||||||
for FILE in dropshell.amd64 dropshell.arm64 install.sh server_autosetup.sh; do
|
for FILE in dropshell.amd64 dropshell.arm64 install.sh server_autosetup.sh; do
|
||||||
if [ -f "build/$FILE" ]; then
|
if [ -f "output/$FILE" ]; then
|
||||||
filetoupload="build/$FILE"
|
filetoupload="output/$FILE"
|
||||||
|
elif [ -f "../$FILE" ]; then
|
||||||
|
filetoupload="../$FILE"
|
||||||
elif [ -f "$FILE" ]; then
|
elif [ -f "$FILE" ]; then
|
||||||
filetoupload="$FILE"
|
filetoupload="$FILE"
|
||||||
else
|
else
|
||||||
@ -93,3 +104,5 @@ for FILE in dropshell.amd64 dropshell.arm64 install.sh server_autosetup.sh; do
|
|||||||
done
|
done
|
||||||
|
|
||||||
echo "Published dropshell version $TAG to $REPO_URL (tag $TAG) with binaries."
|
echo "Published dropshell version $TAG to $REPO_URL (tag $TAG) with binaries."
|
||||||
|
|
||||||
|
cd $OLD_PWD
|
@ -1,427 +0,0 @@
|
|||||||
#include <fstream>
|
|
||||||
#include <filesystem>
|
|
||||||
#include <string>
|
|
||||||
#include <iostream>
|
|
||||||
#include <cstring>
|
|
||||||
|
|
||||||
/*
|
|
||||||
|
|
||||||
THIS FILE IS AUTO-GENERATED BY DEHYDRATE.
|
|
||||||
DO NOT EDIT THIS FILE.
|
|
||||||
|
|
||||||
*/
|
|
||||||
|
|
||||||
|
|
||||||
#include "_agent.hpp"
|
|
||||||
namespace recreate_agent {
|
|
||||||
|
|
||||||
|
|
||||||
// Tiny dependency-free FNV-1a 64-bit hash
|
|
||||||
static uint64_t fnv1a_64(const void* data, size_t len) {
|
|
||||||
const uint8_t* p = static_cast<const uint8_t*>(data);
|
|
||||||
uint64_t h = 0xcbf29ce484222325ULL;
|
|
||||||
for (size_t i = 0; i < len; ++i)
|
|
||||||
h = (h ^ p[i]) * 0x100000001b3ULL;
|
|
||||||
return h;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// Base64 decoding function - no dependencies
|
|
||||||
static void base64_decode(const char* encoded_data, size_t encoded_len, unsigned char* output, size_t* output_len) {
|
|
||||||
const char* base64_chars =
|
|
||||||
"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
|
|
||||||
size_t out_pos = 0;
|
|
||||||
int val = 0, valb = -8;
|
|
||||||
|
|
||||||
for (size_t i = 0; i < encoded_len; i++) {
|
|
||||||
char c = encoded_data[i];
|
|
||||||
if (c == '=') break;
|
|
||||||
|
|
||||||
// Find position in base64_chars
|
|
||||||
const char* pos = strchr(base64_chars, c);
|
|
||||||
if (pos == nullptr) continue; // Skip invalid characters
|
|
||||||
|
|
||||||
val = (val << 6) + static_cast<int>(pos - base64_chars);
|
|
||||||
valb += 6;
|
|
||||||
if (valb >= 0) {
|
|
||||||
output[out_pos++] = static_cast<unsigned char>((val >> valb) & 0xFF);
|
|
||||||
valb -= 8;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
*output_len = out_pos;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Utility function to recreate a file with proper permissions
|
|
||||||
static bool _recreate_file_(const std::filesystem::path& outpath, uint64_t file_hash, std::filesystem::perms file_perms, const unsigned char* filedata, size_t filedata_len) {
|
|
||||||
namespace fs = std::filesystem;
|
|
||||||
bool needs_write = false;
|
|
||||||
|
|
||||||
// Check if file exists and has correct hash
|
|
||||||
if (fs::exists(outpath)) {
|
|
||||||
// Check content hash
|
|
||||||
std::ifstream in(outpath, std::ios::binary);
|
|
||||||
std::ostringstream oss;
|
|
||||||
oss << in.rdbuf();
|
|
||||||
std::string data = oss.str();
|
|
||||||
uint64_t existing_hash = fnv1a_64(data.data(), data.size());
|
|
||||||
needs_write = existing_hash != file_hash;
|
|
||||||
} else {
|
|
||||||
needs_write = true; // File doesn't exist, need to create it
|
|
||||||
}
|
|
||||||
|
|
||||||
bool needs_permission_update = true;
|
|
||||||
if (!needs_write) { // we always update permissions if the file is written or changed. Othewise we check.
|
|
||||||
fs::perms current_perms = fs::status(outpath).permissions();
|
|
||||||
needs_permission_update = current_perms != file_perms;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (needs_write) {
|
|
||||||
fs::create_directories(outpath.parent_path());
|
|
||||||
std::ofstream out(outpath, std::ios::binary);
|
|
||||||
out.write(reinterpret_cast<const char*>(filedata), filedata_len);
|
|
||||||
out.close();
|
|
||||||
// Set the file permissions
|
|
||||||
fs::permissions(outpath, file_perms);
|
|
||||||
|
|
||||||
if (!fs::exists(outpath)) {
|
|
||||||
std::cout << "[dehydrate] " << outpath.filename() << ": created\n";
|
|
||||||
} else {
|
|
||||||
std::cout << "[dehydrate] " << outpath.filename() << ": updated (hash changed)\n";
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (needs_permission_update) {
|
|
||||||
// Update only permissions
|
|
||||||
fs::permissions(outpath, file_perms);
|
|
||||||
std::cout << "[dehydrate] " << outpath.filename() << ": updated (permissions changed)\n";
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool recreate_tree(std::string destination_folder) {
|
|
||||||
namespace fs = std::filesystem;
|
|
||||||
bool any_written = false;
|
|
||||||
{
|
|
||||||
// File: datacommands.sh
|
|
||||||
fs::path outpath = fs::path(destination_folder) / "datacommands.sh";
|
|
||||||
static const char filedata_base64[] = "IyEvYmluL2Jhc2gKCiMgVGhpcyBzY3JpcHQgY29udGFpbnMgdGhlIGNvbW1vbiBjb2RlIGZvciB0"\
|
|
||||||
"aGUgYXV0b2NvbW1hbmRzLgoKTVlJRD0kKGlkIC11KQpNWUdSUD0kKGlkIC1nKQoKX2F1dG9jb21t"\
|
|
||||||
"YW5kcnVuX3ZvbHVtZSgpIHsKICAgIGxvY2FsIGNvbW1hbmQ9IiQxIgogICAgbG9jYWwgdm9sdW1l"\
|
|
||||||
"X25hbWU9IiQyIgogICAgbG9jYWwgYmFja3VwX2ZvbGRlcj0iJDMiCgogICAgY2FzZSAiJGNvbW1h"\
|
|
||||||
"bmQiIGluCiAgICAgICAgY3JlYXRlKQogICAgICAgICAgICBlY2hvICJDcmVhdGluZyB2b2x1bWUg"\
|
|
||||||
"JHt2b2x1bWVfbmFtZX0iCiAgICAgICAgICAgIGRvY2tlciB2b2x1bWUgY3JlYXRlICR7dm9sdW1l"\
|
|
||||||
"X25hbWV9CiAgICAgICAgICAgIDs7CiAgICAgICAgbnVrZSkKICAgICAgICAgICAgZWNobyAiTnVr"\
|
|
||||||
"aW5nIHZvbHVtZSAke3ZvbHVtZV9uYW1lfSIKICAgICAgICAgICAgZG9ja2VyIHZvbHVtZSBybSAk"\
|
|
||||||
"e3ZvbHVtZV9uYW1lfQogICAgICAgICAgICA7OwogICAgICAgIGJhY2t1cCkKICAgICAgICAgICAg"\
|
|
||||||
"ZWNobyAiQmFja2luZyB1cCB2b2x1bWUgJHt2b2x1bWVfbmFtZX0iCiAgICAgICAgICAgIGRvY2tl"\
|
|
||||||
"ciBydW4gLS1ybSAtdiAke3ZvbHVtZV9uYW1lfTovdm9sdW1lIC12ICR7YmFja3VwX2ZvbGRlcn06"\
|
|
||||||
"L2JhY2t1cCBkZWJpYW4gYmFzaCAtYyAidGFyIC1jenZmIC9iYWNrdXAvYmFja3VwLnRneiAtQyAv"\
|
|
||||||
"dm9sdW1lIC4gJiYgY2hvd24gLVIgJE1ZSUQ6JE1ZR1JQIC9iYWNrdXAiCiAgICAgICAgICAgIDs7"\
|
|
||||||
"CiAgICAgICAgcmVzdG9yZSkKICAgICAgICAgICAgZWNobyAiUmVzdG9yaW5nIHZvbHVtZSAke3Zv"\
|
|
||||||
"bHVtZV9uYW1lfSIKICAgICAgICAgICAgZG9ja2VyIHZvbHVtZSBybSAke3ZvbHVtZV9uYW1lfQog"\
|
|
||||||
"ICAgICAgICAgICBkb2NrZXIgdm9sdW1lIGNyZWF0ZSAke3ZvbHVtZV9uYW1lfQogICAgICAgICAg"\
|
|
||||||
"ICBkb2NrZXIgcnVuIC0tcm0gLXYgJHt2b2x1bWVfbmFtZX06L3ZvbHVtZSAtdiAke2JhY2t1cF9m"\
|
|
||||||
"b2xkZXJ9Oi9iYWNrdXAgZGViaWFuIGJhc2ggLWMgInRhciAteHp2ZiAvYmFja3VwL2JhY2t1cC50"\
|
|
||||||
"Z3ogLUMgL3ZvbHVtZSAtLXN0cmlwLWNvbXBvbmVudHM9MSIKICAgICAgICAgICAgOzsKICAgIGVz"\
|
|
||||||
"YWMKfSAgIAoKX2F1dG9jb21tYW5kcnVuX3BhdGgoKSB7CiAgICBsb2NhbCBjb21tYW5kPSIkMSIK"\
|
|
||||||
"ICAgIGxvY2FsIHBhdGg9IiQyIgogICAgbG9jYWwgYmFja3VwX2ZvbGRlcj0iJDMiCgogICAgY2Fz"\
|
|
||||||
"ZSAiJGNvbW1hbmQiIGluCiAgICAgICAgY3JlYXRlKQogICAgICAgICAgICBlY2hvICJDcmVhdGlu"\
|
|
||||||
"ZyBwYXRoICR7cGF0aH0iCiAgICAgICAgICAgIG1rZGlyIC1wICR7cGF0aH0KICAgICAgICAgICAg"\
|
|
||||||
"OzsKICAgICAgICBudWtlKQogICAgICAgICAgICBlY2hvICJOdWtpbmcgcGF0aCAke3BhdGh9Igog"\
|
|
||||||
"ICAgICAgICAgICBsb2NhbCBwYXRoX3BhcmVudD0kKGRpcm5hbWUgJHtwYXRofSkKICAgICAgICAg"\
|
|
||||||
"ICAgbG9jYWwgcGF0aF9jaGlsZD0kKGJhc2VuYW1lICR7cGF0aH0pCiAgICAgICAgICAgIGlmIFsg"\
|
|
||||||
"LWQgIiR7cGF0aF9wYXJlbnR9LyR7cGF0aF9jaGlsZH0iIF07IHRoZW4KICAgICAgICAgICAgICAg"\
|
|
||||||
"IGRvY2tlciBydW4gLS1ybSAtdiAke3BhdGhfcGFyZW50fTovdm9sdW1lIGRlYmlhbiBiYXNoIC1j"\
|
|
||||||
"ICJybSAtcmYgL3ZvbHVtZS8ke3BhdGhfY2hpbGR9IiB8fCBlY2hvICJGYWlsZWQgdG8gbnVrZSBw"\
|
|
||||||
"YXRoICR7cGF0aH0iCiAgICAgICAgICAgIGVsc2UKICAgICAgICAgICAgICAgIGVjaG8gIlBhdGgg"\
|
|
||||||
"JHtwYXRofSBkb2VzIG5vdCBleGlzdCAtIG5vdGhpbmcgdG8gbnVrZSIKICAgICAgICAgICAgZmkK"\
|
|
||||||
"ICAgICAgICAgICAgOzsKICAgICAgICBiYWNrdXApCiAgICAgICAgICAgIGVjaG8gIkJhY2tpbmcg"\
|
|
||||||
"dXAgcGF0aCAke3BhdGh9IgogICAgICAgICAgICBpZiBbIC1kICIke3BhdGh9IiBdOyB0aGVuCiAg"\
|
|
||||||
"ICAgICAgICAgICAgICBkb2NrZXIgcnVuIC0tcm0gLXYgJHtwYXRofTovcGF0aCAtdiAke2JhY2t1"\
|
|
||||||
"cF9mb2xkZXJ9Oi9iYWNrdXAgZGViaWFuIGJhc2ggLWMgInRhciAtY3p2ZiAvYmFja3VwL2JhY2t1"\
|
|
||||||
"cC50Z3ogLUMgL3BhdGggLiAmJiBjaG93biAtUiAkTVlJRDokTVlHUlAgL2JhY2t1cCIKICAgICAg"\
|
|
||||||
"ICAgICAgZWxzZQogICAgICAgICAgICAgICAgZWNobyAiUGF0aCAke3BhdGh9IGRvZXMgbm90IGV4"\
|
|
||||||
"aXN0IC0gbm90aGluZyB0byBiYWNrdXAiCiAgICAgICAgICAgIGZpCiAgICAgICAgICAgIDs7CiAg"\
|
|
||||||
"ICAgICAgcmVzdG9yZSkKICAgICAgICAgICAgZWNobyAiUmVzdG9yaW5nIHBhdGggJHtwYXRofSIK"\
|
|
||||||
"ICAgICAgICAgICAgdGFyIC14enZmICR7YmFja3VwX2ZvbGRlcn0vYmFja3VwLnRneiAtQyAke3Bh"\
|
|
||||||
"dGh9IC0tc3RyaXAtY29tcG9uZW50cz0xCiAgICAgICAgICAgIDs7CiAgICBlc2FjCn0KCl9hdXRv"\
|
|
||||||
"Y29tbWFuZHJ1bl9maWxlKCkgewogICAgbG9jYWwgY29tbWFuZD0iJDEiCiAgICBsb2NhbCBmaWxl"\
|
|
||||||
"cGF0aD0iJDIiCiAgICBsb2NhbCBiYWNrdXBfZm9sZGVyPSIkMyIKCiAgICBjYXNlICIkY29tbWFu"\
|
|
||||||
"ZCIgaW4KICAgICAgICBjcmVhdGUpCiAgICAgICAgICAgIDs7CiAgICAgICAgbnVrZSkKICAgICAg"\
|
|
||||||
"ICAgICAgcm0gLWYgJHtmaWxlcGF0aH0KICAgICAgICAgICAgOzsKICAgICAgICBiYWNrdXApCiAg"\
|
|
||||||
"ICAgICAgICAgIGVjaG8gIkJhY2tpbmcgdXAgZmlsZSAke2ZpbGVwYXRofSIKICAgICAgICAgICAg"\
|
|
||||||
"bG9jYWwgZmlsZV9wYXJlbnQ9JChkaXJuYW1lICR7ZmlsZXBhdGh9KQogICAgICAgICAgICBsb2Nh"\
|
|
||||||
"bCBmaWxlX25hbWU9JChiYXNlbmFtZSAke2ZpbGVwYXRofSkKICAgICAgICAgICAgaWYgWyAtZiAi"\
|
|
||||||
"JHtmaWxlX3BhcmVudH0vJHtmaWxlX25hbWV9IiBdOyB0aGVuCiAgICAgICAgICAgICAgICBkb2Nr"\
|
|
||||||
"ZXIgcnVuIC0tcm0gLXYgJHtmaWxlX3BhcmVudH06L3ZvbHVtZSAtdiAke2JhY2t1cF9mb2xkZXJ9"\
|
|
||||||
"Oi9iYWNrdXAgZGViaWFuIGJhc2ggLWMgImNwIC92b2x1bWUvJHtmaWxlX25hbWV9IC9iYWNrdXAv"\
|
|
||||||
"JHtmaWxlX25hbWV9ICYmIGNob3duIC1SICRNWUlEOiRNWUdSUCAvYmFja3VwIgogICAgICAgICAg"\
|
|
||||||
"ICBlbHNlCiAgICAgICAgICAgICAgICBlY2hvICJGaWxlICR7ZmlsZXBhdGh9IGRvZXMgbm90IGV4"\
|
|
||||||
"aXN0IC0gbm90aGluZyB0byBiYWNrdXAiCiAgICAgICAgICAgIGZpCiAgICAgICAgICAgIDs7CiAg"\
|
|
||||||
"ICAgICAgcmVzdG9yZSkKICAgICAgICAgICAgZWNobyAiUmVzdG9yaW5nIGZpbGUgJHtmaWxlcGF0"\
|
|
||||||
"aH0iCiAgICAgICAgICAgIGxvY2FsIGZpbGVfbmFtZT0kKGJhc2VuYW1lICR7ZmlsZXBhdGh9KQog"\
|
|
||||||
"ICAgICAgICAgICBjcCAke2JhY2t1cF9mb2xkZXJ9LyR7ZmlsZV9uYW1lfSAke2ZpbGVwYXRofQog"\
|
|
||||||
"ICAgICAgICAgICA7OwogICAgZXNhYwp9CgpfYXV0b2NvbW1hbmRwYXJzZSgpIHsKICAgICMgZmly"\
|
|
||||||
"c3QgYXJndW1lbnQgaXMgdGhlIGNvbW1hbmQKICAgICMgaWYgdGhlIGNvbW1hbmQgaXMgYmFja3Vw"\
|
|
||||||
"IG9yIHJlc3RvcmUsIHRoZW4gdGhlIGxhc3QgdHdvIGFyZ3VtZW50cyBhcmUgdGhlIGJhY2t1cCBm"\
|
|
||||||
"aWxlIGFuZCB0aGUgdGVtcG9yYXJ5IHBhdGgKICAgICMgYWxsIG90aGVyIGFyZ3VtZW50cyBhcmUg"\
|
|
||||||
"b2YgZm9ybToKICAgICMga2V5PXZhbHVlCiAgICAjIHdoZXJlIGtleSBjYW4gYmUgb25lIG9mIHZv"\
|
|
||||||
"bHVtZSwgcGF0aCBvciBmaWxlLgogICAgIyB2YWx1ZSBpcyB0aGUgcGF0aCBvciB2b2x1bWUgbmFt"\
|
|
||||||
"ZS4KCiAgICAjIHdlIGl0ZXJhdGUgb3ZlciB0aGUga2V5PXZhbHVlIGFyZ3VtZW50cywgYW5kIGZv"\
|
|
||||||
"ciBlYWNoIHdlIGNhbGw6CiAgICAjICAgIGF1dG9ydW4gPGNvbW1hbmQ+IDxiYWNrdXBmaWxlPiA8"\
|
|
||||||
"a2V5PiA8dmFsdWU+CgogICAgbG9jYWwgY29tbWFuZD0iJDEiCiAgICBzaGlmdAoKICAgIGxvY2Fs"\
|
|
||||||
"IGJhY2t1cF90ZW1wX3BhdGg9IiQxIgogICAgc2hpZnQKCiAgICBlY2hvICJhdXRvY29tbWFuZHBh"\
|
|
||||||
"cnNlOiBjb21tYW5kPSRjb21tYW5kIGJhY2t1cF90ZW1wX3BhdGg9JGJhY2t1cF90ZW1wX3BhdGgi"\
|
|
||||||
"CgogICAgIyBFeHRyYWN0IHRoZSBiYWNrdXAgZmlsZSBhbmQgdGVtcCBwYXRoIChsYXN0IHR3byBh"\
|
|
||||||
"cmd1bWVudHMpCiAgICBsb2NhbCBhcmdzPSgiJEAiKQogICAgbG9jYWwgYXJnX2NvdW50PSR7I2Fy"\
|
|
||||||
"Z3NbQF19CiAgICAKICAgICMgUHJvY2VzcyBhbGwga2V5PXZhbHVlIHBhaXJzCiAgICBmb3IgKChp"\
|
|
||||||
"PTA7IGk8JGFyZ19jb3VudDsgaSsrKSk7IGRvCiAgICAgICAgbG9jYWwgcGFpcj0iJHthcmdzWyRp"\
|
|
||||||
"XX0iCiAgICAgICAgCiAgICAgICAgIyBTa2lwIGlmIG5vdCBpbiBrZXk9dmFsdWUgZm9ybWF0CiAg"\
|
|
||||||
"ICAgICAgaWYgW1sgIiRwYWlyIiAhPSAqIj0iKiBdXTsgdGhlbgogICAgICAgICAgICBjb250aW51"\
|
|
||||||
"ZQogICAgICAgIGZpCiAgICAgICAgCiAgICAgICAgbG9jYWwga2V5PSIke3BhaXIlJT0qfSIKICAg"\
|
|
||||||
"ICAgICBsb2NhbCB2YWx1ZT0iJHtwYWlyIyo9fSIKCiAgICAgICAgIyBjcmVhdGUgYmFja3VwIGZv"\
|
|
||||||
"bGRlciB1bmlxdWUgdG8ga2V5L3ZhbHVlLgogICAgICAgIGxvY2FsIGJmb2xkZXI9JChlY2hvICIk"\
|
|
||||||
"e2tleX1fJHt2YWx1ZX0iIHwgdHIgLWNkICdbOmFsbnVtOl1fLScpCiAgICAgICAgbG9jYWwgdGFy"\
|
|
||||||
"Z2V0cGF0aD0iJHtiYWNrdXBfdGVtcF9wYXRofS8ke2Jmb2xkZXJ9IgogICAgICAgIG1rZGlyIC1w"\
|
|
||||||
"ICR7dGFyZ2V0cGF0aH0KCiAgICAgICAgIyBLZXkgbXVzdCBiZSBvbmUgb2Ygdm9sdW1lLCBwYXRo"\
|
|
||||||
"IG9yIGZpbGUKICAgICAgICBjYXNlICIka2V5IiBpbgogICAgICAgICAgICB2b2x1bWUpCiAgICAg"\
|
|
||||||
"ICAgICAgICAgICBfYXV0b2NvbW1hbmRydW5fdm9sdW1lICIkY29tbWFuZCIgIiR2YWx1ZSIgIiR0"\
|
|
||||||
"YXJnZXRwYXRoIgogICAgICAgICAgICAgICAgOzsKICAgICAgICAgICAgcGF0aCkKICAgICAgICAg"\
|
|
||||||
"ICAgICAgIF9hdXRvY29tbWFuZHJ1bl9wYXRoICIkY29tbWFuZCIgIiR2YWx1ZSIgIiR0YXJnZXRw"\
|
|
||||||
"YXRoIgogICAgICAgICAgICAgICAgOzsKICAgICAgICAgICAgZmlsZSkKICAgICAgICAgICAgICAg"\
|
|
||||||
"IF9hdXRvY29tbWFuZHJ1bl9maWxlICIkY29tbWFuZCIgIiR2YWx1ZSIgIiR0YXJnZXRwYXRoIgog"\
|
|
||||||
"ICAgICAgICAgICAgICAgOzsKICAgICAgICAgICAgKikKICAgICAgICAgICAgICAgIF9kaWUgIlVu"\
|
|
||||||
"a25vd24ga2V5ICRrZXkgcGFzc2VkIHRvIGF1dG8ke2NvbW1hbmR9LiBXZSBvbmx5IHN1cHBvcnQg"\
|
|
||||||
"dm9sdW1lLCBwYXRoIGFuZCBmaWxlLiIKICAgICAgICAgICAgICAgIDs7CiAgICAgICAgZXNhYwog"\
|
|
||||||
"ICAgZG9uZQp9CgoKZGF0YWNyZWF0ZSgpIHsKICAgIF9hdXRvY29tbWFuZHBhcnNlIGNyZWF0ZSBu"\
|
|
||||||
"b25lICIkQCIKfQoKCmRhdGFudWtlKCkgewogICAgX2F1dG9jb21tYW5kcGFyc2UgbnVrZSBub25l"\
|
|
||||||
"ICIkQCIKfQoKZGF0YWJhY2t1cCgpIHsKICAgIF9jaGVja19yZXF1aXJlZF9lbnZfdmFycyAiQkFD"\
|
|
||||||
"S1VQX0ZJTEUiICJURU1QX0RJUiIKICAgIEJBQ0tVUF9URU1QX1BBVEg9IiRURU1QX0RJUi9iYWNr"\
|
|
||||||
"dXAiCgoKICAgIG1rZGlyIC1wICIkQkFDS1VQX1RFTVBfUEFUSCIKICAgIGVjaG8gIl9hdXRvY29t"\
|
|
||||||
"bWFuZHBhcnNlIFtiYWNrdXBdIFskQkFDS1VQX1RFTVBfUEFUSF0gWyRAXSIKICAgIF9hdXRvY29t"\
|
|
||||||
"bWFuZHBhcnNlIGJhY2t1cCAiJEJBQ0tVUF9URU1QX1BBVEgiICIkQCIKCiAgICB0YXIgemN2ZiAi"\
|
|
||||||
"JEJBQ0tVUF9GSUxFIiAtQyAiJEJBQ0tVUF9URU1QX1BBVEgiIC4KfQoKZGF0YXJlc3RvcmUoKSB7"\
|
|
||||||
"CiAgICBfY2hlY2tfcmVxdWlyZWRfZW52X3ZhcnMgIkJBQ0tVUF9GSUxFIiAiVEVNUF9ESVIiCiAg"\
|
|
||||||
"ICBCQUNLVVBfVEVNUF9QQVRIPSIkVEVNUF9ESVIvcmVzdG9yZSIKCiAgICBlY2hvICJfYXV0b2Nv"\
|
|
||||||
"bW1hbmRwYXJzZSBbcmVzdG9yZV0gWyRCQUNLVVBfVEVNUF9QQVRIXSBbJEBdIgoKICAgIG1rZGly"\
|
|
||||||
"IC1wICIkQkFDS1VQX1RFTVBfUEFUSCIKICAgIHRhciB6eHZmICIkQkFDS1VQX0ZJTEUiIC1DICIk"\
|
|
||||||
"QkFDS1VQX1RFTVBfUEFUSCIgLS1zdHJpcC1jb21wb25lbnRzPTEKCiAgICBfYXV0b2NvbW1hbmRw"\
|
|
||||||
"YXJzZSByZXN0b3JlICIkQkFDS1VQX1RFTVBfUEFUSCIgIiRAIgp9Cg==";
|
|
||||||
|
|
||||||
// Decode Base64 data
|
|
||||||
size_t decoded_size = (strlen(filedata_base64) * 3) / 4;
|
|
||||||
unsigned char* decoded_data = new unsigned char[decoded_size];
|
|
||||||
size_t actual_size;
|
|
||||||
base64_decode(filedata_base64, strlen(filedata_base64), decoded_data, &actual_size);
|
|
||||||
|
|
||||||
bool file_written = _recreate_file_(outpath, 6443138635497166205ULL, std::filesystem::perms(493), decoded_data, actual_size);
|
|
||||||
delete[] decoded_data;
|
|
||||||
any_written = any_written || file_written;
|
|
||||||
}
|
|
||||||
{
|
|
||||||
// File: _allservicesstatus.sh
|
|
||||||
fs::path outpath = fs::path(destination_folder) / "_allservicesstatus.sh";
|
|
||||||
static const char filedata_base64[] = "IyEvYmluL2Jhc2gKCiMgVGhpcyBzY3JpcHQgY2hlY2tzIEFMTCBzZXJ2aWNlcyBvbiB0aGUgc2Vy"\
|
|
||||||
"dmVyIGFuZCByZXR1cm5zIGEgc3RhdHVzIGZvciBlYWNoLgoKIyBSZXR1cm4gZm9ybWF0IGlzIHNp"\
|
|
||||||
"bXBsZSBFTlYgd2l0aCB0aGUgZm9sbG93aW5nIGZvcm1hdDoKIyBTRVJWSUNFX05BTUVfSEVBTFRI"\
|
|
||||||
"PWhlYWx0aHl8dW5oZWFsdGh5fHVua25vd24KIyBTRVJWSUNFX05BTUVfUE9SVFM9cG9ydDEscG9y"\
|
|
||||||
"dDIscG9ydDMKCiMgR2V0IGFsbCBzZXJ2aWNlcyBvbiB0aGUgc2VydmVyClNDUklQVF9ESVI9IiQo"\
|
|
||||||
"ZGlybmFtZSAiJDAiKSIKCiAgICAjIC8vLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0t"\
|
|
||||||
"LS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0t"\
|
|
||||||
"LS0tLS0tCiAgICAjIC8vIHJlbW90ZSBwYXRocwogICAgIyAvLyBEUk9QU0hFTExfRElSCiAgICAj"\
|
|
||||||
"IC8vICAgfC0tIGJhY2t1cHMKICAgICMgLy8gICB8LS0gdGVtcF9maWxlcwogICAgIyAvLyAgIHwt"\
|
|
||||||
"LSBhZ2VudAogICAgIyAvLyAgIHwgICB8LS0gYmI2NAogICAgIyAvLyAgIHwgICB8LS0gKG90aGVy"\
|
|
||||||
"IGFnZW50IGZpbGVzLCBpbmNsdWRpbmcgX2FsbHNlcnZpY2Vzc3RhdHVzLnNoKQogICAgIyAvLyAg"\
|
|
||||||
"IHwtLSBzZXJ2aWNlcwogICAgIyAvLyAgICAgICB8LS0gc2VydmljZSBuYW1lCiAgICAjIC8vICAg"\
|
|
||||||
"ICAgICAgICB8LS0gY29uZmlnCiAgICAjIC8vICAgICAgICAgICAgICAgfC0tIHNlcnZpY2UuZW52"\
|
|
||||||
"IChhY3R1YWwgc2VydmljZSBjb25maWcpCiAgICAjIC8vICAgICAgICAgICAgICAgfC0tIC50ZW1w"\
|
|
||||||
"bGF0ZV9pbmZvLmVudgogICAgIyAvLyAgICAgICAgICAgfC0tIHRlbXBsYXRlCiAgICAjIC8vICAg"\
|
|
||||||
"ICAgICAgICAgICAgfC0tIF9kZWZhdWx0LmVudgogICAgIyAvLyAgICAgICAgICAgICAgIHwtLSAo"\
|
|
||||||
"c2NyaXB0IGZpbGVzKQogICAgIyAvLyAgICAgICAgICAgICAgIHwtLSBjb25maWcKICAgICMgLy8g"\
|
|
||||||
"ICAgICAgICAgICAgICAgICAgfC0tIHNlcnZpY2UuZW52IChkZWZhdWx0IHNlcnZpY2UgY29uZmln"\
|
|
||||||
"KQogICAgIyAvLyAgICAgICAgICAgICAgICAgICB8LS0gLnRlbXBsYXRlX2luZm8uZW52CiAgICAj"\
|
|
||||||
"IC8vICAgICAgICAgICAgICAgICAgIHwtLSAob3RoZXIgY29uZmlnIGZpbGVzIGZvciBzcGVjaWZp"\
|
|
||||||
"YyBzZXJ2ZXImc2VydmljZSkKCiMgR2V0IGFsbCBzZXJ2aWNlcyBvbiB0aGUgc2VydmVyClNFUlZJ"\
|
|
||||||
"Q0VTX1BBVEg9JChyZWFscGF0aCAiJHtTQ1JJUFRfRElSfS8uLi9zZXJ2aWNlcy8iKQoKQ1VSUkVO"\
|
|
||||||
"VF9PVVRQVVQ9IiIKQ1VSUkVOVF9FWElUX0NPREU9MAoKbG9hZF9kb3RlbnYoKXsKICAgIGxvY2Fs"\
|
|
||||||
"IGZpbGVfcGF0aD0kMQogICAgaWYgWyAtZiAiJHtmaWxlX3BhdGh9IiBdOyB0aGVuCiAgICAgICAg"\
|
|
||||||
"c291cmNlICIke2ZpbGVfcGF0aH0iCiAgICBmaQp9CgpfY2hlY2tfcmVxdWlyZWRfZW52X3ZhcnNf"\
|
|
||||||
"YWxsc2VydmljZXNzdGF0dXMoKSB7CiAgICBsb2NhbCByZXF1aXJlZF92YXJzPSgiJEAiKQogICAg"\
|
|
||||||
"Zm9yIHZhciBpbiAiJHtyZXF1aXJlZF92YXJzW0BdfSI7IGRvCiAgICAgICAgaWYgWyAteiAiJHsh"\
|
|
||||||
"dmFyfSIgXTsgdGhlbgogICAgICAgICAgICBfZGllICJSZXF1aXJlZCBlbnZpcm9ubWVudCB2YXJp"\
|
|
||||||
"YWJsZSAkdmFyIGlzIG5vdCBzZXQiCiAgICAgICAgZmkKICAgIGRvbmUKfQoKZnVuY3Rpb24gcnVu"\
|
|
||||||
"X2NvbW1hbmQoKSB7CiAgICBsb2NhbCBzZXJ2aWNlX3BhdGg9JDEKICAgIGxvY2FsIGNvbW1hbmQ9"\
|
|
||||||
"JDIKICAgIGxvY2FsIGNhcHR1cmVfb3V0cHV0PSR7MzotZmFsc2V9ICAjIGRlZmF1bHQgdG8gZmFs"\
|
|
||||||
"c2UgaWYgbm90IHNwZWNpZmllZAoKICAgICMgY2hlY2sgaWYgdGhlIGNvbW1hbmQgaXMgYSBmaWxl"\
|
|
||||||
"CiAgICBpZiBbICEgLWYgIiR7c2VydmljZV9wYXRofS90ZW1wbGF0ZS8ke2NvbW1hbmR9LnNoIiBd"\
|
|
||||||
"OyB0aGVuCiAgICAgICAgcmV0dXJuOwogICAgZmkKCiAgICAjIHJ1biB0aGUgY29tbWFuZCBpbiBh"\
|
|
||||||
"IHN1YnNoZWxsIHRvIHByZXZlbnQgZW52aXJvbm1lbnQgY2hhbmdlcwogICAgQ1VSUkVOVF9PVVRQ"\
|
|
||||||
"VVQ9JCgKICAgICAgICBzZXQgLWEKCiAgICAgICAgbG9hZF9kb3RlbnYgIiR7c2VydmljZV9wYXRo"\
|
|
||||||
"fS90ZW1wbGF0ZS9fZGVmYXVsdC5lbnYiCiAgICAgICAgbG9hZF9kb3RlbnYgIiR7c2VydmljZV9w"\
|
|
||||||
"YXRofS9jb25maWcvc2VydmljZS5lbnYiCiAgICAgICAgbG9hZF9kb3RlbnYgIiR7c2VydmljZV9w"\
|
|
||||||
"YXRofS9jb25maWcvLnRlbXBsYXRlX2luZm8uZW52IgoKICAgICAgICAjIHVwZGF0ZSB0aGUgbWFp"\
|
|
||||||
"biB2YXJpYWJsZXMuCiAgICAgICAgQ09ORklHX1BBVEg9IiR7c2VydmljZV9wYXRofS9jb25maWci"\
|
|
||||||
"CiAgICAgICAgU0VSVklDRT0iJHtTRVJWSUNFX05BTUV9IgoKICAgICAgICBzZXQgK2EKCiAgICAg"\
|
|
||||||
"ICAgX2NoZWNrX3JlcXVpcmVkX2Vudl92YXJzX2FsbHNlcnZpY2Vzc3RhdHVzICJDT05GSUdfUEFU"\
|
|
||||||
"SCIgIlNFUlZFUiIgIlNFUlZJQ0UiICJBR0VOVF9QQVRIIiAiSE9TVF9OQU1FIiAiVEVNUExBVEUi"\
|
|
||||||
"CgogICAgICAgIGlmIFsgIiRjYXB0dXJlX291dHB1dCIgPSAidHJ1ZSIgXTsgdGhlbgogICAgICAg"\
|
|
||||||
"ICAgICAjIENhcHR1cmUgYW5kIHJldHVybiBvdXRwdXQKICAgICAgICAgICAgYmFzaCAiJHtzZXJ2"\
|
|
||||||
"aWNlX3BhdGh9L3RlbXBsYXRlLyR7Y29tbWFuZH0uc2giIDI+JjEKICAgICAgICBlbHNlCiAgICAg"\
|
|
||||||
"ICAgICAgICMgUnVuIHNpbGVudGx5IGFuZCByZXR1cm4gZXhpdCBjb2RlCiAgICAgICAgICAgIGJh"\
|
|
||||||
"c2ggIiR7c2VydmljZV9wYXRofS90ZW1wbGF0ZS8ke2NvbW1hbmR9LnNoIiA+IC9kZXYvbnVsbCAy"\
|
|
||||||
"PiYxCiAgICAgICAgZmkKICAgICkKICAgIENVUlJFTlRfRVhJVF9DT0RFPSQ/Cn0KCmZ1bmN0aW9u"\
|
|
||||||
"IGNvbW1hbmRfZXhpc3RzKCkgewogICAgbG9jYWwgc2VydmljZV9wYXRoPSQxCiAgICBsb2NhbCBj"\
|
|
||||||
"b21tYW5kPSQyCiAgICBpZiBbICEgLWYgIiR7c2VydmljZV9wYXRofS90ZW1wbGF0ZS8ke2NvbW1h"\
|
|
||||||
"bmR9LnNoIiBdOyB0aGVuCiAgICAgICAgcmV0dXJuIDEKICAgIGZpCiAgICByZXR1cm4gMAp9CgoK"\
|
|
||||||
"CiMgR2V0IGFsbCBzZXJ2aWNlIG5hbWVzClNFUlZJQ0VfTkFNRVM9JChscyAiJHtTRVJWSUNFU19Q"\
|
|
||||||
"QVRIfSIpCgojIEl0ZXJhdGUgb3ZlciBhbGwgc2VydmljZSBuYW1lcwpmb3IgU0VSVklDRV9OQU1F"\
|
|
||||||
"IGluICR7U0VSVklDRV9OQU1FU307IGRvCgogICAgU0VSVklDRV9QQVRIPSQocmVhbHBhdGggIiR7"\
|
|
||||||
"U0VSVklDRVNfUEFUSH0vJHtTRVJWSUNFX05BTUV9IikKCiAgICAjLS0tLS0tLS0tLS0tLS0tLS0t"\
|
|
||||||
"LS0tLS0tLS0tLS0tLS0KICAgICMgR2V0IHRoZSBzZXJ2aWNlIGhlYWx0aAogICAgaWYgISBjb21t"\
|
|
||||||
"YW5kX2V4aXN0cyAiJHtTRVJWSUNFX1BBVEh9IiAic3RhdHVzIjsgdGhlbgogICAgICAgIFNFUlZJ"\
|
|
||||||
"Q0VfSEVBTFRIPSJ1bmtub3duIgogICAgZWxzZQogICAgICAgIHJ1bl9jb21tYW5kICIke1NFUlZJ"\
|
|
||||||
"Q0VfUEFUSH0iICJzdGF0dXMiICJmYWxzZSIKICAgICAgICBpZiBbICIke0NVUlJFTlRfRVhJVF9D"\
|
|
||||||
"T0RFfSIgLWVxIDAgXTsgdGhlbgogICAgICAgICAgICBTRVJWSUNFX0hFQUxUSD0iaGVhbHRoeSIK"\
|
|
||||||
"ICAgICAgICBlbHNlCiAgICAgICAgICAgIFNFUlZJQ0VfSEVBTFRIPSJ1bmhlYWx0aHkiCiAgICAg"\
|
|
||||||
"ICAgZmkKICAgIGZpCgogICAgIy0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tCiAgICAj"\
|
|
||||||
"IEdldCB0aGUgc2VydmljZSBwb3J0cwogICAgaWYgISBjb21tYW5kX2V4aXN0cyAiJHtTRVJWSUNF"\
|
|
||||||
"X1BBVEh9IiAicG9ydHMiOyB0aGVuCiAgICAgICAgU0VSVklDRV9QT1JUUz0iIgogICAgZWxzZQog"\
|
|
||||||
"ICAgICAgIHJ1bl9jb21tYW5kICIke1NFUlZJQ0VfUEFUSH0iICJwb3J0cyIgInRydWUiCiAgICAg"\
|
|
||||||
"ICAgU0VSVklDRV9QT1JUUz0iJHtDVVJSRU5UX09VVFBVVH0iCiAgICBmaQoKICAgICMtLS0tLS0t"\
|
|
||||||
"LS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLQogICAgIyByZXR1cm4gdGhlIGhlYWx0aCBhbmQgcG9y"\
|
|
||||||
"dHMKICAgIGVjaG8gIiR7U0VSVklDRV9OQU1FfV9IRUFMVEg9JHtTRVJWSUNFX0hFQUxUSH0iCiAg"\
|
|
||||||
"ICBlY2hvICIke1NFUlZJQ0VfTkFNRX1fUE9SVFM9JHtTRVJWSUNFX1BPUlRTfSIKZG9uZQo=";
|
|
||||||
|
|
||||||
// Decode Base64 data
|
|
||||||
size_t decoded_size = (strlen(filedata_base64) * 3) / 4;
|
|
||||||
unsigned char* decoded_data = new unsigned char[decoded_size];
|
|
||||||
size_t actual_size;
|
|
||||||
base64_decode(filedata_base64, strlen(filedata_base64), decoded_data, &actual_size);
|
|
||||||
|
|
||||||
bool file_written = _recreate_file_(outpath, 4383289270743338040ULL, std::filesystem::perms(493), decoded_data, actual_size);
|
|
||||||
delete[] decoded_data;
|
|
||||||
any_written = any_written || file_written;
|
|
||||||
}
|
|
||||||
{
|
|
||||||
// File: common.sh
|
|
||||||
fs::path outpath = fs::path(destination_folder) / "common.sh";
|
|
||||||
static const char filedata_base64[] = "IyBDT01NT04gRlVOQ1RJT05TCiMgSkRFCiMgMjAyNS0wNS0wMwoKIyBUaGlzIGZpbGUgaXMgYXZh"\
|
|
||||||
"aWxhYmxlIFRPICoqKkFMTCoqKiB0ZW1wbGF0ZXMsIGFzICR7QUdFTlRfUEFUSH0vX2NvbW1vbi5z"\
|
|
||||||
"aAoKIyAtLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0t"\
|
|
||||||
"LS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tCgoj"\
|
|
||||||
"IHN1bW1hcnkgb2YgZnVuY3Rpb25zOgojICAgX2RpZSAibWVzc2FnZSIgICAgICAgICAgICAgICAg"\
|
|
||||||
"ICAgICAgICAgOiBQcmludHMgYW4gZXJyb3IgbWVzc2FnZSBpbiByZWQgYW5kIGV4aXRzIHdpdGgg"\
|
|
||||||
"c3RhdHVzIGNvZGUgMS4KIyAgIF9ncmV5X3N0YXJ0ICAgICAgICAgICAgICAgICAgICAgICAgICAg"\
|
|
||||||
"IDogU3dpdGNoZXMgdGVybWluYWwgb3V0cHV0IGNvbG9yIHRvIGdyZXkuCiMgICBfZ3JleV9lbmQg"\
|
|
||||||
"ICAgICAgICAgICAgICAgICAgICAgICAgICAgICA6IFJlc2V0cyB0ZXJtaW5hbCBvdXRwdXQgY29s"\
|
|
||||||
"b3IgZnJvbSBncmV5LgojICAgX2NyZWF0ZV9hbmRfc3RhcnRfY29udGFpbmVyICI8cnVuX2NtZD4i"\
|
|
||||||
"IDxjb250YWluZXJfbmFtZT4gOiBDcmVhdGVzL3N0YXJ0cyBhIGNvbnRhaW5lciwgdmVyaWZ5aW5n"\
|
|
||||||
"IGl0IHJ1bnMuCiMgICBfY3JlYXRlX2ZvbGRlciA8Zm9sZGVyX3BhdGg+ICAgICAgICAgICA6IENy"\
|
|
||||||
"ZWF0ZXMgYSBkaXJlY3RvcnkgaWYgaXQgZG9lc24ndCBleGlzdCAoY2htb2QgNzc3KS4KIyAgIF9j"\
|
|
||||||
"aGVja19kb2NrZXJfaW5zdGFsbGVkICAgICAgICAgICAgICAgOiBDaGVja3MgaWYgRG9ja2VyIGlz"\
|
|
||||||
"IGluc3RhbGxlZCwgcnVubmluZywgYW5kIHVzZXIgaGFzIHBlcm1pc3Npb24uIFJldHVybnMgMSBv"\
|
|
||||||
"biBmYWlsdXJlLgojICAgX2lzX2NvbnRhaW5lcl9leGlzdHMgPGNvbnRhaW5lcl9uYW1lPiA6IENo"\
|
|
||||||
"ZWNrcyBpZiBhIGNvbnRhaW5lciAoYW55IHN0YXRlKSBleGlzdHMuIFJldHVybnMgMSBpZiBub3Qg"\
|
|
||||||
"Zm91bmQuCiMgICBfaXNfY29udGFpbmVyX3J1bm5pbmcgPGNvbnRhaW5lcl9uYW1lPjogQ2hlY2tz"\
|
|
||||||
"IGlmIGEgY29udGFpbmVyIGlzIGN1cnJlbnRseSBydW5uaW5nLiBSZXR1cm5zIDEgaWYgbm90IHJ1"\
|
|
||||||
"bm5pbmcuCiMgICBfZ2V0X2NvbnRhaW5lcl9pZCA8Y29udGFpbmVyX25hbWU+ICAgIDogUHJpbnRz"\
|
|
||||||
"IHRoZSBJRCBvZiB0aGUgbmFtZWQgY29udGFpbmVyLgojICAgX2dldF9jb250YWluZXJfc3RhdHVz"\
|
|
||||||
"IDxjb250YWluZXJfbmFtZT46IFByaW50cyB0aGUgc3RhdHVzIHN0cmluZyBvZiB0aGUgbmFtZWQg"\
|
|
||||||
"Y29udGFpbmVyLgojICAgX3N0YXJ0X2NvbnRhaW5lciA8Y29udGFpbmVyX25hbWU+ICAgICA6IFN0"\
|
|
||||||
"YXJ0cyBhbiBleGlzdGluZywgc3RvcHBlZCBjb250YWluZXIuCiMgICBfc3RvcF9jb250YWluZXIg"\
|
|
||||||
"PGNvbnRhaW5lcl9uYW1lPiAgICAgIDogU3RvcHMgYSBydW5uaW5nIGNvbnRhaW5lci4KIyAgIF9y"\
|
|
||||||
"ZW1vdmVfY29udGFpbmVyIDxjb250YWluZXJfbmFtZT4gICAgOiBTdG9wcyAoaWYgbmVlZGVkKSBh"\
|
|
||||||
"bmQgcmVtb3ZlcyBhIGNvbnRhaW5lci4KIyAgIF9nZXRfY29udGFpbmVyX2xvZ3MgPGNvbnRhaW5l"\
|
|
||||||
"cl9uYW1lPiAgOiBQcmludHMgdGhlIGxvZ3MgZm9yIGEgY29udGFpbmVyLgojICAgX2NoZWNrX3Jl"\
|
|
||||||
"cXVpcmVkX2Vudl92YXJzICJWQVIxIiAuLi4gICAgOiBDaGVja3MgaWYgbGlzdGVkIGVudmlyb25t"\
|
|
||||||
"ZW50IHZhcmlhYmxlcyBhcmUgc2V0OyBjYWxscyBfZGllKCkgaWYgYW55IGFyZSBtaXNzaW5nLgoj"\
|
|
||||||
"ICAgX3Jvb3RfcmVtb3ZlX3RyZWUgPHBhdGg+ICAgICAgICAgICAgICA6IFJlbW92ZXMgYSBwYXRo"\
|
|
||||||
"IHVzaW5nIGEgcm9vdCBEb2NrZXIgY29udGFpbmVyIChmb3IgcGVybWlzc2lvbnMpLgoKIyAtLS0t"\
|
|
||||||
"LS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0t"\
|
|
||||||
"LS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tLS0tCgojIFByaW50cyBh"\
|
|
||||||
"biBlcnJvciBtZXNzYWdlIGluIHJlZCBhbmQgZXhpdHMgd2l0aCBzdGF0dXMgY29kZSAxLgpfZGll"\
|
|
||||||
"KCkgewogICAgZWNobyAtZSAiXDAzM1s5MW1FcnJvcjogJDFcMDMzWzBtIgogICAgZXhpdCAxCn0K"\
|
|
||||||
"CiMgU3dpdGNoZXMgdGVybWluYWwgb3V0cHV0IGNvbG9yIHRvIGdyZXkuCl9ncmV5X3N0YXJ0KCkg"\
|
|
||||||
"ewogICAgZWNobyAtZSAtbiAiXDAzM1s5MG0iCn0KCiMgUmVzZXRzIHRlcm1pbmFsIG91dHB1dCBj"\
|
|
||||||
"b2xvciBmcm9tIGdyZXkuCl9ncmV5X2VuZCgpIHsKICAgIGVjaG8gLWUgLW4gIlwwMzNbMG0iCn0K"\
|
|
||||||
"CiMgQ3JlYXRlcy9zdGFydHMgYSBjb250YWluZXIsIHZlcmlmeWluZyBpdCBydW5zLgpfY3JlYXRl"\
|
|
||||||
"X2FuZF9zdGFydF9jb250YWluZXIoKSB7CiAgICBpZiBbIC16ICIkMSIgXSB8fCBbIC16ICIkMiIg"\
|
|
||||||
"XTsgdGhlbgogICAgICAgIF9kaWUgIlRlbXBsYXRlIGVycm9yOiBjcmVhdGVfYW5kX3N0YXJ0X2Nv"\
|
|
||||||
"bnRhaW5lciA8cnVuX2NtZD4gPGNvbnRhaW5lcl9uYW1lPiIKICAgIGZpCgogICAgbG9jYWwgcnVu"\
|
|
||||||
"X2NtZD0iJDEiCiAgICBsb2NhbCBjb250YWluZXJfbmFtZT0iJDIiCgogICAgaWYgX2lzX2NvbnRh"\
|
|
||||||
"aW5lcl9leGlzdHMgJGNvbnRhaW5lcl9uYW1lOyB0aGVuCiAgICAgICAgX2lzX2NvbnRhaW5lcl9y"\
|
|
||||||
"dW5uaW5nICRjb250YWluZXJfbmFtZSAmJiByZXR1cm4gMAogICAgICAgIF9zdGFydF9jb250YWlu"\
|
|
||||||
"ZXIgJGNvbnRhaW5lcl9uYW1lCiAgICBlbHNlCiAgICAgICAgX2dyZXlfc3RhcnQKICAgICAgICAk"\
|
|
||||||
"cnVuX2NtZAogICAgICAgIF9ncmV5X2VuZAogICAgZmkKCiAgICBpZiAhIF9pc19jb250YWluZXJf"\
|
|
||||||
"cnVubmluZyAkY29udGFpbmVyX25hbWU7IHRoZW4KICAgICAgICBfZGllICJDb250YWluZXIgJHtj"\
|
|
||||||
"b250YWluZXJfbmFtZX0gZmFpbGVkIHRvIHN0YXJ0IgogICAgZmkKCiAgICBJRD0kKF9nZXRfY29u"\
|
|
||||||
"dGFpbmVyX2lkICRjb250YWluZXJfbmFtZSkKICAgIGVjaG8gIkNvbnRhaW5lciAke2NvbnRhaW5l"\
|
|
||||||
"cl9uYW1lfSBpcyBydW5uaW5nIHdpdGggSUQgJHtJRH0iCn0KCiMgQ3JlYXRlcyBhIGRpcmVjdG9y"\
|
|
||||||
"eSBpZiBpdCBkb2Vzbid0IGV4aXN0IChjaG1vZCA3NzcpLgpfY3JlYXRlX2ZvbGRlcigpIHsKICAg"\
|
|
||||||
"IGxvY2FsIGZvbGRlcj0iJDEiCiAgICBpZiBbIC1kICIkZm9sZGVyIiBdOyB0aGVuCiAgICAgICAg"\
|
|
||||||
"cmV0dXJuIDAKICAgIGZpCiAgICBpZiAhIG1rZGlyIC1wICIkZm9sZGVyIjsgdGhlbgogICAgICAg"\
|
|
||||||
"IF9kaWUgIkZhaWxlZCB0byBjcmVhdGUgZm9sZGVyOiAkZm9sZGVyIgogICAgZmkKICAgIGNobW9k"\
|
|
||||||
"IDc3NyAiJGZvbGRlciIKICAgIGVjaG8gIkZvbGRlciBjcmVhdGVkOiAkZm9sZGVyIgp9CgojIENo"\
|
|
||||||
"ZWNrcyBpZiBEb2NrZXIgaXMgaW5zdGFsbGVkLCBydW5uaW5nLCBhbmQgdXNlciBoYXMgcGVybWlz"\
|
|
||||||
"c2lvbi4gUmV0dXJucyAxIG9uIGZhaWx1cmUuCl9jaGVja19kb2NrZXJfaW5zdGFsbGVkKCkgewog"\
|
|
||||||
"ICAgaWYgISBjb21tYW5kIC12IGRvY2tlciAmPiAvZGV2L251bGw7IHRoZW4KICAgICAgICBlY2hv"\
|
|
||||||
"ICJEb2NrZXIgaXMgbm90IGluc3RhbGxlZCIKICAgICAgICByZXR1cm4gMQogICAgZmkKCiAgICAj"\
|
|
||||||
"IGNoZWNrIGlmIGRvY2tlciBkYWVtb24gaXMgcnVubmluZwogICAgaWYgISBkb2NrZXIgaW5mbyAm"\
|
|
||||||
"PiAvZGV2L251bGw7IHRoZW4KICAgICAgICBlY2hvICJEb2NrZXIgZGFlbW9uIGlzIG5vdCBydW5u"\
|
|
||||||
"aW5nIgogICAgICAgIHJldHVybiAxCiAgICBmaQoKICAgICMgY2hlY2sgaWYgdXNlciBoYXMgcGVy"\
|
|
||||||
"bWlzc2lvbiB0byBydW4gZG9ja2VyCiAgICBpZiAhIGRvY2tlciBydW4gLS1ybSBoZWxsby13b3Js"\
|
|
||||||
"ZCAmPiAvZGV2L251bGw7IHRoZW4KICAgICAgICBlY2hvICJVc2VyIGRvZXMgbm90IGhhdmUgcGVy"\
|
|
||||||
"bWlzc2lvbiB0byBydW4gZG9ja2VyIgogICAgICAgIHJldHVybiAxCiAgICBmaQoKICAgIHJldHVy"\
|
|
||||||
"biAwCn0KCiMgQ2hlY2tzIGlmIGEgY29udGFpbmVyIChhbnkgc3RhdGUpIGV4aXN0cy4gUmV0dXJu"\
|
|
||||||
"cyAxIGlmIG5vdCBmb3VuZC4KX2lzX2NvbnRhaW5lcl9leGlzdHMoKSB7CiAgICBpZiAhIGRvY2tl"\
|
|
||||||
"ciBwcyAtYSAtLWZvcm1hdCAie3suTmFtZXN9fSIgfCBncmVwIC1xICJeJDEkIjsgdGhlbgogICAg"\
|
|
||||||
"ICAgIHJldHVybiAxCiAgICBmaQogICAgcmV0dXJuIDAKfQoKIyBDaGVja3MgaWYgYSBjb250YWlu"\
|
|
||||||
"ZXIgaXMgY3VycmVudGx5IHJ1bm5pbmcuIFJldHVybnMgMSBpZiBub3QgcnVubmluZy4KX2lzX2Nv"\
|
|
||||||
"bnRhaW5lcl9ydW5uaW5nKCkgewogICAgaWYgISBkb2NrZXIgcHMgLS1mb3JtYXQgInt7Lk5hbWVz"\
|
|
||||||
"fX0iIHwgZ3JlcCAtcSAiXiQxJCI7IHRoZW4KICAgICAgICByZXR1cm4gMQogICAgZmkKICAgIHJl"\
|
|
||||||
"dHVybiAwCn0KCiMgUHJpbnRzIHRoZSBJRCBvZiB0aGUgbmFtZWQgY29udGFpbmVyLgpfZ2V0X2Nv"\
|
|
||||||
"bnRhaW5lcl9pZCgpIHsKICAgIGRvY2tlciBwcyAtLWZvcm1hdCAie3suSUR9fSIgLS1maWx0ZXIg"\
|
|
||||||
"Im5hbWU9JDEiCn0KCiMgUHJpbnRzIHRoZSBzdGF0dXMgc3RyaW5nIG9mIHRoZSBuYW1lZCBjb250"\
|
|
||||||
"YWluZXIuCl9nZXRfY29udGFpbmVyX3N0YXR1cygpIHsKICAgIGRvY2tlciBwcyAtLWZvcm1hdCAi"\
|
|
||||||
"e3suU3RhdHVzfX0iIC0tZmlsdGVyICJuYW1lPSQxIgp9CgojIFN0YXJ0cyBhbiBleGlzdGluZywg"\
|
|
||||||
"c3RvcHBlZCBjb250YWluZXIuCl9zdGFydF9jb250YWluZXIoKSB7CiAgICBfaXNfY29udGFpbmVy"\
|
|
||||||
"X2V4aXN0cyAkMSB8fCByZXR1cm4gMQogICAgX2lzX2NvbnRhaW5lcl9ydW5uaW5nICQxICYmIHJl"\
|
|
||||||
"dHVybiAwCiAgICBkb2NrZXIgc3RhcnQgJDEKfQoKIyBTdG9wcyBhIHJ1bm5pbmcgY29udGFpbmVy"\
|
|
||||||
"Lgpfc3RvcF9jb250YWluZXIoKSB7CiAgICBfaXNfY29udGFpbmVyX3J1bm5pbmcgJDEgfHwgcmV0"\
|
|
||||||
"dXJuIDA7CiAgICBkb2NrZXIgc3RvcCAkMQp9ICAgCgojIFN0b3BzIChpZiBuZWVkZWQpIGFuZCBy"\
|
|
||||||
"ZW1vdmVzIGEgY29udGFpbmVyLgpfcmVtb3ZlX2NvbnRhaW5lcigpIHsKICAgIF9zdG9wX2NvbnRh"\
|
|
||||||
"aW5lciAkMQogICAgX2lzX2NvbnRhaW5lcl9leGlzdHMgJDEgfHwgcmV0dXJuIDA7CiAgICBkb2Nr"\
|
|
||||||
"ZXIgcm0gJDEKfQoKIyBQcmludHMgdGhlIGxvZ3MgZm9yIGEgY29udGFpbmVyLgpfZ2V0X2NvbnRh"\
|
|
||||||
"aW5lcl9sb2dzKCkgewogICAgaWYgISBfaXNfY29udGFpbmVyX2V4aXN0cyAkMTsgdGhlbgogICAg"\
|
|
||||||
"ICAgIGVjaG8gIkNvbnRhaW5lciAkMSBkb2VzIG5vdCBleGlzdCIKICAgICAgICByZXR1cm4gMQog"\
|
|
||||||
"ICAgZmkKCiAgICBkb2NrZXIgbG9ncyAkMQp9CgojIENoZWNrcyBpZiBsaXN0ZWQgZW52aXJvbm1l"\
|
|
||||||
"bnQgdmFyaWFibGVzIGFyZSBzZXQ7IGNhbGxzIF9kaWUoKSBpZiBhbnkgYXJlIG1pc3NpbmcuCl9j"\
|
|
||||||
"aGVja19yZXF1aXJlZF9lbnZfdmFycygpIHsKICAgIGxvY2FsIHJlcXVpcmVkX3ZhcnM9KCIkQCIp"\
|
|
||||||
"CiAgICBmb3IgdmFyIGluICIke3JlcXVpcmVkX3ZhcnNbQF19IjsgZG8KICAgICAgICBpZiBbIC16"\
|
|
||||||
"ICIkeyF2YXJ9IiBdOyB0aGVuCiAgICAgICAgICAgIF9kaWUgIlJlcXVpcmVkIGVudmlyb25tZW50"\
|
|
||||||
"IHZhcmlhYmxlICR2YXIgaXMgbm90IHNldCIKICAgICAgICBmaQogICAgZG9uZQp9CgojIFJlbW92"\
|
|
||||||
"ZXMgYSBwYXRoIHVzaW5nIGEgcm9vdCBEb2NrZXIgY29udGFpbmVyIChmb3IgcGVybWlzc2lvbnMp"\
|
|
||||||
"Lgpfcm9vdF9yZW1vdmVfdHJlZSgpIHsKICAgIGxvY2FsIHRvX3JlbW92ZT0iJDEiCiAgICBwYXJl"\
|
|
||||||
"bnQ9JChkaXJuYW1lICIkdG9fcmVtb3ZlIikKICAgIGFic19wYXJlbnQ9JChyZWFscGF0aCAiJHBh"\
|
|
||||||
"cmVudCIpCiAgICBjaGlsZD0kKGJhc2VuYW1lICIkdG9fcmVtb3ZlIikKICAgIGRvY2tlciBydW4g"\
|
|
||||||
"LS1ybSAtdiAiJGFic19wYXJlbnQiOi9kYXRhIGFscGluZSBybSAtcmYgIi9kYXRhLyRjaGlsZCIK"\
|
|
||||||
"fQoKCiMgTG9hZCBhdXRvY29tbWFuZHMKc291cmNlICIke0FHRU5UX1BBVEh9L2RhdGFjb21tYW5k"\
|
|
||||||
"cy5zaCI=";
|
|
||||||
|
|
||||||
// Decode Base64 data
|
|
||||||
size_t decoded_size = (strlen(filedata_base64) * 3) / 4;
|
|
||||||
unsigned char* decoded_data = new unsigned char[decoded_size];
|
|
||||||
size_t actual_size;
|
|
||||||
base64_decode(filedata_base64, strlen(filedata_base64), decoded_data, &actual_size);
|
|
||||||
|
|
||||||
bool file_written = _recreate_file_(outpath, 15958097326741776083ULL, std::filesystem::perms(493), decoded_data, actual_size);
|
|
||||||
delete[] decoded_data;
|
|
||||||
any_written = any_written || file_written;
|
|
||||||
}
|
|
||||||
return any_written;
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,15 +0,0 @@
|
|||||||
|
|
||||||
#pragma once
|
|
||||||
|
|
||||||
/*
|
|
||||||
|
|
||||||
THIS FILE IS AUTO-GENERATED BY DEHYDRATE.
|
|
||||||
DO NOT EDIT THIS FILE.
|
|
||||||
|
|
||||||
*/
|
|
||||||
|
|
||||||
|
|
||||||
#include <string>
|
|
||||||
namespace recreate_agent {
|
|
||||||
bool recreate_tree(std::string destination_folder);
|
|
||||||
}
|
|
@ -1,15 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
// DUMMY VERSION - replaced by build process.
|
|
||||||
|
|
||||||
#include <string>
|
|
||||||
|
|
||||||
namespace dropshell {
|
|
||||||
|
|
||||||
// Version information
|
|
||||||
const std::string VERSION = "DEV";
|
|
||||||
const std::string RELEASE_DATE = "NEVER";
|
|
||||||
const std::string AUTHOR = "j842";
|
|
||||||
const std::string LICENSE = "MIT";
|
|
||||||
|
|
||||||
} // namespace dropshell
|
|
174
source/src/commands/backupdata.cpp
Normal file
174
source/src/commands/backupdata.cpp
Normal file
@ -0,0 +1,174 @@
|
|||||||
|
#include <unistd.h>
|
||||||
|
#include <cstring>
|
||||||
|
#include <iostream>
|
||||||
|
#include <sstream>
|
||||||
|
#include <filesystem>
|
||||||
|
|
||||||
|
#include "utils/output.hpp"
|
||||||
|
#include "utils/assert.hpp"
|
||||||
|
#include "utils/utils.hpp"
|
||||||
|
#include "command_registry.hpp"
|
||||||
|
#include "config.hpp"
|
||||||
|
#include "services.hpp"
|
||||||
|
#include "servers.hpp"
|
||||||
|
#include "server_env_manager.hpp"
|
||||||
|
#include "templates.hpp"
|
||||||
|
#include "utils/directories.hpp"
|
||||||
|
#include "shared_commands.hpp"
|
||||||
|
|
||||||
|
namespace dropshell
|
||||||
|
{
|
||||||
|
|
||||||
|
int backupdata_handler(const CommandContext &ctx);
|
||||||
|
|
||||||
|
static std::vector<std::string> backupdata_name_list = {"backupdata", "bd", "backup", "bup"};
|
||||||
|
|
||||||
|
// Static registration
|
||||||
|
struct BackupDataCommandRegister
|
||||||
|
{
|
||||||
|
BackupDataCommandRegister()
|
||||||
|
{
|
||||||
|
CommandRegistry::instance().register_command({backupdata_name_list,
|
||||||
|
backupdata_handler,
|
||||||
|
shared_commands::std_autocomplete_allowall,
|
||||||
|
false, // hidden
|
||||||
|
true, // requires_config
|
||||||
|
true, // requires_install
|
||||||
|
2, // min_args (after command)
|
||||||
|
2, // max_args (after command)
|
||||||
|
"backupdata SERVER SERVICE",
|
||||||
|
"Backup data for a service on a server.",
|
||||||
|
// heredoc
|
||||||
|
R"(
|
||||||
|
backupdata SERVER SERVICE Backup data for a service on a server.
|
||||||
|
backupdata SERVER all Backup data for all services on a server.
|
||||||
|
|
||||||
|
Note: This command will not create any data or configuration.
|
||||||
|
It will simply backup the data on the remote server, saving it to a local file.
|
||||||
|
Restore the data with restore.
|
||||||
|
)"});
|
||||||
|
}
|
||||||
|
} backupdata_command_register;
|
||||||
|
|
||||||
|
namespace shared_commands
|
||||||
|
{
|
||||||
|
|
||||||
|
bool backupdata_service(const std::string &server, const std::string &service)
|
||||||
|
{
|
||||||
|
server_env_manager server_env(server);
|
||||||
|
if (!server_env.is_valid())
|
||||||
|
{
|
||||||
|
error << "Server " << server << " is not valid" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
LocalServiceInfo sinfo = get_service_info(server, service);
|
||||||
|
if (!SIvalid(sinfo))
|
||||||
|
{
|
||||||
|
error << "Service " << service << " is not valid" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const std::string command = "backup";
|
||||||
|
|
||||||
|
if (!gTemplateManager().template_command_exists(sinfo.template_name, command))
|
||||||
|
{
|
||||||
|
info << service << " has no data to backup" << std::endl;
|
||||||
|
debug << "(no backup script for " << sinfo.template_name << ")" << std::endl;
|
||||||
|
return true; // nothing to back up.
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if basic installed stuff is in place.
|
||||||
|
std::string remote_service_template_path = remotepath::service_template(server, service);
|
||||||
|
std::string remote_command_script_file = remote_service_template_path + "/" + command + ".sh";
|
||||||
|
std::string remote_service_config_path = remotepath::service_config(server, service);
|
||||||
|
if (!server_env.check_remote_items_exist({remotepath::service(server, service),
|
||||||
|
remote_command_script_file,
|
||||||
|
remotefile::service_env(server, service)}))
|
||||||
|
{
|
||||||
|
error << "Error: Required service directories not found on remote server" << std::endl;
|
||||||
|
info << "Is the service installed?" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create backups directory on server if it doesn't exist
|
||||||
|
std::string remote_backups_dir = remotepath::backups(server);
|
||||||
|
debug << "Remote backups directory on " << server << ": " << remote_backups_dir << std::endl;
|
||||||
|
std::string mkdir_cmd = "mkdir -p " + quote(remote_backups_dir);
|
||||||
|
if (!execute_ssh_command(server_env.get_SSH_INFO(), sCommand("", mkdir_cmd, {}), cMode::Defaults))
|
||||||
|
{
|
||||||
|
error << "Failed to create backups directory on server" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create backups directory locally if it doesn't exist
|
||||||
|
std::string local_backups_dir = gConfig().get_local_backup_path();
|
||||||
|
if (local_backups_dir.empty())
|
||||||
|
{
|
||||||
|
error << "Error: Local backups directory not found" << std::endl;
|
||||||
|
info << "Run 'dropshell edit' to configure DropShell" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (!std::filesystem::exists(local_backups_dir))
|
||||||
|
std::filesystem::create_directories(local_backups_dir);
|
||||||
|
|
||||||
|
// Get current datetime for backup filename
|
||||||
|
shared_commands::cBackupFileName backup_filename_construction(server, service, sinfo.template_name);
|
||||||
|
if (!backup_filename_construction.is_valid())
|
||||||
|
{
|
||||||
|
error << "Invalid backup filename" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Construct backup filename
|
||||||
|
std::string backup_filename = backup_filename_construction.get_filename();
|
||||||
|
std::string remote_backup_file_path = remote_backups_dir + "/" + backup_filename;
|
||||||
|
std::string local_backup_file_path = (std::filesystem::path(local_backups_dir) / backup_filename).string();
|
||||||
|
|
||||||
|
// assert that the backup filename is valid - -_- appears exactly 3 times in local_backup_file_path.
|
||||||
|
ASSERT(3 == count_substring(magic_string(), local_backup_file_path), "Invalid backup filename");
|
||||||
|
|
||||||
|
{ // Run backup script
|
||||||
|
shared_commands::cRemoteTempFolder remote_temp_folder(server_env);
|
||||||
|
if (!server_env.run_remote_template_command(service, command, {}, false, {{"BACKUP_FILE", remote_backup_file_path}, {"TEMP_DIR", remote_temp_folder.path()}}))
|
||||||
|
{
|
||||||
|
error << "Backup script failed on remote server: " << remote_backup_file_path << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Copy backup file from server to local
|
||||||
|
if (!shared_commands::scp_file_from_remote(server_env, remote_backup_file_path, local_backup_file_path, false))
|
||||||
|
{
|
||||||
|
error << "Failed to copy backup file from server" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
} // dtor of remote_temp_folder will clean up the temp folder on the server
|
||||||
|
|
||||||
|
info << "Backup created successfully. Restore with:" << std::endl;
|
||||||
|
info << " dropshell restore " << server << " " << service << " " << backup_filename << std::endl;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
} // namespace shared_commands
|
||||||
|
|
||||||
|
int backupdata_handler(const CommandContext &ctx)
|
||||||
|
{
|
||||||
|
ASSERT(ctx.args.size() == 2, "Invalid number of arguments");
|
||||||
|
|
||||||
|
std::string server = safearg(ctx.args, 0);
|
||||||
|
std::string service = safearg(ctx.args, 1);
|
||||||
|
|
||||||
|
if (service == "all")
|
||||||
|
{
|
||||||
|
// backup all services on the server
|
||||||
|
maketitle("Backing up data for all services on " + server);
|
||||||
|
bool okay = true;
|
||||||
|
std::vector<LocalServiceInfo> services = get_server_services_info(server);
|
||||||
|
for (const LocalServiceInfo &si : services)
|
||||||
|
okay &= shared_commands::backupdata_service(server, si.service_name);
|
||||||
|
return okay ? 0 : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return shared_commands::backupdata_service(server, service) ? 0 : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace dropshell
|
44
source/src/commands/colours.cpp
Normal file
44
source/src/commands/colours.cpp
Normal file
@ -0,0 +1,44 @@
|
|||||||
|
#include "command_registry.hpp"
|
||||||
|
#include "config.hpp"
|
||||||
|
#include "utils/output.hpp"
|
||||||
|
#include "utils/assert.hpp"
|
||||||
|
|
||||||
|
namespace dropshell {
|
||||||
|
|
||||||
|
void colours_autocomplete(const CommandContext& ctx) {}
|
||||||
|
int colours_handler(const CommandContext& ctx)
|
||||||
|
{
|
||||||
|
info << "Colours:" << std::endl;
|
||||||
|
debug << "Debug Example: The quick brown fox jumps over the lazy dog." << std::endl;
|
||||||
|
info << "Info Example: The quick brown fox jumps over the lazy dog." << std::endl;
|
||||||
|
warning << "Warning Example: The quick brown fox jumps over the lazy dog." << std::endl;
|
||||||
|
error << "Error Example: The quick brown fox jumps over the lazy dog." << std::endl;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static std::vector<std::string> colours_name_list={"colours","c","--colours","-c"};
|
||||||
|
|
||||||
|
// Static registration
|
||||||
|
struct ColoursCommandRegister {
|
||||||
|
ColoursCommandRegister() {
|
||||||
|
CommandRegistry::instance().register_command({
|
||||||
|
colours_name_list,
|
||||||
|
colours_handler,
|
||||||
|
colours_autocomplete,
|
||||||
|
true, // hidden
|
||||||
|
false, // requires_config
|
||||||
|
false, // requires_install
|
||||||
|
0, // min_args (after command)
|
||||||
|
0, // max_args (after command)
|
||||||
|
"colours",
|
||||||
|
"Show the colours used by dropshell.",
|
||||||
|
// heredoc
|
||||||
|
R"(
|
||||||
|
Show the colours used by dropshell.
|
||||||
|
)"
|
||||||
|
});
|
||||||
|
}
|
||||||
|
} colours_command_register;
|
||||||
|
|
||||||
|
|
||||||
|
} // namespace dropshell
|
@ -11,15 +11,14 @@ namespace dropshell
|
|||||||
{
|
{
|
||||||
|
|
||||||
int create_service_handler(const CommandContext &ctx);
|
int create_service_handler(const CommandContext &ctx);
|
||||||
bool create_service(const std::string &server_name, const std::string &template_name, const std::string &service_name, bool silent);
|
|
||||||
void create_service_autocomplete(const CommandContext &ctx);
|
void create_service_autocomplete(const CommandContext &ctx);
|
||||||
|
|
||||||
static std::vector<std::string> create_service_name_list = {"create-service"};
|
static std::vector<std::string> create_service_name_list = {"create-service"};
|
||||||
|
|
||||||
// Static registration
|
// Static registration
|
||||||
struct UninstallCommandRegister
|
struct CreateServiceCommandRegister
|
||||||
{
|
{
|
||||||
UninstallCommandRegister()
|
CreateServiceCommandRegister()
|
||||||
{
|
{
|
||||||
CommandRegistry::instance().register_command({create_service_name_list,
|
CommandRegistry::instance().register_command({create_service_name_list,
|
||||||
create_service_handler,
|
create_service_handler,
|
||||||
@ -45,7 +44,7 @@ namespace dropshell
|
|||||||
std::string service = safearg(ctx.args, 1);
|
std::string service = safearg(ctx.args, 1);
|
||||||
std::string template_name = safearg(ctx.args, 2);
|
std::string template_name = safearg(ctx.args, 2);
|
||||||
|
|
||||||
return create_service(server, template_name, service, false) ? 0 : 1;
|
return shared_commands::create_service(server, template_name, service) ? 0 : 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
void create_service_autocomplete(const CommandContext &ctx)
|
void create_service_autocomplete(const CommandContext &ctx)
|
||||||
@ -58,12 +57,15 @@ namespace dropshell
|
|||||||
{
|
{
|
||||||
std::set<std::string> templates = gTemplateManager().get_template_list();
|
std::set<std::string> templates = gTemplateManager().get_template_list();
|
||||||
for (const auto &template_name : templates)
|
for (const auto &template_name : templates)
|
||||||
std::cout << template_name << std::endl;
|
rawout << template_name << std::endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool create_service(const std::string &server_name, const std::string &template_name, const std::string &service_name, bool silent)
|
namespace shared_commands
|
||||||
|
{
|
||||||
|
|
||||||
|
bool create_service(const std::string &server_name, const std::string &template_name, const std::string &service_name)
|
||||||
{
|
{
|
||||||
if (server_name.empty() || template_name.empty() || service_name.empty())
|
if (server_name.empty() || template_name.empty() || service_name.empty())
|
||||||
return false;
|
return false;
|
||||||
@ -72,44 +74,34 @@ namespace dropshell
|
|||||||
|
|
||||||
if (service_dir.empty())
|
if (service_dir.empty())
|
||||||
{
|
{
|
||||||
if (!silent)
|
error << "Couldn't locate server " << server_name << " in any config directory" << std::endl;
|
||||||
{
|
info << "Please check the server name is correct and try again" << std::endl;
|
||||||
std::cerr << "Error: Couldn't locate server " << server_name << " in any config directory" << std::endl;
|
info << "You can list all servers with 'dropshell servers'" << std::endl;
|
||||||
std::cerr << "Please check the server name is correct and try again" << std::endl;
|
info << "You can create a new server with 'dropshell create-server " << server_name << "'" << std::endl;
|
||||||
std::cerr << "You can list all servers with 'dropshell servers'" << std::endl;
|
|
||||||
std::cerr << "You can create a new server with 'dropshell create-server " << server_name << "'" << std::endl;
|
|
||||||
}
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (std::filesystem::exists(service_dir))
|
if (std::filesystem::exists(service_dir))
|
||||||
{
|
{
|
||||||
if (!silent)
|
error << "Service already exists: " << service_name << std::endl;
|
||||||
{
|
debug << "Current service path: " << service_dir << std::endl;
|
||||||
std::cerr << "Error: Service already exists: " << service_name << std::endl;
|
|
||||||
std::cerr << "Current service path: " << service_dir << std::endl;
|
|
||||||
}
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
template_info tinfo = gTemplateManager().get_template_info(template_name);
|
template_info tinfo = gTemplateManager().get_template_info(template_name);
|
||||||
if (!tinfo.is_set())
|
if (!tinfo.is_set())
|
||||||
{
|
{
|
||||||
if (!silent)
|
error << "Template '" << template_name << "' not found" << std::endl;
|
||||||
{
|
info << "Please check the template name is correct and try again" << std::endl;
|
||||||
std::cerr << "Error: Template '" << template_name << "' not found" << std::endl;
|
info << "You can list all templates with 'dropshell templates'" << std::endl;
|
||||||
std::cerr << "Please check the template name is correct and try again" << std::endl;
|
info << "You can create a new template with 'dropshell create-template " << template_name << "'" << std::endl;
|
||||||
std::cerr << "You can list all templates with 'dropshell templates'" << std::endl;
|
|
||||||
std::cerr << "You can create a new template with 'dropshell create-template " << template_name << "'" << std::endl;
|
|
||||||
}
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
// check template is all good.
|
// check template is all good.
|
||||||
if (!gTemplateManager().test_template(tinfo.local_template_path()))
|
if (!gTemplateManager().test_template(tinfo.local_template_path()))
|
||||||
{
|
{
|
||||||
if (!silent)
|
error << "Template '" << template_name << "' is not valid" << std::endl;
|
||||||
std::cerr << "Error: Template '" << template_name << "' is not valid" << std::endl;
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -119,15 +111,14 @@ namespace dropshell
|
|||||||
// copy the template config files to the service directory
|
// copy the template config files to the service directory
|
||||||
recursive_copy(tinfo.local_template_path() / "config", service_dir);
|
recursive_copy(tinfo.local_template_path() / "config", service_dir);
|
||||||
|
|
||||||
if (!silent)
|
info << "Service " << service_name << " created successfully" << std::endl;
|
||||||
{
|
info << std::endl;
|
||||||
std::cout << "Service " << service_name << " created successfully" << std::endl;
|
info << "To complete the installation, please:" << std::endl;
|
||||||
std::cout << std::endl;
|
info << "1. edit the service config file: dropshell edit " << server_name << " " << service_name << std::endl;
|
||||||
std::cout << "To complete the installation, please:" << std::endl;
|
info << "2. install the remote service: dropshell install " << server_name << " " << service_name << std::endl;
|
||||||
std::cout << "1. edit the service config file: dropshell edit " << server_name << " " << service_name << std::endl;
|
|
||||||
std::cout << "2. install the remote service: dropshell install " << server_name << " " << service_name << std::endl;
|
|
||||||
}
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
} // namespace shared_commands
|
||||||
|
|
||||||
} // namespace dropshell
|
} // namespace dropshell
|
@ -66,16 +66,16 @@ bool edit_file(const std::string &file_path, bool has_bb64)
|
|||||||
// Check if stdin is connected to a terminal if EDITOR is not set
|
// Check if stdin is connected to a terminal if EDITOR is not set
|
||||||
editor_cmd = "nano -w " + quote(file_path);
|
editor_cmd = "nano -w " + quote(file_path);
|
||||||
} else {
|
} else {
|
||||||
std::cerr << "Error: Standard input is not a terminal and EDITOR environment variable is not set." << std::endl;
|
error << "Standard input is not a terminal and EDITOR environment variable is not set." << std::endl;
|
||||||
std::cerr << "Try setting the EDITOR environment variable (e.g., export EDITOR=nano) or run in an interactive terminal." << std::endl;
|
info << "Try setting the EDITOR environment variable (e.g., export EDITOR=nano) or run in an interactive terminal." << std::endl;
|
||||||
std::cerr << "You can manually edit the file at: " << file_path << std::endl;
|
info << "You can manually edit the file at: " << file_path << std::endl;
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::cout << "Editing file: " << file_path << std::endl;
|
info << "Editing file: " << file_path << std::endl;
|
||||||
|
|
||||||
if (has_bb64) {
|
if (has_bb64) {
|
||||||
return execute_local_command(editor_cmd, nullptr, cMode::Interactive);
|
return execute_local_command("", editor_cmd, {}, nullptr, cMode::Interactive);
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
// might not have bb64 at this early stage. Direct edit.
|
// might not have bb64 at this early stage. Direct edit.
|
||||||
@ -119,14 +119,14 @@ int edit_server(const std::string &server_name)
|
|||||||
std::string config_file = localfile::server_json(server_name);
|
std::string config_file = localfile::server_json(server_name);
|
||||||
|
|
||||||
if (!edit_file(config_file, true)) {
|
if (!edit_file(config_file, true)) {
|
||||||
std::cerr << "Error: Failed to edit server config" << std::endl;
|
error << "Failed to edit server config" << std::endl;
|
||||||
std::cerr << "You can manually edit this file at: " << config_file << std::endl;
|
info << "You can manually edit this file at: " << config_file << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::cout << "If you have changed DROPSHELL_DIR, you should manually move the files to the new location NOW." << std::endl
|
info << "If you have changed DROPSHELL_DIR, you should manually move the files to the new location NOW." << std::endl;
|
||||||
<< "You can ssh in to the remote server with: dropshell ssh "<<server_name<< std::endl
|
info << "You can ssh in to the remote server with: dropshell ssh "<<server_name<< std::endl;
|
||||||
<< "Once moved, reinstall all services with: dropshell install " << server_name << std::endl;
|
info << "Once moved, reinstall all services with: dropshell install " << server_name << std::endl;
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -138,12 +138,12 @@ int edit_service_config(const std::string &server, const std::string &service)
|
|||||||
std::string config_file = localfile::service_env(server, service);
|
std::string config_file = localfile::service_env(server, service);
|
||||||
if (!std::filesystem::exists(config_file))
|
if (!std::filesystem::exists(config_file))
|
||||||
{
|
{
|
||||||
std::cerr << "Error: Service config file not found: " << config_file << std::endl;
|
error << "Service config file not found: " << config_file << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (edit_file(config_file, true) && std::filesystem::exists(config_file))
|
if (edit_file(config_file, true) && std::filesystem::exists(config_file))
|
||||||
std::cout << "To apply your changes, run:\n dropshell install " + server + " " + service << std::endl;
|
info << "To apply your changes, run:\n dropshell install " + server + " " + service << std::endl;
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -167,7 +167,7 @@ int edit_handler(const CommandContext& ctx) {
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::cout << "Edit handler called with " << ctx.args.size() << " args\n";
|
info << "Edit handler called with " << ctx.args.size() << " args\n";
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -45,7 +45,7 @@ namespace dropshell
|
|||||||
{
|
{
|
||||||
if (ctx.args.size() < 1)
|
if (ctx.args.size() < 1)
|
||||||
{
|
{
|
||||||
std::cerr << "Error: Server name is required" << std::endl;
|
error << "Server name is required" << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -48,7 +48,7 @@ void help_autocomplete(const CommandContext& ctx) {
|
|||||||
if (ctx.args.size() == 1) {
|
if (ctx.args.size() == 1) {
|
||||||
// list all commands
|
// list all commands
|
||||||
for (const auto& cmd : CommandRegistry::instance().list_primary_commands(false)) {
|
for (const auto& cmd : CommandRegistry::instance().list_primary_commands(false)) {
|
||||||
std::cout << cmd << std::endl;
|
rawout << cmd << std::endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return;
|
return;
|
||||||
@ -57,11 +57,14 @@ void help_autocomplete(const CommandContext& ctx) {
|
|||||||
void show_command(const std::string& cmd) {
|
void show_command(const std::string& cmd) {
|
||||||
const auto& cmd_info = CommandRegistry::instance().find_command(cmd);
|
const auto& cmd_info = CommandRegistry::instance().find_command(cmd);
|
||||||
if (!cmd_info)
|
if (!cmd_info)
|
||||||
std::cout << "Unknown command: " << cmd << std::endl;
|
{
|
||||||
|
error << "Unknown command: " << cmd << std::endl;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
std::cout << " ";
|
info << " ";
|
||||||
print_left_aligned(cmd_info->help_usage, 30);
|
info << left_align(cmd_info->help_usage, 32);
|
||||||
std::cout << cmd_info->help_description << std::endl;
|
info << cmd_info->help_description << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
extern const std::string VERSION;
|
extern const std::string VERSION;
|
||||||
@ -74,28 +77,26 @@ int show_command_help(const std::string& cmd) {
|
|||||||
const auto& cmd_info = CommandRegistry::instance().find_command(cmd);
|
const auto& cmd_info = CommandRegistry::instance().find_command(cmd);
|
||||||
if (!cmd_info)
|
if (!cmd_info)
|
||||||
{
|
{
|
||||||
std::cout << "Unknown command: " << cmd << std::endl;
|
error << "Unknown command: " << cmd << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
info << "Command " << cmd << " usage:" << std::endl;
|
||||||
|
info << " ";
|
||||||
|
info << left_align(cmd_info->help_usage, 32);
|
||||||
|
info << cmd_info->help_description << std::endl;
|
||||||
|
|
||||||
std::cout << std::endl;
|
info << std::endl;
|
||||||
std::cout << "Usage: " << std::endl;
|
|
||||||
std::cout << " ";
|
|
||||||
print_left_aligned(cmd_info->help_usage, 30);
|
|
||||||
std::cout << cmd_info->help_description << std::endl;
|
|
||||||
|
|
||||||
std::cout << std::endl;
|
info << " Equivalent names: ";
|
||||||
|
|
||||||
std::cout << " Equivalent names: ";
|
|
||||||
bool first = true;
|
bool first = true;
|
||||||
for (const auto& name : cmd_info->names) {
|
for (const auto& name : cmd_info->names) {
|
||||||
if (!first) std::cout << ", ";
|
if (!first) info << ", ";
|
||||||
std::cout << name;
|
info << name;
|
||||||
first = false;
|
first = false;
|
||||||
}
|
}
|
||||||
std::cout << std::endl << std::endl;
|
info << std::endl;
|
||||||
|
|
||||||
std::cout << cmd_info->full_help << std::endl << std::endl;
|
info << cmd_info->full_help << std::endl << std::endl;
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
@ -105,12 +106,12 @@ int help_handler(const CommandContext& ctx) {
|
|||||||
if (ctx.args.size() > 0)
|
if (ctx.args.size() > 0)
|
||||||
return show_command_help(ctx.args[0]);
|
return show_command_help(ctx.args[0]);
|
||||||
|
|
||||||
std::cout << std::endl;
|
info << std::endl;
|
||||||
maketitle("DropShell version " + VERSION);
|
maketitle("DropShell version " + VERSION);
|
||||||
std::cout << std::endl;
|
info << std::endl;
|
||||||
std::cout << "A tool for managing remote servers, by " << AUTHOR << std::endl;
|
info << "A tool for managing remote servers, by " << AUTHOR << std::endl;
|
||||||
std::cout << std::endl;
|
info << std::endl;
|
||||||
std::cout << "dropshell ..." << std::endl;
|
info << "dropshell ..." << std::endl;
|
||||||
|
|
||||||
show_command("help");
|
show_command("help");
|
||||||
show_command("edit");
|
show_command("edit");
|
||||||
@ -119,60 +120,20 @@ int help_handler(const CommandContext& ctx) {
|
|||||||
{
|
{
|
||||||
// show more!
|
// show more!
|
||||||
show_command("list");
|
show_command("list");
|
||||||
|
std::cout << std::endl;
|
||||||
|
show_command("install");
|
||||||
|
show_command("uninstall");
|
||||||
|
show_command("nuke");
|
||||||
|
std::cout << std::endl;
|
||||||
|
show_command("start");
|
||||||
|
show_command("stop");
|
||||||
|
std::cout << std::endl;
|
||||||
|
show_command("ssh");
|
||||||
|
std::cout << std::endl;
|
||||||
}
|
}
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
// void show_command(const std::string& cmd) {
|
|
||||||
// const auto& cmd_info = CommandRegistry::instance().find_command(cmd);
|
|
||||||
// if (cmd_info) {
|
|
||||||
// std::cout << " " << cmd_info->help_usage
|
|
||||||
// << std::string(' ', std::min(1,(int)(30-cmd_info->help_usage.length())))
|
|
||||||
// << cmd_info->help_description << std::endl;
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
|
|
||||||
// bool print_help() {
|
|
||||||
// std::cout << std::endl;
|
|
||||||
// maketitle("DropShell version " + VERSION);
|
|
||||||
// std::cout << std::endl;
|
|
||||||
// std::cout << "A tool for managing server configurations" << std::endl;
|
|
||||||
// std::cout << std::endl;
|
|
||||||
// std::cout << "dropshell ..." << std::endl;
|
|
||||||
// show_command("help");
|
|
||||||
// show_command("edit");
|
|
||||||
|
|
||||||
// if (gConfig().is_config_set()) {
|
|
||||||
// std::cout << " templates List all available templates" << std::endl;
|
|
||||||
// std::cout << std::endl;
|
|
||||||
// std::cout << std::endl;
|
|
||||||
// std::cout << "Service commands: (if no service is specified, all services for the server are affected)" << std::endl;
|
|
||||||
// std::cout << " list [SERVER] [SERVICE] List status/details of all servers/server/service." << std::endl;
|
|
||||||
// std::cout << " edit [SERVER] [SERVICE] Edit the configuration of dropshell/server/service." << std::endl;
|
|
||||||
// std::cout << std::endl;
|
|
||||||
// std::cout << " install SERVER [SERVICE] Install/reinstall/update service(s). Safe/non-destructive." << std::endl;
|
|
||||||
// std::cout << " uninstall SERVER [SERVICE] Uninstalls the service on the remote server. Leaves data intact." << std::endl;
|
|
||||||
// std::cout << " nuke SERVER SERVICE Nuke the service, deleting ALL local and remote data." << std::endl;
|
|
||||||
// std::cout << std::endl;
|
|
||||||
// std::cout << " COMMAND SERVER [SERVICE] Run a command on service(s), e.g." << std::endl;
|
|
||||||
// std::cout << " backup, restore, start, stop, logs" << std::endl;
|
|
||||||
// std::cout << std::endl;
|
|
||||||
// std::cout << " ssh SERVER SERVICE Launch an interactive shell on a server or service" << std::endl;
|
|
||||||
// std::cout << std::endl;
|
|
||||||
// std::cout << "Creation commands: (apply to the first local config directory)"<<std::endl;
|
|
||||||
// std::cout << " create-template TEMPLATE" << std::endl;
|
|
||||||
// std::cout << " create-server SERVER" << std::endl;
|
|
||||||
// std::cout << " create-service SERVER TEMPLATE SERVICE" << std::endl;
|
|
||||||
// }
|
|
||||||
// else {
|
|
||||||
// show_command("help");
|
|
||||||
// show_command("edit");
|
|
||||||
// std::cout << std::endl;
|
|
||||||
// std::cout << "Other commands available once initialised." << std::endl;
|
|
||||||
// }
|
|
||||||
// return true;
|
|
||||||
// }
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
} // namespace dropshell
|
} // namespace dropshell
|
||||||
|
@ -5,8 +5,10 @@
|
|||||||
#include "templates.hpp"
|
#include "templates.hpp"
|
||||||
#include "shared_commands.hpp"
|
#include "shared_commands.hpp"
|
||||||
#include "utils/hash.hpp"
|
#include "utils/hash.hpp"
|
||||||
#include "autogen/_agent.hpp"
|
#include "autogen/_agent-local.hpp"
|
||||||
|
#include "autogen/_agent-remote.hpp"
|
||||||
#include "services.hpp"
|
#include "services.hpp"
|
||||||
|
#include "utils/output.hpp"
|
||||||
|
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
@ -14,6 +16,7 @@
|
|||||||
#include <sstream>
|
#include <sstream>
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
#include "utils/assert.hpp"
|
#include "utils/assert.hpp"
|
||||||
|
#include "servers.hpp"
|
||||||
|
|
||||||
namespace dropshell
|
namespace dropshell
|
||||||
{
|
{
|
||||||
@ -35,15 +38,15 @@ namespace dropshell
|
|||||||
false, // requires_install
|
false, // requires_install
|
||||||
0, // min_args (after command)
|
0, // min_args (after command)
|
||||||
2, // max_args (after command)
|
2, // max_args (after command)
|
||||||
"install SERVER [SERVICE|all]",
|
"install [SERVER] [SERVICE|all]",
|
||||||
"Install/reinstall service(s). Safe/non-destructive way to update.",
|
"Install/reinstall host, remote servers, or service(s). Safe/non-destructive way to update.",
|
||||||
// heredoc
|
// heredoc
|
||||||
R"(
|
R"(
|
||||||
Install components on a server. This is safe to re-run (non-destructive) and used to update
|
Install components on a server. This is safe to re-run (non-destructive) and used to update
|
||||||
servers and their services.
|
servers and their services.
|
||||||
|
|
||||||
install (re)install dropshell components on this computer.
|
install (re)install dropshell components on this computer, and on all servers.
|
||||||
install SERVER (re)install dropshell agent on the given server.
|
install SERVER (re)install dropshell agent on the particular given server.
|
||||||
install SERVER [SERVICE|all] (re)install the given service (or all services) on the given server.
|
install SERVER [SERVICE|all] (re)install the given service (or all services) on the given server.
|
||||||
|
|
||||||
Note you need to create the service first with:
|
Note you need to create the service first with:
|
||||||
@ -52,11 +55,13 @@ namespace dropshell
|
|||||||
}
|
}
|
||||||
} install_command_register;
|
} install_command_register;
|
||||||
|
|
||||||
|
namespace shared_commands
|
||||||
|
{
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
// install service over ssh : SHARED COMMAND
|
// install service over ssh : SHARED COMMAND
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
bool install_service(const std::string &server, const std::string &service, bool silent)
|
bool install_service(const std::string &server, const std::string &service)
|
||||||
{
|
{
|
||||||
LocalServiceInfo service_info = get_service_info(server, service);
|
LocalServiceInfo service_info = get_service_info(server, service);
|
||||||
if (!SIvalid(service_info))
|
if (!SIvalid(service_info))
|
||||||
@ -100,20 +105,20 @@ namespace dropshell
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Copy template files
|
// Copy template files
|
||||||
std::cout << "Copying: [LOCAL] " << tinfo.local_template_path() << std::endl
|
debug << "Copying: [LOCAL] " << tinfo.local_template_path() << std::endl
|
||||||
<< std::string(8, ' ') << "[REMOTE] " << remotepath::service_template(server, service) << "/" << std::endl;
|
<< std::string(8, ' ') << "[REMOTE] " << remotepath::service_template(server, service) << "/" << std::endl;
|
||||||
if (!shared_commands::rsync_tree_to_remote(tinfo.local_template_path().string(), remotepath::service_template(server, service),
|
if (!shared_commands::rsync_tree_to_remote(tinfo.local_template_path().string(), remotepath::service_template(server, service),
|
||||||
server_env, silent))
|
server_env, false))
|
||||||
{
|
{
|
||||||
std::cerr << "Failed to copy template files using rsync" << std::endl;
|
std::cerr << "Failed to copy template files using rsync" << std::endl;
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Copy service files
|
// Copy service files
|
||||||
std::cout << "Copying: [LOCAL] " << localpath::service(server, service) << std::endl
|
debug << "Copying: [LOCAL] " << localpath::service(server, service) << std::endl
|
||||||
<< std::string(8, ' ') << "[REMOTE] " << remotepath::service_config(server, service) << std::endl;
|
<< std::string(8, ' ') << "[REMOTE] " << remotepath::service_config(server, service) << std::endl;
|
||||||
if (!shared_commands::rsync_tree_to_remote(localpath::service(server, service), remotepath::service_config(server, service),
|
if (!shared_commands::rsync_tree_to_remote(localpath::service(server, service), remotepath::service_config(server, service),
|
||||||
server_env, silent))
|
server_env, false))
|
||||||
{
|
{
|
||||||
std::cerr << "Failed to copy service files using rsync" << std::endl;
|
std::cerr << "Failed to copy service files using rsync" << std::endl;
|
||||||
return false;
|
return false;
|
||||||
@ -121,14 +126,16 @@ namespace dropshell
|
|||||||
|
|
||||||
// Run install script
|
// Run install script
|
||||||
{
|
{
|
||||||
server_env.run_remote_template_command(service, "install", {}, silent, {});
|
info << "Running " << service_info.template_name << " install script on " << server << "..." << std::endl;
|
||||||
|
server_env.run_remote_template_command(service, "install", {}, false, {});
|
||||||
}
|
}
|
||||||
|
|
||||||
// print health tick
|
// print health tick
|
||||||
std::cout << "Health: " << shared_commands::healthtick(server, service) << std::endl;
|
info << "Health: " << shared_commands::healthtick(server, service) << std::endl;
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
} // namespace shared_commands
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
// update_dropshell
|
// update_dropshell
|
||||||
@ -154,6 +161,8 @@ namespace dropshell
|
|||||||
|
|
||||||
int update_dropshell()
|
int update_dropshell()
|
||||||
{
|
{
|
||||||
|
maketitle("Updating dropshell on this computer...");
|
||||||
|
|
||||||
// determine path to this executable
|
// determine path to this executable
|
||||||
std::filesystem::path dropshell_path = std::filesystem::canonical("/proc/self/exe");
|
std::filesystem::path dropshell_path = std::filesystem::canonical("/proc/self/exe");
|
||||||
std::filesystem::path parent_path = dropshell_path.parent_path();
|
std::filesystem::path parent_path = dropshell_path.parent_path();
|
||||||
@ -197,8 +206,8 @@ namespace dropshell
|
|||||||
|
|
||||||
if (currentver >= newver)
|
if (currentver >= newver)
|
||||||
{
|
{
|
||||||
std::cout << "Current dropshell version: " << currentver << ", published version: " << newver << std::endl;
|
info << "Current dropshell version: " << currentver << ", published version: " << newver << std::endl;
|
||||||
std::cout << "No update needed." << std::endl;
|
info << "Release version is not newer, no update needed." << std::endl;
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -209,20 +218,22 @@ namespace dropshell
|
|||||||
rval = system(bash_script_2.c_str());
|
rval = system(bash_script_2.c_str());
|
||||||
if (rval != 0)
|
if (rval != 0)
|
||||||
{
|
{
|
||||||
std::cerr << "Failed to install new version of dropshell." << std::endl;
|
error << "Failed to install new version of dropshell." << std::endl;
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::cout << "Successfully updated " << dropshell_path << " to the latest " << arch << " version." << std::endl;
|
info << "Successfully updated " << dropshell_path << " to the latest " << arch << " version." << std::endl;
|
||||||
|
|
||||||
// execute the new version
|
// execute the new version
|
||||||
execlp("bash", "bash", "-c", (parent_path / "dropshell").c_str(), "install", (char *)nullptr);
|
execlp("bash", "bash", "-c", (parent_path / "dropshell").c_str(), "install", (char *)nullptr);
|
||||||
std::cerr << "Failed to execute new version of dropshell." << std::endl;
|
error << "Failed to execute new version of dropshell." << std::endl;
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int install_local_agent()
|
int install_local_agent()
|
||||||
{
|
{
|
||||||
|
maketitle("Installing dropshell agent on this computer...");
|
||||||
|
|
||||||
std::vector<std::filesystem::path> paths = {
|
std::vector<std::filesystem::path> paths = {
|
||||||
gConfig().get_local_template_cache_path(),
|
gConfig().get_local_template_cache_path(),
|
||||||
gConfig().get_local_backup_path(),
|
gConfig().get_local_backup_path(),
|
||||||
@ -234,32 +245,64 @@ namespace dropshell
|
|||||||
for (auto &p : paths)
|
for (auto &p : paths)
|
||||||
if (!std::filesystem::exists(p))
|
if (!std::filesystem::exists(p))
|
||||||
{
|
{
|
||||||
std::cout << "Creating directory: " << p << std::endl;
|
info << "Creating directory: " << p << std::endl;
|
||||||
std::filesystem::create_directories(p);
|
std::filesystem::create_directories(p);
|
||||||
}
|
}
|
||||||
|
|
||||||
// download bb64 for the host architecture.
|
// create the agent-local directory.
|
||||||
if (!std::filesystem::exists(localpath::agent() + "bb64"))
|
recreate_agent_local::recreate_tree(localpath::agent());
|
||||||
{
|
|
||||||
std::string cmd = "cd " + localpath::agent() + " && curl -fsSL -o bb64 https://gitea.jde.nz/public/bb64/releases/download/latest/bb64.amd64 && chmod a+x bb64";
|
|
||||||
int ret = system(cmd.c_str());
|
|
||||||
if (EXITSTATUSCHECK(ret))
|
|
||||||
std::cout << "Downloaded local bb64 to " << localpath::agent() << std::endl;
|
|
||||||
else
|
|
||||||
std::cerr << "Failed to download local bb64 to " << localpath::agent() << std::endl;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
std::cout << "Updating local bb64..." << std::endl;
|
|
||||||
system((localpath::agent() + "bb64 -u").c_str()); // update.
|
|
||||||
}
|
|
||||||
|
|
||||||
std::cout << "Creating local files to copy to remote agents..." << std::endl;
|
// run the local agent installer.
|
||||||
recreate_agent::recreate_tree(localpath::files_for_remote_agent());
|
execute_local_command(localpath::agent(), "agent-install.sh",{}, nullptr, cMode::Defaults | cMode::NoBB64);
|
||||||
|
|
||||||
|
// create the agent-remote directory.
|
||||||
|
info << "Creating local files to copy to remote agents..." << std::endl;
|
||||||
|
recreate_agent_remote::recreate_tree(localpath::files_for_remote_agent());
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int install_server(const std::string &server)
|
||||||
|
{
|
||||||
|
// install the dropshell agent on the given server.
|
||||||
|
maketitle("Installing dropshell agent on " + server, sColour::INFO);
|
||||||
|
|
||||||
|
std::string agent_path = remotepath::agent(server);
|
||||||
|
if (agent_path.empty())
|
||||||
|
{
|
||||||
|
error << "Failed to get agent path for " << server << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
server_env_manager server_env(server);
|
||||||
|
if (!server_env.is_valid())
|
||||||
|
{
|
||||||
|
error << "Invalid server environment for " << server << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// now create the agent.
|
||||||
|
// copy across from the local agent files.
|
||||||
|
info << "Copying local agent files to remote server... " << std::flush;
|
||||||
|
shared_commands::rsync_tree_to_remote(localpath::files_for_remote_agent(), agent_path, server_env, false);
|
||||||
|
info << "done." << std::endl;
|
||||||
|
|
||||||
|
// run the agent installer. Can't use BB64 yet, as we're installing it on the remote server.
|
||||||
|
|
||||||
|
bool okay = execute_ssh_command(server_env.get_SSH_INFO(), sCommand(agent_path, "agent-install.sh",{}), cMode::Defaults | cMode::NoBB64, nullptr);
|
||||||
|
if (!okay)
|
||||||
|
{
|
||||||
|
error << "ERROR: Failed to install remote agent on " << server << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
info << "Installation on " << server << " complete." << std::endl;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------------------------------------------------------------------------
|
||||||
|
// install_host
|
||||||
|
// ------------------------------------------------------------------------------------------------
|
||||||
int install_host()
|
int install_host()
|
||||||
{
|
{
|
||||||
// update dropshell.
|
// update dropshell.
|
||||||
@ -273,60 +316,19 @@ namespace dropshell
|
|||||||
if (rval != 0)
|
if (rval != 0)
|
||||||
return rval;
|
return rval;
|
||||||
|
|
||||||
|
// install the dropshell agent on all servers.
|
||||||
|
std::vector<ServerInfo> servers = get_configured_servers();
|
||||||
|
for (const auto &server : servers)
|
||||||
|
{
|
||||||
|
rval = install_server(server.name);
|
||||||
|
if (rval != 0)
|
||||||
|
return rval;
|
||||||
|
}
|
||||||
|
|
||||||
std::cout << "Installation complete." << std::endl;
|
std::cout << "Installation complete." << std::endl;
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int install_server(const std::string &server)
|
|
||||||
{
|
|
||||||
// install the dropshell agent on the given server.
|
|
||||||
std::cout << "Installing dropshell agent on " << server << std::endl;
|
|
||||||
|
|
||||||
std::string agent_path = remotepath::agent(server);
|
|
||||||
if (agent_path.empty())
|
|
||||||
{
|
|
||||||
std::cerr << "Failed to get agent path for " << server << std::endl;
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
server_env_manager server_env(server);
|
|
||||||
if (!server_env.is_valid())
|
|
||||||
{
|
|
||||||
std::cerr << "Invalid server environment for " << server << std::endl;
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
// now create the agent.
|
|
||||||
// copy across from the local agent files.
|
|
||||||
shared_commands::rsync_tree_to_remote(localpath::files_for_remote_agent(), agent_path, server_env, false);
|
|
||||||
|
|
||||||
// add in bb64. We can't use execute_remote_command() here, as that relies on bb64 which we're installing!
|
|
||||||
std::cout << "Installing bb64 on " << server << std::endl << std::flush;
|
|
||||||
|
|
||||||
std::string remote_cmd =
|
|
||||||
"ssh -p " + server_env.get_SSH_INFO().port + " " + server_env.get_SSH_INFO().user + "@" + server_env.get_SSH_INFO().host +
|
|
||||||
" 'mkdir -p " + quote(agent_path) + " && curl -fsSL \"https://gitea.jde.nz/public/bb64/releases/download/latest/install.sh\" | bash -s -- " +
|
|
||||||
quote(agent_path) + " " + quote("$(id -u " + server_env.get_SSH_USER() + "):$(id -g " + server_env.get_SSH_USER() + ")") + "'";
|
|
||||||
|
|
||||||
//std::cout << "Executing: " << remote_cmd << std::endl;
|
|
||||||
if (!execute_local_command(remote_cmd, nullptr, cMode::Silent))
|
|
||||||
std::cerr << "Failed to download bb64 to " << agent_path << " on remote server." << std::endl;
|
|
||||||
else
|
|
||||||
std::cout << "Downloaded bb64 to " << agent_path << " on remote server." << std::endl;
|
|
||||||
|
|
||||||
// just test all is ok
|
|
||||||
std::string output;
|
|
||||||
bool okay = execute_ssh_command(server_env.get_SSH_INFO(), sCommand(agent_path, "./bb64 -i VGhlIGRyb3BzaGVsbCByZW1vdGUgYWdlbnQgaXMgY29ycmVjdGx5IGluc3RhbGxlZC4=", {}), cMode::CaptureOutput, &output);
|
|
||||||
if (!okay)
|
|
||||||
{
|
|
||||||
std::cerr << "Failed to install bb64 on " << server << std::endl;
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::cout << output << std::endl;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
// install command implementation
|
// install command implementation
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
@ -339,7 +341,7 @@ namespace dropshell
|
|||||||
|
|
||||||
if (!gConfig().is_config_set())
|
if (!gConfig().is_config_set())
|
||||||
{
|
{
|
||||||
std::cerr << "Error: Dropshell is not configured. Please run 'dropshell edit' to configure it." << std::endl;
|
error << "Dropshell is not configured. Please run 'dropshell edit' to configure it." << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -359,7 +361,7 @@ namespace dropshell
|
|||||||
std::vector<LocalServiceInfo> services = get_server_services_info(server);
|
std::vector<LocalServiceInfo> services = get_server_services_info(server);
|
||||||
for (const auto &service : services)
|
for (const auto &service : services)
|
||||||
{
|
{
|
||||||
if (!install_service(server, service.service_name, false))
|
if (!shared_commands::install_service(server, service.service_name))
|
||||||
okay = false;
|
okay = false;
|
||||||
}
|
}
|
||||||
return okay ? 0 : 1;
|
return okay ? 0 : 1;
|
||||||
@ -367,7 +369,7 @@ namespace dropshell
|
|||||||
else
|
else
|
||||||
{ // install the specific service.
|
{ // install the specific service.
|
||||||
std::string service = safearg(ctx.args, 1);
|
std::string service = safearg(ctx.args, 1);
|
||||||
return install_service(server, service, false) ? 0 : 1;
|
return shared_commands::install_service(server, service) ? 0 : 1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -7,6 +7,7 @@
|
|||||||
#include "tableprint.hpp"
|
#include "tableprint.hpp"
|
||||||
#include "transwarp.hpp"
|
#include "transwarp.hpp"
|
||||||
#include "server_env_manager.hpp"
|
#include "server_env_manager.hpp"
|
||||||
|
#include "services.hpp"
|
||||||
|
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
@ -62,7 +63,7 @@ int list_handler(const CommandContext& ctx) {
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::cout << "List handler called with " << ctx.args.size() << " args\n";
|
debug << "List handler called with " << ctx.args.size() << " args\n";
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -73,16 +74,16 @@ void list_servers() {
|
|||||||
auto servers = get_configured_servers();
|
auto servers = get_configured_servers();
|
||||||
|
|
||||||
if (servers.empty()) {
|
if (servers.empty()) {
|
||||||
std::cout << "No servers found" << std::endl;
|
error << "No servers found" << std::endl;
|
||||||
std::cout << "Please run 'dropshell edit' to set up dropshell." << std::endl;
|
info << "Please run 'dropshell edit' to set up dropshell." << std::endl;
|
||||||
std::cout << "Then run 'dropshell create-server' to create a server." << std::endl;
|
info << "Then run 'dropshell create-server' to create a server." << std::endl;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
tableprint tp("All DropShell Servers");
|
tableprint tp("All DropShell Servers");
|
||||||
tp.add_row({"Name", "User", "Address", "Health", "Ports"});
|
tp.add_row({"Name", "User", "Address", "Health", "Ports"});
|
||||||
|
|
||||||
std::cout << "Checking "<<servers.size() << " servers: " << std::flush;
|
info << "Checking "<<servers.size() << " servers: " << std::flush;
|
||||||
int checked = 0;
|
int checked = 0;
|
||||||
|
|
||||||
transwarp::parallel exec{servers.size()};
|
transwarp::parallel exec{servers.size()};
|
||||||
@ -102,10 +103,10 @@ void list_servers() {
|
|||||||
tp.add_row({server.name, server.ssh_user, server.ssh_host, serviceticks, ports_used_str});
|
tp.add_row({server.name, server.ssh_user, server.ssh_host, serviceticks, ports_used_str});
|
||||||
++checked;
|
++checked;
|
||||||
// print out a tick character for each server checked.
|
// print out a tick character for each server checked.
|
||||||
std::cout << checked << " ✓ " << std::flush;
|
info << checked << " ✓ " << std::flush;
|
||||||
});
|
});
|
||||||
task->wait();
|
task->wait();
|
||||||
std::cout << std::endl << std::endl;
|
info << std::endl << std::endl;
|
||||||
tp.print();
|
tp.print();
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -115,7 +116,7 @@ void list_servers() {
|
|||||||
void show_server_details(const std::string& server_name) {
|
void show_server_details(const std::string& server_name) {
|
||||||
server_env_manager env(server_name);
|
server_env_manager env(server_name);
|
||||||
if (!env.is_valid()) {
|
if (!env.is_valid()) {
|
||||||
std::cerr << "Error: Invalid server environment file: " << server_name << std::endl;
|
error << "Error: Invalid server environment file: " << server_name << std::endl;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -125,14 +126,14 @@ void show_server_details(const std::string& server_name) {
|
|||||||
std::string ssh_user = env.get_SSH_USER();
|
std::string ssh_user = env.get_SSH_USER();
|
||||||
std::string ssh_port = env.get_SSH_PORT();
|
std::string ssh_port = env.get_SSH_PORT();
|
||||||
if (!ssh_address.empty()) {
|
if (!ssh_address.empty()) {
|
||||||
std::cout << std::endl << "Server Status:" << std::endl;
|
info << std::endl << "Server Status:" << std::endl;
|
||||||
std::cout << std::string(40, '-') << std::endl;
|
info << std::string(40, '-') << std::endl;
|
||||||
|
|
||||||
// Try to connect to the server
|
// Try to connect to the server
|
||||||
std::string cmd = "ssh -o ConnectTimeout=5 " + ssh_user + "@" + ssh_address + " -p " + ssh_port + " 'echo connected' 2>/dev/null";
|
std::string cmd = "ssh -o ConnectTimeout=5 " + ssh_user + "@" + ssh_address + " -p " + ssh_port + " 'true' 2>/dev/null";
|
||||||
int result = system(cmd.c_str());
|
int result = system(cmd.c_str());
|
||||||
if (result == 0) {
|
if (result == 0) {
|
||||||
std::cout << "Status: Online" << std::endl;
|
info << "Status: Online" << std::endl;
|
||||||
|
|
||||||
// // Get uptime if possible
|
// // Get uptime if possible
|
||||||
// cmd = "ssh " + ssh_address + " 'uptime' 2>/dev/null";
|
// cmd = "ssh " + ssh_address + " 'uptime' 2>/dev/null";
|
||||||
@ -141,10 +142,10 @@ void show_server_details(const std::string& server_name) {
|
|||||||
// std::cout << "Error: Failed to get uptime" << std::endl;
|
// std::cout << "Error: Failed to get uptime" << std::endl;
|
||||||
// }
|
// }
|
||||||
} else {
|
} else {
|
||||||
std::cout << "Status: Offline" << std::endl;
|
warning << "Status: Offline" << std::endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
std::cout << std::endl;
|
info << std::endl;
|
||||||
|
|
||||||
//---------------------
|
//---------------------
|
||||||
{
|
{
|
||||||
@ -161,7 +162,7 @@ void show_server_details(const std::string& server_name) {
|
|||||||
// list services, and run healthcheck on each
|
// list services, and run healthcheck on each
|
||||||
{
|
{
|
||||||
tableprint tp("Services: " + server_name, false);
|
tableprint tp("Services: " + server_name, false);
|
||||||
tp.add_row({"Status", "Service", "Ports"});
|
tp.add_row({"Status", "Service", "Template","Ports"});
|
||||||
|
|
||||||
|
|
||||||
std::map<std::string, shared_commands::ServiceStatus> status = shared_commands::get_all_services_status(server_name);
|
std::map<std::string, shared_commands::ServiceStatus> status = shared_commands::get_all_services_status(server_name);
|
||||||
@ -175,7 +176,7 @@ void show_server_details(const std::string& server_name) {
|
|||||||
for (const auto& port : service_status.ports)
|
for (const auto& port : service_status.ports)
|
||||||
ports_str += std::to_string(port) + " ";
|
ports_str += std::to_string(port) + " ";
|
||||||
|
|
||||||
tp.add_row({healthy, service_name, ports_str});
|
tp.add_row({healthy, service_name, get_service_info(server_name,service_name).template_name, ports_str});
|
||||||
} // end of for (const auto& service : services)
|
} // end of for (const auto& service : services)
|
||||||
tp.print();
|
tp.print();
|
||||||
} // end of list services
|
} // end of list services
|
||||||
|
@ -10,18 +10,18 @@
|
|||||||
|
|
||||||
#include "utils/assert.hpp"
|
#include "utils/assert.hpp"
|
||||||
|
|
||||||
#pragma message ("TODO: Fix issues with Nuke below.")
|
namespace dropshell
|
||||||
|
{
|
||||||
|
|
||||||
namespace dropshell {
|
int nuke_handler(const CommandContext &ctx);
|
||||||
|
static std::vector<std::string> nuke_name_list = {"nuke"};
|
||||||
|
|
||||||
int nuke_handler(const CommandContext& ctx);
|
// Static registration
|
||||||
static std::vector<std::string> nuke_name_list={"nuke"};
|
struct NukeCommandRegister
|
||||||
|
{
|
||||||
// Static registration
|
NukeCommandRegister()
|
||||||
struct NukeCommandRegister {
|
{
|
||||||
NukeCommandRegister() {
|
CommandRegistry::instance().register_command({nuke_name_list,
|
||||||
CommandRegistry::instance().register_command({
|
|
||||||
nuke_name_list,
|
|
||||||
nuke_handler,
|
nuke_handler,
|
||||||
shared_commands::std_autocomplete,
|
shared_commands::std_autocomplete,
|
||||||
false, // hidden
|
false, // hidden
|
||||||
@ -33,16 +33,25 @@ struct NukeCommandRegister {
|
|||||||
"Nuke a service on a server. Destroys everything, both local and remote!",
|
"Nuke a service on a server. Destroys everything, both local and remote!",
|
||||||
// heredoc
|
// heredoc
|
||||||
R"(
|
R"(
|
||||||
Nuke a service on a server. Destroys everything, both local and remote!
|
Nuke a service.
|
||||||
|
|
||||||
|
Examples:
|
||||||
nuke SERVER SERVICE nuke the given service on the given server.
|
nuke SERVER SERVICE nuke the given service on the given server.
|
||||||
nuke SERVER all nuke all services on the given server.
|
nuke SERVER all nuke all services on the given server.
|
||||||
)"
|
|
||||||
});
|
|
||||||
}
|
|
||||||
} nuke_command_register;
|
|
||||||
|
|
||||||
int nuke_one(std::string server, std::string service)
|
Note: This command is destructive and will destroy all data and all configuration,
|
||||||
{
|
both on the dropshell host and on the remote server.
|
||||||
|
|
||||||
|
Use with caution!
|
||||||
|
)"});
|
||||||
|
}
|
||||||
|
} nuke_command_register;
|
||||||
|
|
||||||
|
namespace shared_commands
|
||||||
|
{
|
||||||
|
|
||||||
|
bool nuke_service(const std::string &server, const std::string &service)
|
||||||
|
{
|
||||||
server_env_manager server_env(server);
|
server_env_manager server_env(server);
|
||||||
|
|
||||||
// step 1 - nuke on remote server.
|
// step 1 - nuke on remote server.
|
||||||
@ -52,7 +61,7 @@ int nuke_one(std::string server, std::string service)
|
|||||||
|
|
||||||
service_info = get_service_info(server, service);
|
service_info = get_service_info(server, service);
|
||||||
if (!SIvalid(service_info))
|
if (!SIvalid(service_info))
|
||||||
std::cerr << "Warning: Invalid service: " << service << std::endl;
|
error << "Invalid service: " << service << std::endl;
|
||||||
|
|
||||||
if (server_env.check_remote_dir_exists(remotepath::service(server, service)))
|
if (server_env.check_remote_dir_exists(remotepath::service(server, service)))
|
||||||
{
|
{
|
||||||
@ -60,50 +69,54 @@ int nuke_one(std::string server, std::string service)
|
|||||||
// otherwise just uninstall.
|
// otherwise just uninstall.
|
||||||
if (gTemplateManager().template_command_exists(service_info.template_name, "nuke"))
|
if (gTemplateManager().template_command_exists(service_info.template_name, "nuke"))
|
||||||
{
|
{
|
||||||
|
info << "Running nuke script for " << service << " on " << server << std::endl;
|
||||||
if (!server_env.run_remote_template_command(service, "nuke", {}, false, {}))
|
if (!server_env.run_remote_template_command(service, "nuke", {}, false, {}))
|
||||||
std::cerr << "Warning: Failed to run nuke script: " << service << std::endl;
|
warning << "Failed to run nuke script: " << service << std::endl;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
|
info << "No nuke script found for " << service << " on " << server << std::endl;
|
||||||
|
info << "Running uninstall script instead and will clean directories." << std::endl;
|
||||||
if (!server_env.run_remote_template_command(service, "uninstall", {}, false, {}))
|
if (!server_env.run_remote_template_command(service, "uninstall", {}, false, {}))
|
||||||
std::cerr << "Warning: Failed to uninstall service: " << service << std::endl;
|
warning << "Failed to uninstall service: " << service << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Remove the service directory from the server, running in a docker container as root.
|
// Remove the service directory from the server, running in a docker container as root.
|
||||||
if (server_env.remove_remote_dir(remotepath::service(server, service), true))
|
if (server_env.remove_remote_dir(remotepath::service(server, service), true))
|
||||||
{
|
{
|
||||||
ASSERT(!server_env.check_remote_dir_exists(remotepath::service(server, service)), "Service directory still found on server after uninstall");
|
ASSERT(!server_env.check_remote_dir_exists(remotepath::service(server, service)), "Service directory still found on server after uninstall");
|
||||||
std::cout << "Remote service directory removed: " << remotepath::service(server, service) << std::endl;
|
info << "Remote service directory removed: " << remotepath::service(server, service) << std::endl;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
std::cerr << "Warning: Failed to remove remote service directory" << std::endl;
|
warning << "Failed to remove remote service directory" << std::endl;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
std::cerr << "Warning: Service not found on remote server: " << remotepath::service(server, service) << std::endl;
|
warning << "Service not found on remote server: " << remotepath::service(server, service) << std::endl;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
std::cerr << "Warning: Can't nuke the remote service as the server is invalid: " << server << std::endl;
|
warning << "Can't nuke the remote service as the server is invalid: " << server << std::endl;
|
||||||
|
|
||||||
// step 2 - nuke the local service directory.
|
// step 2 - nuke the local service directory.
|
||||||
std::string local_service_path = localpath::service(server, service);
|
std::string local_service_path = localpath::service(server, service);
|
||||||
if (local_service_path.empty() || !std::filesystem::exists(local_service_path))
|
if (local_service_path.empty() || !std::filesystem::exists(local_service_path))
|
||||||
{
|
{
|
||||||
std::cerr << "Warning: Local service directory not found: " << local_service_path << std::endl;
|
warning << "Local service directory not found: " << local_service_path << std::endl;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
auto itemsdeleted = std::filesystem::remove_all(local_service_path);
|
auto itemsdeleted = std::filesystem::remove_all(local_service_path);
|
||||||
if (itemsdeleted == 0)
|
if (itemsdeleted == 0)
|
||||||
std::cerr << "Error: Failed to remove local service directory" << std::endl;
|
error << "Failed to remove local service directory" << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::cout << "Nuked service " << service << " on server " << server << std::endl;
|
info << "Nuked service " << service << " on server " << server << std::endl;
|
||||||
|
|
||||||
return 0;
|
return true;
|
||||||
}
|
}
|
||||||
|
} // namespace shared_commands
|
||||||
|
|
||||||
int nuke_handler(const CommandContext &ctx)
|
int nuke_handler(const CommandContext &ctx)
|
||||||
{
|
{
|
||||||
ASSERT(ctx.args.size() == 2, "Usage: nuke SERVER SERVICE|all (requires 2 args - you supplied " + std::to_string(ctx.args.size()) + ")");
|
ASSERT(ctx.args.size() == 2, "Usage: nuke SERVER SERVICE|all (requires 2 args - you supplied " + std::to_string(ctx.args.size()) + ")");
|
||||||
ASSERT(gConfig().is_config_set(), "No configuration found. Please run 'dropshell config' to set up your configuration.");
|
ASSERT(gConfig().is_config_set(), "No configuration found. Please run 'dropshell config' to set up your configuration.");
|
||||||
|
|
||||||
@ -118,24 +131,24 @@ int nuke_handler(const CommandContext &ctx)
|
|||||||
std::string server_path = localpath::server(server);
|
std::string server_path = localpath::server(server);
|
||||||
if (server_path.empty())
|
if (server_path.empty())
|
||||||
{
|
{
|
||||||
std::cerr << "Error: Server not found: " << server << std::endl;
|
error << "Server not found: " << server << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
for (const auto& entry : std::filesystem::directory_iterator(server_path))
|
for (const auto &entry : std::filesystem::directory_iterator(server_path))
|
||||||
{
|
{
|
||||||
if (entry.is_directory() && entry.path().filename().string().find(".") != 0)
|
if (entry.is_directory() && entry.path().filename().string().find(".") != 0)
|
||||||
{
|
{
|
||||||
std::string service_name = entry.path().filename().string();
|
std::string service_name = entry.path().filename().string();
|
||||||
rval |= nuke_one(server, service_name);
|
rval |= (shared_commands::nuke_service(server, service_name) ? 0 : 1);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return rval;
|
return rval;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
return nuke_one(server, service);
|
return (shared_commands::nuke_service(server, service) ? 0 : 1);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace dropshell
|
} // namespace dropshell
|
265
source/src/commands/restoredata.cpp
Normal file
265
source/src/commands/restoredata.cpp
Normal file
@ -0,0 +1,265 @@
|
|||||||
|
#include <unistd.h>
|
||||||
|
#include <cstring>
|
||||||
|
#include <iostream>
|
||||||
|
#include <sstream>
|
||||||
|
#include <filesystem>
|
||||||
|
|
||||||
|
#include "utils/output.hpp"
|
||||||
|
#include "utils/assert.hpp"
|
||||||
|
#include "utils/utils.hpp"
|
||||||
|
#include "command_registry.hpp"
|
||||||
|
#include "config.hpp"
|
||||||
|
#include "services.hpp"
|
||||||
|
#include "servers.hpp"
|
||||||
|
#include "server_env_manager.hpp"
|
||||||
|
#include "templates.hpp"
|
||||||
|
#include "utils/directories.hpp"
|
||||||
|
#include "shared_commands.hpp"
|
||||||
|
|
||||||
|
namespace dropshell
|
||||||
|
{
|
||||||
|
|
||||||
|
int restoredata_handler(const CommandContext &ctx);
|
||||||
|
|
||||||
|
void restoredata_autocomplete(const CommandContext &ctx);
|
||||||
|
|
||||||
|
static std::vector<std::string> restoredata_name_list = {"restoredata", "rd", "restore", "rest"};
|
||||||
|
|
||||||
|
// Static registration
|
||||||
|
struct RestoreDataCommandRegister
|
||||||
|
{
|
||||||
|
RestoreDataCommandRegister()
|
||||||
|
{
|
||||||
|
CommandRegistry::instance().register_command({restoredata_name_list,
|
||||||
|
restoredata_handler,
|
||||||
|
restoredata_autocomplete,
|
||||||
|
false, // hidden
|
||||||
|
true, // requires_config
|
||||||
|
true, // requires_install
|
||||||
|
3, // min_args (after command)
|
||||||
|
3, // max_args (after command)
|
||||||
|
"restoredata SERVER SERVICE BACKUP_FILE|latest",
|
||||||
|
"Restore data for a service on a server, overwriting the existing data.",
|
||||||
|
// heredoc
|
||||||
|
R"(
|
||||||
|
restoredata SERVER SERVICE BACKUP_FILE Restore data to a service on a server. Destructive.
|
||||||
|
restoredata SERVER SERVICE latest Restore the latest backup for the given service.
|
||||||
|
|
||||||
|
Note: This command will not create any service configuration, you need
|
||||||
|
to have a valid service installed first.
|
||||||
|
The backup file must be in the local backups directory.
|
||||||
|
|
||||||
|
WARNING: This will permanently overwrite the service's data on the remote server!
|
||||||
|
)"});
|
||||||
|
}
|
||||||
|
} restoredata_command_register;
|
||||||
|
|
||||||
|
|
||||||
|
std::vector<shared_commands::cBackupFileName> get_backup_files(const std::string &server, const std::string &match_service = "", const std::string &match_template_name = "")
|
||||||
|
{
|
||||||
|
std::string local_backups_dir = gConfig().get_local_backup_path();
|
||||||
|
if (local_backups_dir.empty() || !std::filesystem::exists(local_backups_dir))
|
||||||
|
{
|
||||||
|
error << "Error: Local backups directory not found: " << local_backups_dir << std::endl;
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<shared_commands::cBackupFileName> backups;
|
||||||
|
for (const auto &entry : std::filesystem::directory_iterator(local_backups_dir))
|
||||||
|
{
|
||||||
|
if (!entry.is_regular_file())
|
||||||
|
continue;
|
||||||
|
std::string filename = entry.path().filename().string();
|
||||||
|
shared_commands::cBackupFileName backup_details(filename);
|
||||||
|
if (backup_details.is_valid())
|
||||||
|
if (match_service.empty() || backup_details.get_service() == match_service)
|
||||||
|
if (match_template_name.empty() || backup_details.get_template_name() == match_template_name)
|
||||||
|
backups.push_back(backup_details);
|
||||||
|
}
|
||||||
|
|
||||||
|
// sort backups by datetime
|
||||||
|
std::sort(backups.begin(), backups.end(), [](const shared_commands::cBackupFileName &a, const shared_commands::cBackupFileName &b)
|
||||||
|
{ return a.get_datetime() > b.get_datetime(); });
|
||||||
|
|
||||||
|
return backups;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int restoredata_handler(const CommandContext &ctx)
|
||||||
|
{
|
||||||
|
ASSERT(ctx.args.size() == 3, "Invalid number of arguments");
|
||||||
|
std::string server = ctx.args[0];
|
||||||
|
std::string service = ctx.args[1];
|
||||||
|
std::string backup_arg = ctx.args[2];
|
||||||
|
|
||||||
|
server_env_manager server_env(server);
|
||||||
|
if (!server_env.is_valid())
|
||||||
|
{
|
||||||
|
error << "Server " << server << " is not valid" << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
LocalServiceInfo service_info = get_service_info(server, service);
|
||||||
|
if (!SIvalid(service_info))
|
||||||
|
{
|
||||||
|
error << "Service " << service << " is not valid" << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::optional<shared_commands::cBackupFileName> backup_details;
|
||||||
|
if (backup_arg == "latest")
|
||||||
|
{ // special case.
|
||||||
|
std::vector<shared_commands::cBackupFileName> backups = get_backup_files(server, service, service_info.template_name); // this service only (and also match template in case something changed there!).
|
||||||
|
if (backups.empty())
|
||||||
|
{
|
||||||
|
error << "No backups found for " << server << "/" << service << std::endl;
|
||||||
|
debug << "Template also has to match with the service template: " << service_info.template_name << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
backup_details = backups[0];
|
||||||
|
} else {
|
||||||
|
backup_details = shared_commands::cBackupFileName(backup_arg);
|
||||||
|
if (!backup_details->is_valid())
|
||||||
|
{
|
||||||
|
error << "Invalid backup file: " << backup_arg << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ASSERT(backup_details.has_value() && backup_details->is_valid(), "Invalid backup file.");
|
||||||
|
|
||||||
|
debug << "Backup details: " << std::endl;
|
||||||
|
debug << " Backup filename: " << backup_details->get_filename() << std::endl;
|
||||||
|
debug << " Backup template: " << backup_details->get_template_name() << std::endl;
|
||||||
|
debug << " Backup taken from server: " << backup_details->get_server() << std::endl;
|
||||||
|
debug << " Backup taken from service: " << backup_details->get_service() << std::endl;
|
||||||
|
debug << " " << std::endl;
|
||||||
|
debug << "Restoring to:" << std::endl;
|
||||||
|
debug << " Server: " << server << std::endl;
|
||||||
|
debug << " Service: " << service << std::endl;
|
||||||
|
|
||||||
|
std::string local_backups_dir = gConfig().get_local_backup_path();
|
||||||
|
if (local_backups_dir.empty() || !std::filesystem::exists(local_backups_dir))
|
||||||
|
{
|
||||||
|
error << "Error: Local backups directory not found: " << local_backups_dir << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
std::string local_backup_file_path = (std::filesystem::path(local_backups_dir) / backup_details->get_filename()).string();
|
||||||
|
if (!std::filesystem::exists(local_backup_file_path))
|
||||||
|
{
|
||||||
|
error << "Error: Backup file not found at " << local_backup_file_path << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (backup_details->get_template_name() != service_info.template_name)
|
||||||
|
{
|
||||||
|
error << "Error: Backup template does not match service template. Can't restore." << std::endl;
|
||||||
|
info << "Backup template: " << backup_details->get_template_name() << std::endl;
|
||||||
|
info << "Service template: " << service_info.template_name << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
warning << "*** ALL DATA FOR " << server << "/" << service << " WILL BE OVERWRITTEN! ***" << std::endl;
|
||||||
|
|
||||||
|
// run the restore script
|
||||||
|
info << "OK, here goes..." << std::endl;
|
||||||
|
|
||||||
|
{ // backup existing service
|
||||||
|
info << "1) Backing up old service... " << std::endl;
|
||||||
|
if (!shared_commands::backupdata_service(server, service))
|
||||||
|
{
|
||||||
|
error << "Backup failed, restore aborted." << std::endl;
|
||||||
|
info << "You can try using dropshell install " << server << " " << service << " to install the service afresh." << std::endl;
|
||||||
|
info << "Otherwise, stop the service, create and initialise a new one, then restore to that." << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
info << "Backup complete." << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
{ // nuke the old service
|
||||||
|
info << "2) Nuking old service..." << std::endl;
|
||||||
|
if (!shared_commands::nuke_service(server, service))
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
{ // create the new service
|
||||||
|
info << "3) Creating new service..." << std::endl;
|
||||||
|
if (!shared_commands::create_service(server, service_info.template_name, service))
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
{ // installing fresh service
|
||||||
|
info << "4) Install of fresh service..." << std::endl;
|
||||||
|
if (!shared_commands::install_service(server, service))
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
{ // restore service from backup
|
||||||
|
info << "5) Restoring service data from backup..." << std::endl;
|
||||||
|
std::string remote_backups_dir = remotepath::backups(server);
|
||||||
|
std::string remote_backup_file_path = remote_backups_dir + "/" + backup_details->get_filename();
|
||||||
|
|
||||||
|
debug << "Copying backup file from local to server: " << local_backup_file_path << " -> " << remote_backup_file_path << std::endl;
|
||||||
|
|
||||||
|
// Copy backup file from local to server
|
||||||
|
if (!shared_commands::scp_file_to_remote(server_env, local_backup_file_path, remote_backup_file_path, false))
|
||||||
|
{
|
||||||
|
error << "Failed to copy backup file from local to server" << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
shared_commands::cRemoteTempFolder remote_temp_folder(server_env);
|
||||||
|
|
||||||
|
debug << "Running restore script on server: " << server << std::endl;
|
||||||
|
debug << " BACKUP_FILE: " << remote_backup_file_path << std::endl;
|
||||||
|
debug << " TEMP_DIR: " << remote_temp_folder.path() << std::endl;
|
||||||
|
|
||||||
|
server_env.run_remote_template_command(service, "restore", {}, false, {{"BACKUP_FILE", remote_backup_file_path}, {"TEMP_DIR", remote_temp_folder.path()}});
|
||||||
|
} // dtor of remote_temp_folder will clean up the temp folder on the server
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
{ // healthcheck the service
|
||||||
|
info << "5) Healthchecking service..." << std::endl;
|
||||||
|
std::string green_tick = "\033[32m✓\033[0m";
|
||||||
|
std::string red_cross = "\033[31m✗\033[0m";
|
||||||
|
bool healthy = (server_env.run_remote_template_command(service, "status", {}, false, {}));
|
||||||
|
info << (healthy ? green_tick : red_cross) << " Service is " << (healthy ? "healthy" : "NOT healthy") << std::endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
void restoredata_autocomplete(const CommandContext &ctx)
|
||||||
|
{
|
||||||
|
shared_commands::std_autocomplete(ctx);
|
||||||
|
if (ctx.args.size() == 2) // next arg is the backup file
|
||||||
|
{
|
||||||
|
std::string server = ctx.args[0];
|
||||||
|
std::string service = ctx.args[1];
|
||||||
|
|
||||||
|
LocalServiceInfo service_info = get_service_info(server, service);
|
||||||
|
if (!SIvalid(service_info))
|
||||||
|
{
|
||||||
|
error << "Service " << service << " is not valid" << std::endl;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
std::string template_name = service_info.template_name;
|
||||||
|
|
||||||
|
std::vector<shared_commands::cBackupFileName> backups = get_backup_files(server, "", template_name); // any service, but must match template.
|
||||||
|
|
||||||
|
// print most recent backup for each {host,service} pair
|
||||||
|
std::map<std::string, std::string> unique_backups;
|
||||||
|
for (const auto &backup : backups)
|
||||||
|
{
|
||||||
|
std::string key = backup.get_server() + "-" + backup.get_service();
|
||||||
|
if (unique_backups.find(key) == unique_backups.end())
|
||||||
|
unique_backups[key] = backup.get_filename();
|
||||||
|
}
|
||||||
|
for (const auto &[key, value] : unique_backups)
|
||||||
|
rawout << value << std::endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace dropshell
|
@ -5,6 +5,7 @@
|
|||||||
#include "directories.hpp"
|
#include "directories.hpp"
|
||||||
#include "services.hpp"
|
#include "services.hpp"
|
||||||
#include "servers.hpp"
|
#include "servers.hpp"
|
||||||
|
#include "utils/output.hpp"
|
||||||
|
|
||||||
namespace dropshell
|
namespace dropshell
|
||||||
{
|
{
|
||||||
@ -23,7 +24,7 @@ namespace dropshell
|
|||||||
std::vector<ServerInfo> servers = get_configured_servers();
|
std::vector<ServerInfo> servers = get_configured_servers();
|
||||||
for (const auto &server : servers)
|
for (const auto &server : servers)
|
||||||
{
|
{
|
||||||
std::cout << server.name << std::endl;
|
rawout << server.name << std::endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else if (ctx.args.size() == 1)
|
else if (ctx.args.size() == 1)
|
||||||
@ -32,7 +33,7 @@ namespace dropshell
|
|||||||
std::vector<LocalServiceInfo> services = get_server_services_info(ctx.args[0]);
|
std::vector<LocalServiceInfo> services = get_server_services_info(ctx.args[0]);
|
||||||
for (const auto &service : services)
|
for (const auto &service : services)
|
||||||
{
|
{
|
||||||
std::cout << service.service_name << std::endl;
|
rawout << service.service_name << std::endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -44,7 +45,7 @@ namespace dropshell
|
|||||||
{
|
{
|
||||||
std_autocomplete(ctx);
|
std_autocomplete(ctx);
|
||||||
if (ctx.args.size() == 1)
|
if (ctx.args.size() == 1)
|
||||||
std::cout << "all" << std::endl;
|
rawout << "all" << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
@ -62,7 +63,7 @@ namespace dropshell
|
|||||||
quote(local_path + "/") + " " +
|
quote(local_path + "/") + " " +
|
||||||
quote(server_env.get_SSH_USER() + "@" + server_env.get_SSH_HOST() + ":" +
|
quote(server_env.get_SSH_USER() + "@" + server_env.get_SSH_HOST() + ":" +
|
||||||
remote_path + "/");
|
remote_path + "/");
|
||||||
return execute_local_command(rsync_cmd, nullptr, (silent ? cMode::Silent : cMode::Defaults));
|
return execute_local_command("", rsync_cmd, {}, nullptr, (silent ? cMode::Silent : cMode::Defaults));
|
||||||
}
|
}
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
@ -88,7 +89,7 @@ namespace dropshell
|
|||||||
std::string p = remotepath::temp_files(server_env.get_server_name()) + "/" + random_alphanumeric_string(10);
|
std::string p = remotepath::temp_files(server_env.get_server_name()) + "/" + random_alphanumeric_string(10);
|
||||||
std::string mkdir_cmd = "mkdir -p " + quote(p);
|
std::string mkdir_cmd = "mkdir -p " + quote(p);
|
||||||
if (!execute_ssh_command(server_env.get_SSH_INFO(), sCommand("", mkdir_cmd, {}), cMode::Silent))
|
if (!execute_ssh_command(server_env.get_SSH_INFO(), sCommand("", mkdir_cmd, {}), cMode::Silent))
|
||||||
std::cerr << "Failed to create temp directory on server" << std::endl;
|
error << "Failed to create temp directory on server" << std::endl;
|
||||||
else
|
else
|
||||||
mPath = p;
|
mPath = p;
|
||||||
}
|
}
|
||||||
@ -114,12 +115,14 @@ namespace dropshell
|
|||||||
server_env_manager env(server_name);
|
server_env_manager env(server_name);
|
||||||
if (!env.is_valid())
|
if (!env.is_valid())
|
||||||
{
|
{
|
||||||
std::cerr << "Error: Invalid server environment" << std::endl;
|
error << "Invalid server environment" << std::endl;
|
||||||
return status;
|
return status;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string output;
|
std::string output;
|
||||||
if (!execute_ssh_command(env.get_SSH_INFO(), sCommand(remotepath::agent(server_name), "./_allservicesstatus.sh", {{"HOST_NAME", server_name}, {"SERVER", server_name}, {"AGENT_PATH", remotepath::agent(server_name)}}), cMode::CaptureOutput, &output))
|
if (!execute_ssh_command(env.get_SSH_INFO(), sCommand(remotepath::agent(server_name), "./_allservicesstatus.sh", {{"HOST_NAME", server_name}, {"SERVER", server_name}, {"AGENT_PATH", remotepath::agent(server_name)}}),
|
||||||
|
cMode::Silent,
|
||||||
|
&output))
|
||||||
return status;
|
return status;
|
||||||
|
|
||||||
std::stringstream ss(output);
|
std::stringstream ss(output);
|
||||||
@ -162,8 +165,6 @@ namespace dropshell
|
|||||||
return status;
|
return status;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
// healthtick : SHARED COMMAND
|
// healthtick : SHARED COMMAND
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
@ -202,7 +203,6 @@ namespace dropshell
|
|||||||
return ":error:";
|
return ":error:";
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
// is_healthy : SHARED COMMAND
|
// is_healthy : SHARED COMMAND
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
@ -211,7 +211,7 @@ namespace dropshell
|
|||||||
server_env_manager env(server);
|
server_env_manager env(server);
|
||||||
if (!env.is_valid())
|
if (!env.is_valid())
|
||||||
{
|
{
|
||||||
std::cerr << "Error: Server service not initialized" << std::endl;
|
error << "Server service not initialized" << std::endl;
|
||||||
return HealthStatus::ERROR;
|
return HealthStatus::ERROR;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -232,7 +232,6 @@ namespace dropshell
|
|||||||
return HealthStatus::HEALTHY;
|
return HealthStatus::HEALTHY;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
// healthmark : SHARED COMMAND
|
// healthmark : SHARED COMMAND
|
||||||
// ------------------------------------------------------------------------------------------------
|
// ------------------------------------------------------------------------------------------------
|
||||||
@ -242,6 +241,91 @@ namespace dropshell
|
|||||||
return HealthStatus2String(status);
|
return HealthStatus2String(status);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ------------------------------------------------------------------------------------------------
|
||||||
|
// cBackupFileName : SHARED CLASS
|
||||||
|
// ------------------------------------------------------------------------------------------------
|
||||||
|
cBackupFileName::cBackupFileName(const std::string &server, const std::string &service, const std::string &template_name)
|
||||||
|
{
|
||||||
|
mServer = server;
|
||||||
|
mService = service;
|
||||||
|
mTemplateName = template_name;
|
||||||
|
// Get current datetime for backup filename
|
||||||
|
auto now = std::chrono::system_clock::now();
|
||||||
|
auto time = std::chrono::system_clock::to_time_t(now);
|
||||||
|
std::stringstream datetime;
|
||||||
|
datetime << std::put_time(std::localtime(&time), "%Y-%m-%d_%H-%M-%S");
|
||||||
|
mDatetime = datetime.str();
|
||||||
|
}
|
||||||
|
|
||||||
|
cBackupFileName::cBackupFileName(const std::string &filename)
|
||||||
|
{
|
||||||
|
// Parse the filename according to the format:
|
||||||
|
// server + magic_string() + template_name + magic_string() + service + magic_string() + datetime + ".tgz"
|
||||||
|
std::string name = filename;
|
||||||
|
if (name.size() > 4 && name.substr(name.size() - 4) == ".tgz")
|
||||||
|
name = name.substr(0, name.size() - 4);
|
||||||
|
std::string sep = magic_string();
|
||||||
|
size_t first = name.find(sep);
|
||||||
|
size_t second = name.find(sep, first + sep.size());
|
||||||
|
size_t third = name.find(sep, second + sep.size());
|
||||||
|
if (first == std::string::npos || second == std::string::npos || third == std::string::npos)
|
||||||
|
{
|
||||||
|
mServer = mService = mTemplateName = mDatetime = "";
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
mServer = name.substr(0, first);
|
||||||
|
mTemplateName = name.substr(first + sep.size(), second - (first + sep.size()));
|
||||||
|
mService = name.substr(second + sep.size(), third - (second + sep.size()));
|
||||||
|
mDatetime = name.substr(third + sep.size());
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string cBackupFileName::get_filename() const
|
||||||
|
{
|
||||||
|
return mServer + magic_string() + mTemplateName + magic_string() + mService + magic_string() + mDatetime + ".tgz";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string cBackupFileName::get_server() const { return mServer; }
|
||||||
|
std::string cBackupFileName::get_service() const { return mService; }
|
||||||
|
std::string cBackupFileName::get_template_name() const { return mTemplateName; }
|
||||||
|
std::string cBackupFileName::get_datetime() const { return mDatetime; }
|
||||||
|
|
||||||
|
bool cBackupFileName::is_valid() const
|
||||||
|
{
|
||||||
|
// All fields must be non-empty, and none may contain the magic string
|
||||||
|
return !mServer.empty() && !mService.empty() && !mTemplateName.empty() && !mDatetime.empty() &&
|
||||||
|
!has_magic_string(mServer) && !has_magic_string(mService) && !has_magic_string(mTemplateName);
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------------------------------------------------------------------------
|
||||||
|
// scp_file_to_remote : SHARED COMMAND
|
||||||
|
// ------------------------------------------------------------------------------------------------
|
||||||
|
bool scp_file_to_remote(const server_env_manager &server_env, const std::string &local_path, const std::string &remote_path, bool silent)
|
||||||
|
{
|
||||||
|
if (!server_env.is_valid())
|
||||||
|
{
|
||||||
|
error << "Invalid server environment" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
ASSERT(!remote_path.empty() && !local_path.empty(), "Remote or local path not specified. Can't scp.");
|
||||||
|
std::string scp_cmd = "scp -P " + server_env.get_SSH_PORT() + " " + quote(local_path) + " " + server_env.get_SSH_USER() + "@" + server_env.get_SSH_HOST() + ":" + quote(remote_path) + (silent ? " > /dev/null 2>&1" : "");
|
||||||
|
return execute_local_command("", scp_cmd, {}, nullptr, (silent ? cMode::Silent : cMode::Defaults));
|
||||||
|
}
|
||||||
|
|
||||||
|
// ------------------------------------------------------------------------------------------------
|
||||||
|
// scp_file_from_remote : SHARED COMMAND
|
||||||
|
// ------------------------------------------------------------------------------------------------
|
||||||
|
bool scp_file_from_remote(const server_env_manager &server_env, const std::string &remote_path, const std::string &local_path, bool silent)
|
||||||
|
{
|
||||||
|
if (!server_env.is_valid())
|
||||||
|
{
|
||||||
|
error << "Invalid server environment" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
ASSERT(!remote_path.empty() && !local_path.empty(), "Remote or local path not specified. Can't scp.");
|
||||||
|
std::string scp_cmd = "scp -P " + server_env.get_SSH_PORT() + " " + server_env.get_SSH_USER() + "@" + server_env.get_SSH_HOST() + ":" + quote(remote_path) + " " + quote(local_path) + (silent ? " > /dev/null 2>&1" : "");
|
||||||
|
return execute_local_command("", scp_cmd, {}, nullptr, (silent ? cMode::Silent : cMode::Defaults));
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace shared_commands
|
} // namespace shared_commands
|
||||||
|
|
||||||
} // namespace dropshell
|
} // namespace dropshell
|
@ -57,6 +57,44 @@ namespace dropshell
|
|||||||
void std_autocomplete(const CommandContext &ctx);
|
void std_autocomplete(const CommandContext &ctx);
|
||||||
void std_autocomplete_allowall(const CommandContext &ctx);
|
void std_autocomplete_allowall(const CommandContext &ctx);
|
||||||
|
|
||||||
|
class cBackupFileName
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
cBackupFileName(const std::string &server, const std::string &service, const std::string &template_name);
|
||||||
|
cBackupFileName(const std::string &filename);
|
||||||
|
|
||||||
|
std::string get_filename() const;
|
||||||
|
std::string get_server() const;
|
||||||
|
std::string get_service() const;
|
||||||
|
std::string get_template_name() const;
|
||||||
|
std::string get_datetime() const;
|
||||||
|
|
||||||
|
bool is_valid() const;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::string mServer;
|
||||||
|
std::string mService;
|
||||||
|
std::string mTemplateName;
|
||||||
|
std::string mDatetime;
|
||||||
|
};
|
||||||
|
|
||||||
|
bool scp_file_to_remote(const server_env_manager &server_env, const std::string &local_path, const std::string &remote_path, bool silent);
|
||||||
|
bool scp_file_from_remote(const server_env_manager &server_env, const std::string &remote_path, const std::string &local_path, bool silent);
|
||||||
|
|
||||||
|
// defined in backupdata.cpp, used by restoredata.cpp.
|
||||||
|
bool backupdata_service(const std::string& server, const std::string& service);
|
||||||
|
|
||||||
|
// defined in uninstall.cpp
|
||||||
|
bool uninstall_service(const std::string &server, const std::string &service);
|
||||||
|
|
||||||
|
// defined in nuke.cpp
|
||||||
|
bool nuke_service(const std::string &server, const std::string &service);
|
||||||
|
|
||||||
|
// defined in install.cpp
|
||||||
|
bool install_service(const std::string &server, const std::string &service);
|
||||||
|
|
||||||
|
// defined in create-service.cpp
|
||||||
|
bool create_service(const std::string &server_name, const std::string &template_name, const std::string &service_name);
|
||||||
|
|
||||||
} // namespace shared_commands
|
} // namespace shared_commands
|
||||||
} // namespace dropshell
|
} // namespace dropshell
|
||||||
|
109
source/src/commands/ssh.cpp
Normal file
109
source/src/commands/ssh.cpp
Normal file
@ -0,0 +1,109 @@
|
|||||||
|
#include "command_registry.hpp"
|
||||||
|
#include "config.hpp"
|
||||||
|
#include "utils/utils.hpp"
|
||||||
|
#include "utils/directories.hpp"
|
||||||
|
#include "shared_commands.hpp"
|
||||||
|
#include "server_env_manager.hpp"
|
||||||
|
#include "services.hpp"
|
||||||
|
#include "servers.hpp"
|
||||||
|
#include "templates.hpp"
|
||||||
|
|
||||||
|
namespace dropshell
|
||||||
|
{
|
||||||
|
|
||||||
|
int ssh_handler(const CommandContext &ctx);
|
||||||
|
|
||||||
|
static std::vector<std::string> ssh_name_list = {"ssh"};
|
||||||
|
|
||||||
|
// Static registration
|
||||||
|
struct SSHCommandRegister
|
||||||
|
{
|
||||||
|
SSHCommandRegister()
|
||||||
|
{
|
||||||
|
CommandRegistry::instance().register_command({ssh_name_list,
|
||||||
|
ssh_handler,
|
||||||
|
shared_commands::std_autocomplete,
|
||||||
|
false, // hidden
|
||||||
|
true, // requires_config
|
||||||
|
true, // requires_install
|
||||||
|
1, // min_args (after command)
|
||||||
|
2, // max_args (after command)
|
||||||
|
"ssh SERVER",
|
||||||
|
"SSH into a server, or into a docker container for a service.",
|
||||||
|
R"(
|
||||||
|
|
||||||
|
ssh SERVER SERVICE SSH into a docker container for a service.
|
||||||
|
ssh SERVER SSH into a server.
|
||||||
|
)"});
|
||||||
|
}
|
||||||
|
} ssh_command_register;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bool ssh_into_server(const std::string &server)
|
||||||
|
{
|
||||||
|
server_env_manager server_env(server);
|
||||||
|
if (!server_env.is_valid())
|
||||||
|
{
|
||||||
|
error << "Server " << server << " is not valid" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
execute_ssh_command(server_env.get_SSH_INFO(), sCommand(remotepath::DROPSHELL_DIR(server), "ls --color && bash", {}), cMode::Interactive);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ssh_into_service(const std::string &server, const std::string &service)
|
||||||
|
{
|
||||||
|
server_env_manager server_env(server);
|
||||||
|
if (!server_env.is_valid())
|
||||||
|
{
|
||||||
|
error << "Server " << server << " is not valid" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
LocalServiceInfo sinfo = get_service_info(server, service);
|
||||||
|
if (!SIvalid(sinfo))
|
||||||
|
{
|
||||||
|
error << "Service " << service << " is not valid" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!gTemplateManager().has_template(sinfo.template_name))
|
||||||
|
{
|
||||||
|
error << "Template " << sinfo.template_name << " is not valid" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!gTemplateManager().template_command_exists(sinfo.template_name, "ssh"))
|
||||||
|
{
|
||||||
|
error << "Template " << sinfo.template_name << " does not have an ssh command" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
server_env.run_remote_template_command(service,"ssh",{},false,{}); // explicitly supports interactive ssh!
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ssh_handler(const CommandContext &ctx)
|
||||||
|
{
|
||||||
|
if (ctx.args.size() < 1)
|
||||||
|
{
|
||||||
|
error << "Server name is required" << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string server = safearg(ctx.args, 0);
|
||||||
|
|
||||||
|
if (ctx.args.size() < 2)
|
||||||
|
{
|
||||||
|
// ssh into the server
|
||||||
|
return ssh_into_server(server) ? 0 : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string service = safearg(ctx.args, 1);
|
||||||
|
|
||||||
|
// ssh into the specific service.
|
||||||
|
return ssh_into_service(server, service) ? 0 : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace dropshell
|
92
source/src/commands/start.cpp
Normal file
92
source/src/commands/start.cpp
Normal file
@ -0,0 +1,92 @@
|
|||||||
|
#include "command_registry.hpp"
|
||||||
|
#include "config.hpp"
|
||||||
|
#include "utils/utils.hpp"
|
||||||
|
#include "utils/directories.hpp"
|
||||||
|
#include "shared_commands.hpp"
|
||||||
|
#include "server_env_manager.hpp"
|
||||||
|
#include "services.hpp"
|
||||||
|
#include "servers.hpp"
|
||||||
|
#include "utils/output.hpp"
|
||||||
|
|
||||||
|
namespace dropshell
|
||||||
|
{
|
||||||
|
|
||||||
|
int start_handler(const CommandContext &ctx);
|
||||||
|
|
||||||
|
static std::vector<std::string> start_name_list = {"start", "start-service"};
|
||||||
|
|
||||||
|
// Static registration
|
||||||
|
struct StartCommandRegister
|
||||||
|
{
|
||||||
|
StartCommandRegister()
|
||||||
|
{
|
||||||
|
CommandRegistry::instance().register_command({start_name_list,
|
||||||
|
start_handler,
|
||||||
|
shared_commands::std_autocomplete_allowall,
|
||||||
|
false, // hidden
|
||||||
|
true, // requires_config
|
||||||
|
true, // requires_install
|
||||||
|
1, // min_args (after command)
|
||||||
|
2, // max_args (after command)
|
||||||
|
"start SERVER SERVICE|all",
|
||||||
|
"Start a service or all services on a server.",
|
||||||
|
R"(
|
||||||
|
|
||||||
|
start SERVER SERVICE Starts the given service on the given server.
|
||||||
|
start SERVER all Starts all services on the given server.
|
||||||
|
|
||||||
|
Note: This command will not create any data or configuration.
|
||||||
|
It will simply start the service on the remote server.
|
||||||
|
Stop the service with stop, or uninstall with uninstall.
|
||||||
|
)"});
|
||||||
|
}
|
||||||
|
} start_command_register;
|
||||||
|
|
||||||
|
bool start_service(const std::string &server, const std::string &service)
|
||||||
|
{
|
||||||
|
server_env_manager server_env(server);
|
||||||
|
if (!server_env.is_valid())
|
||||||
|
{
|
||||||
|
error << "Server " << server << " is not valid" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// run the start script.
|
||||||
|
bool started = server_env.run_remote_template_command(service, "start", {}, false, {});
|
||||||
|
|
||||||
|
if (started)
|
||||||
|
{
|
||||||
|
info << "Service " << service << " on server " << server << " started." << std::endl;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
error << "Failed to start service " << service << " on server " << server << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
int start_handler(const CommandContext &ctx)
|
||||||
|
{
|
||||||
|
if (ctx.args.size() < 2)
|
||||||
|
{
|
||||||
|
std::cerr << "Error: Server name and service name are both required" << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string server = safearg(ctx.args, 0);
|
||||||
|
std::string service = safearg(ctx.args, 1);
|
||||||
|
|
||||||
|
if (service == "all")
|
||||||
|
{
|
||||||
|
// install all services on the server
|
||||||
|
maketitle("Stopping all services on " + server);
|
||||||
|
bool okay = true;
|
||||||
|
std::vector<LocalServiceInfo> services = get_server_services_info(server);
|
||||||
|
for (const auto &service : services)
|
||||||
|
okay &= start_service(server, service.service_name);
|
||||||
|
return okay ? 0 : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// start the specific service.
|
||||||
|
return start_service(server, service) ? 0 : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace dropshell
|
92
source/src/commands/stop.cpp
Normal file
92
source/src/commands/stop.cpp
Normal file
@ -0,0 +1,92 @@
|
|||||||
|
#include "command_registry.hpp"
|
||||||
|
#include "config.hpp"
|
||||||
|
#include "utils/utils.hpp"
|
||||||
|
#include "utils/directories.hpp"
|
||||||
|
#include "shared_commands.hpp"
|
||||||
|
#include "server_env_manager.hpp"
|
||||||
|
#include "services.hpp"
|
||||||
|
#include "servers.hpp"
|
||||||
|
#include "utils/output.hpp"
|
||||||
|
|
||||||
|
namespace dropshell
|
||||||
|
{
|
||||||
|
|
||||||
|
int stop_handler(const CommandContext &ctx);
|
||||||
|
|
||||||
|
static std::vector<std::string> stop_name_list = {"stop", "stop-service"};
|
||||||
|
|
||||||
|
// Static registration
|
||||||
|
struct StopCommandRegister
|
||||||
|
{
|
||||||
|
StopCommandRegister()
|
||||||
|
{
|
||||||
|
CommandRegistry::instance().register_command({stop_name_list,
|
||||||
|
stop_handler,
|
||||||
|
shared_commands::std_autocomplete_allowall,
|
||||||
|
false, // hidden
|
||||||
|
true, // requires_config
|
||||||
|
true, // requires_install
|
||||||
|
1, // min_args (after command)
|
||||||
|
2, // max_args (after command)
|
||||||
|
"stop SERVER SERVICE|all",
|
||||||
|
"Stop a service or all services on a server.",
|
||||||
|
R"(
|
||||||
|
|
||||||
|
stop SERVER SERVICE Stops the given service on the given server.
|
||||||
|
stop SERVER all Stops all services on the given server.
|
||||||
|
|
||||||
|
Note: This command will not destroy any data or configuration.
|
||||||
|
It will simply stop the service on the remote server.
|
||||||
|
Restart the service with start, or update and start it with install.
|
||||||
|
)"});
|
||||||
|
}
|
||||||
|
} stop_command_register;
|
||||||
|
|
||||||
|
bool stop_service(const std::string &server, const std::string &service)
|
||||||
|
{
|
||||||
|
server_env_manager server_env(server);
|
||||||
|
if (!server_env.is_valid())
|
||||||
|
{
|
||||||
|
error << "Server " << server << " is not valid" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
// run the stop script.
|
||||||
|
bool stopped = server_env.run_remote_template_command(service, "stop", {}, false, {});
|
||||||
|
|
||||||
|
if (stopped)
|
||||||
|
{
|
||||||
|
info << "Service " << service << " on server " << server << " stopped." << std::endl;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
error << "Failed to stop service " << service << " on server " << server << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
int stop_handler(const CommandContext &ctx)
|
||||||
|
{
|
||||||
|
if (ctx.args.size() < 2)
|
||||||
|
{
|
||||||
|
error << "Server name and service name are both required" << std::endl;
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string server = safearg(ctx.args, 0);
|
||||||
|
std::string service = safearg(ctx.args, 1);
|
||||||
|
|
||||||
|
if (service == "all")
|
||||||
|
{
|
||||||
|
// install all services on the server
|
||||||
|
maketitle("Stopping all services on " + server);
|
||||||
|
bool okay = true;
|
||||||
|
std::vector<LocalServiceInfo> services = get_server_services_info(server);
|
||||||
|
for (const auto &service : services)
|
||||||
|
okay &= stop_service(server, service.service_name);
|
||||||
|
return okay ? 0 : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// stop the specific service.
|
||||||
|
return stop_service(server, service) ? 0 : 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace dropshell
|
@ -31,62 +31,59 @@ namespace dropshell
|
|||||||
"Uninstall a service on a server. Does not remove configuration or user data.",
|
"Uninstall a service on a server. Does not remove configuration or user data.",
|
||||||
// heredoc
|
// heredoc
|
||||||
R"(
|
R"(
|
||||||
Uninstall a service on a server. Does not remove configuration or user data.
|
Uninstall a service, leaving all configuration and data intact.
|
||||||
uninstall SERVER SERVICE uninstall the given service on the given server.
|
|
||||||
uninstall SERVER all uninstall all services on the given server.
|
uninstall SERVER SERVICE Uninstall the given service on the given server.
|
||||||
|
uninstall SERVER all Uninstall all services on the given server.
|
||||||
|
|
||||||
|
Update and reinstall the service with install, or delete all configuration and data with nuke.
|
||||||
)"});
|
)"});
|
||||||
}
|
}
|
||||||
} uninstall_command_register;
|
} uninstall_command_register;
|
||||||
|
|
||||||
|
namespace shared_commands {
|
||||||
|
bool uninstall_service(const std::string &server, const std::string &service)
|
||||||
|
|
||||||
bool uninstall_service(const std::string &server, const std::string &service, bool silent=false)
|
|
||||||
{
|
{
|
||||||
if (!silent)
|
|
||||||
maketitle("Uninstalling " + service + " on " + server);
|
maketitle("Uninstalling " + service + " on " + server);
|
||||||
|
|
||||||
server_env_manager server_env(server);
|
server_env_manager server_env(server);
|
||||||
if (!server_env.is_valid())
|
if (!server_env.is_valid())
|
||||||
{
|
{
|
||||||
std::cerr << "Invalid server: " << server << std::endl;
|
error << "Invalid server: " << server << std::endl;
|
||||||
return false; // should never hit this.
|
return false; // should never hit this.
|
||||||
}
|
}
|
||||||
|
|
||||||
// 2. Check if service directory exists on server
|
// 2. Check if service directory exists on server
|
||||||
if (!server_env.check_remote_dir_exists(remotepath::service(server, service)))
|
if (!server_env.check_remote_dir_exists(remotepath::service(server, service)))
|
||||||
{
|
{
|
||||||
std::cerr << "Service is not installed: " << service << std::endl;
|
error << "Service is not installed: " << service << std::endl;
|
||||||
return true; // Nothing to uninstall
|
return true; // Nothing to uninstall
|
||||||
}
|
}
|
||||||
|
|
||||||
// 3. Run uninstall script if it exists
|
// 3. Run uninstall script if it exists
|
||||||
std::string uninstall_script = remotepath::service_template(server, service) + "/uninstall.sh";
|
std::string uninstall_script = remotepath::service_template(server, service) + "/uninstall.sh";
|
||||||
if (!server_env.run_remote_template_command(service, "uninstall", {}, silent, {}))
|
if (!server_env.run_remote_template_command(service, "uninstall", {}, false, {}))
|
||||||
if (!silent)
|
warning << "Uninstall script failed, but continuing with directory removal" << std::endl;
|
||||||
std::cerr << "Warning: Uninstall script failed, but continuing with directory removal" << std::endl;
|
|
||||||
|
|
||||||
// 4. Remove the service directory from the server, running in a docker container as root.
|
// 4. Remove the service directory from the server, running in a docker container as root.
|
||||||
if (server_env.remove_remote_dir(remotepath::service(server, service), silent))
|
if (server_env.remove_remote_dir(remotepath::service(server, service), false))
|
||||||
{
|
{
|
||||||
ASSERT(!server_env.check_remote_dir_exists(remotepath::service(server, service)), "Service directory still found on server after uninstall");
|
ASSERT(!server_env.check_remote_dir_exists(remotepath::service(server, service)), "Service directory still found on server after uninstall");
|
||||||
if (!silent)
|
info << "Removed remote service directory " << remotepath::service(server, service) << std::endl;
|
||||||
std::cout << "Removed remote service directory " << remotepath::service(server, service) << std::endl;
|
|
||||||
}
|
}
|
||||||
else if (!silent)
|
else
|
||||||
std::cerr << "Warning: Failed to remove remote service directory" << std::endl;
|
warning << "Failed to remove remote service directory" << std::endl;
|
||||||
|
|
||||||
if (!silent)
|
info << "Completed service " << service << " uninstall on " << server << std::endl;
|
||||||
std::cout << "Completed service " << service << " uninstall on " << server << std::endl;
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
} // namespace shared_commands
|
||||||
|
|
||||||
int uninstall_handler(const CommandContext &ctx)
|
int uninstall_handler(const CommandContext &ctx)
|
||||||
{
|
{
|
||||||
if (ctx.args.size() < 1)
|
if (ctx.args.size() < 1)
|
||||||
{
|
{
|
||||||
std::cerr << "Error: uninstall requires a server and a service (or all)" << std::endl;
|
error << "uninstall requires a server and a service (or all)" << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -99,15 +96,14 @@ namespace dropshell
|
|||||||
std::vector<LocalServiceInfo> services = get_server_services_info(server);
|
std::vector<LocalServiceInfo> services = get_server_services_info(server);
|
||||||
for (const auto &service : services)
|
for (const auto &service : services)
|
||||||
{
|
{
|
||||||
if (!uninstall_service(server, service.service_name))
|
if (!shared_commands::uninstall_service(server, service.service_name))
|
||||||
okay = false;
|
okay = false;
|
||||||
}
|
}
|
||||||
return okay ? 0 : 1;
|
return okay ? 0 : 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string service = safearg(ctx.args, 1);
|
std::string service = safearg(ctx.args, 1);
|
||||||
return uninstall_service(server, service) ? 0 : 1;
|
return shared_commands::uninstall_service(server, service) ? 0 : 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
} // namespace dropshell
|
} // namespace dropshell
|
@ -6,6 +6,9 @@
|
|||||||
#include "utils/json.hpp"
|
#include "utils/json.hpp"
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
#include "utils/execute.hpp"
|
#include "utils/execute.hpp"
|
||||||
|
#include "output.hpp"
|
||||||
|
|
||||||
|
|
||||||
namespace dropshell {
|
namespace dropshell {
|
||||||
|
|
||||||
|
|
||||||
@ -150,7 +153,7 @@ std::vector<std::string> config::get_local_server_definition_paths() {
|
|||||||
if (path.is_string() && !path.empty())
|
if (path.is_string() && !path.empty())
|
||||||
paths.push_back(path);
|
paths.push_back(path);
|
||||||
else
|
else
|
||||||
std::cerr << "Warning: Invalid server definition path: " << path << std::endl;
|
warning << "Invalid server definition path: " << path << std::endl;
|
||||||
}
|
}
|
||||||
return paths;
|
return paths;
|
||||||
}
|
}
|
||||||
|
@ -8,6 +8,7 @@
|
|||||||
#include "autocomplete.hpp"
|
#include "autocomplete.hpp"
|
||||||
#include "utils/hash.hpp"
|
#include "utils/hash.hpp"
|
||||||
#include "command_registry.hpp"
|
#include "command_registry.hpp"
|
||||||
|
#include "output.hpp"
|
||||||
|
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
@ -49,33 +50,33 @@ int main(int argc, char* argv[]) {
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
const CommandInfo* info = CommandRegistry::instance().find_command(ctx.command);
|
const CommandInfo* cmdinfo = CommandRegistry::instance().find_command(ctx.command);
|
||||||
if (!info) {
|
if (!cmdinfo) {
|
||||||
std::cerr << "Unknown command: " << ctx.command << std::endl;
|
error << "Unknown command: " << ctx.command << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
if (info->requires_config && !gConfig().is_config_set()) {
|
if (cmdinfo->requires_config && !gConfig().is_config_set()) {
|
||||||
std::cerr << "Valid dropshell configuration required for command: " << ctx.command << std::endl;
|
error << "Valid dropshell configuration required for command: " << ctx.command << std::endl;
|
||||||
std::cerr << "Please run 'dropshell edit' to set up the dropshell configuration." << std::endl;
|
info << "Please run 'dropshell edit' to set up the dropshell configuration." << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
if (info->requires_install && !gConfig().is_agent_installed()) {
|
if (cmdinfo->requires_install && !gConfig().is_agent_installed()) {
|
||||||
std::cerr << "Dropshell agent not installed for command: " << ctx.command << std::endl;
|
error << "Dropshell agent not installed for command: " << ctx.command << std::endl;
|
||||||
std::cerr << "Please run 'dropshell install' to install the local dropshell agent." << std::endl;
|
info << "Please run 'dropshell install' to install the local dropshell agent." << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int arg_count = ctx.args.size();
|
int arg_count = ctx.args.size();
|
||||||
if (arg_count < info->min_args || (info->max_args != -1 && arg_count > info->max_args)) {
|
if (arg_count < cmdinfo->min_args || (cmdinfo->max_args != -1 && arg_count > cmdinfo->max_args)) {
|
||||||
std::cerr << "Invalid number of arguments for command: " << ctx.command << std::endl;
|
error << "Invalid number of arguments for command: " << ctx.command << std::endl;
|
||||||
std::cerr << "(" << ctx.args.size() << " args provided, " << ctx.command << " requires " << (info->min_args) << " to " << (info->max_args) << " args)" << std::endl;
|
debug << "(" << ctx.args.size() << " args provided, " << ctx.command << " requires " << (cmdinfo->min_args) << " to " << (cmdinfo->max_args) << " args)" << std::endl;
|
||||||
std::cerr << "Usage: " << std::endl;
|
info << "Usage: " << std::endl;
|
||||||
std::cerr << " ";
|
info << " ";
|
||||||
print_left_aligned(info->help_usage,30);
|
info << left_align(cmdinfo->help_usage,30);
|
||||||
std::cout << info->help_description << std::endl;
|
info << cmdinfo->help_description << std::endl;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
return info->handler(ctx);
|
return cmdinfo->handler(ctx);
|
||||||
|
|
||||||
}
|
}
|
||||||
catch (const std::exception& e) {
|
catch (const std::exception& e) {
|
||||||
@ -115,13 +116,6 @@ bool getCLIServices(const std::string & arg2, const std::string & arg3,
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
void printversion() {
|
|
||||||
maketitle("DropShell version " + VERSION);
|
|
||||||
std::cout << "Release date: " << RELEASE_DATE << std::endl;
|
|
||||||
std::cout << "Author: " << AUTHOR << std::endl;
|
|
||||||
std::cout << "License: " << LICENSE << std::endl;
|
|
||||||
}
|
|
||||||
|
|
||||||
auto command_match = [](const std::string& cmd_list, int argc, char* argv[]) -> bool {
|
auto command_match = [](const std::string& cmd_list, int argc, char* argv[]) -> bool {
|
||||||
std::istringstream iss(cmd_list);
|
std::istringstream iss(cmd_list);
|
||||||
std::string cmd_item;
|
std::string cmd_item;
|
||||||
|
@ -204,7 +204,7 @@ bool server_env_manager::run_remote_template_command(const std::string &service_
|
|||||||
|
|
||||||
if (scommand->get_command_to_run().empty())
|
if (scommand->get_command_to_run().empty())
|
||||||
return false;
|
return false;
|
||||||
cMode mode = (command=="ssh") ? (cMode::Interactive) : cMode::Silent;
|
cMode mode = (command=="ssh") ? (cMode::Interactive) : (silent ? cMode::Silent : cMode::Defaults);
|
||||||
return execute_ssh_command(get_SSH_INFO(), scommand.value(), mode);
|
return execute_ssh_command(get_SSH_INFO(), scommand.value(), mode);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -218,8 +218,7 @@ bool server_env_manager::run_remote_template_command_and_capture_output(const st
|
|||||||
for (const auto& [key, value] : extra_env_vars)
|
for (const auto& [key, value] : extra_env_vars)
|
||||||
scommand->add_env_var(key, value);
|
scommand->add_env_var(key, value);
|
||||||
|
|
||||||
cMode mode = cMode::CaptureOutput;
|
return execute_ssh_command(get_SSH_INFO(), scommand.value(), cMode::Defaults, &output);
|
||||||
return execute_ssh_command(get_SSH_INFO(), scommand.value(), mode, &output);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -91,7 +91,6 @@ namespace fs = std::filesystem;
|
|||||||
|
|
||||||
namespace dropshell {
|
namespace dropshell {
|
||||||
|
|
||||||
static const std::string magic_string = "-_-";
|
|
||||||
|
|
||||||
service_runner::service_runner(const std::string& server_name, const std::string& service_name) :
|
service_runner::service_runner(const std::string& server_name, const std::string& service_name) :
|
||||||
mServerEnv(server_name), mServer(server_name), mService(service_name), mValid(false)
|
mServerEnv(server_name), mServer(server_name), mService(service_name), mValid(false)
|
||||||
@ -122,12 +121,12 @@ bool service_runner::nuke(bool silent)
|
|||||||
|
|
||||||
std::string remote_service_path = remotepath::service(mServer, mService);
|
std::string remote_service_path = remotepath::service(mServer, mService);
|
||||||
|
|
||||||
std::cout << "Service " << mService << " successfully nuked from " << mServer << std::endl;
|
info << "Service " << mService << " successfully nuked from " << mServer << std::endl;
|
||||||
|
|
||||||
if (!silent) {
|
if (!silent) {
|
||||||
std::cout << "There's nothing left on the remote server." << std::endl;
|
info << "There's nothing left on the remote server." << std::endl;
|
||||||
std::cout << "You can remove the local files with:" << std::endl;
|
info << "You can remove the local files with:" << std::endl;
|
||||||
std::cout << " rm -rf " << localpath::service(mServer,mService) << std::endl;
|
info << " rm -rf " << localpath::service(mServer,mService) << std::endl;
|
||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
@ -136,19 +135,19 @@ bool service_runner::fullnuke()
|
|||||||
{
|
{
|
||||||
if (!nuke(true))
|
if (!nuke(true))
|
||||||
{
|
{
|
||||||
std::cerr << "Warning: Nuke script failed, aborting fullnuke!" << std::endl;
|
warning << "Nuke script failed, aborting." << std::endl;
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string local_service_path = mServiceInfo.local_service_path;
|
std::string local_service_path = mServiceInfo.local_service_path;
|
||||||
if (local_service_path.empty() || !fs::exists(local_service_path)) {
|
if (local_service_path.empty() || !fs::exists(local_service_path)) {
|
||||||
std::cerr << "Error: Service directory not found: " << local_service_path << std::endl;
|
error << "Service directory not found: " << local_service_path << std::endl;
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string rm_cmd = "rm -rf " + quote(local_service_path);
|
std::string rm_cmd = "rm -rf " + quote(local_service_path);
|
||||||
if (!execute_local_command(rm_cmd, nullptr, cMode::Silent)) {
|
if (!execute_local_command("", rm_cmd, {}, nullptr, cMode::Silent)) {
|
||||||
std::cerr << "Failed to remove service directory" << std::endl;
|
error << "Failed to remove service directory" << std::endl;
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -267,13 +266,13 @@ bool service_runner::interactive_ssh_service()
|
|||||||
bool service_runner::scp_file_to_remote(const std::string &local_path, const std::string &remote_path, bool silent)
|
bool service_runner::scp_file_to_remote(const std::string &local_path, const std::string &remote_path, bool silent)
|
||||||
{
|
{
|
||||||
std::string scp_cmd = "scp -P " + mServerEnv.get_SSH_PORT() + " " + quote(local_path) + " " + mServerEnv.get_SSH_USER() + "@" + mServerEnv.get_SSH_HOST() + ":" + quote(remote_path) + (silent ? " > /dev/null 2>&1" : "");
|
std::string scp_cmd = "scp -P " + mServerEnv.get_SSH_PORT() + " " + quote(local_path) + " " + mServerEnv.get_SSH_USER() + "@" + mServerEnv.get_SSH_HOST() + ":" + quote(remote_path) + (silent ? " > /dev/null 2>&1" : "");
|
||||||
return execute_local_command(scp_cmd, nullptr, (silent ? cMode::Silent : cMode::Defaults));
|
return execute_local_command("", scp_cmd, {}, nullptr, (silent ? cMode::Silent : cMode::Defaults));
|
||||||
}
|
}
|
||||||
|
|
||||||
bool service_runner::scp_file_from_remote(const std::string &remote_path, const std::string &local_path, bool silent)
|
bool service_runner::scp_file_from_remote(const std::string &remote_path, const std::string &local_path, bool silent)
|
||||||
{
|
{
|
||||||
std::string scp_cmd = "scp -P " + mServerEnv.get_SSH_PORT() + " " + mServerEnv.get_SSH_USER() + "@" + mServerEnv.get_SSH_HOST() + ":" + quote(remote_path) + " " + quote(local_path) + (silent ? " > /dev/null 2>&1" : "");
|
std::string scp_cmd = "scp -P " + mServerEnv.get_SSH_PORT() + " " + mServerEnv.get_SSH_USER() + "@" + mServerEnv.get_SSH_HOST() + ":" + quote(remote_path) + " " + quote(local_path) + (silent ? " > /dev/null 2>&1" : "");
|
||||||
return execute_local_command(scp_cmd, nullptr, (silent ? cMode::Silent : cMode::Defaults));
|
return execute_local_command("", scp_cmd, {}, nullptr, (silent ? cMode::Silent : cMode::Defaults));
|
||||||
}
|
}
|
||||||
|
|
||||||
bool service_runner::restore(std::string backup_file, bool silent)
|
bool service_runner::restore(std::string backup_file, bool silent)
|
||||||
@ -298,7 +297,7 @@ bool service_runner::restore(std::string backup_file, bool silent)
|
|||||||
}
|
}
|
||||||
|
|
||||||
// split the backup filename into parts based on the magic string
|
// split the backup filename into parts based on the magic string
|
||||||
std::vector<std::string> parts = dropshell::split(backup_file, magic_string);
|
std::vector<std::string> parts = dropshell::split(backup_file, "-_-");
|
||||||
if (parts.size() != 4) {
|
if (parts.size() != 4) {
|
||||||
std::cerr << "Error: Backup file format is incompatible, - in one of the names?" << std::endl;
|
std::cerr << "Error: Backup file format is incompatible, - in one of the names?" << std::endl;
|
||||||
return false;
|
return false;
|
||||||
@ -383,12 +382,6 @@ bool service_runner::restore(std::string backup_file, bool silent)
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
bool name_breaks_backups(std::string name)
|
|
||||||
{
|
|
||||||
// if name contains -_-, return true
|
|
||||||
return name.find("-_-") != std::string::npos;
|
|
||||||
}
|
|
||||||
|
|
||||||
// backup the service over ssh, using the credentials from server.env (via server_env.hpp)
|
// backup the service over ssh, using the credentials from server.env (via server_env.hpp)
|
||||||
// 1. run backup.sh on the server
|
// 1. run backup.sh on the server
|
||||||
// 2. create a backup file with format server-service-datetime.tgz
|
// 2. create a backup file with format server-service-datetime.tgz
|
||||||
@ -452,17 +445,18 @@ bool service_runner::backup(bool silent) {
|
|||||||
std::stringstream datetime;
|
std::stringstream datetime;
|
||||||
datetime << std::put_time(std::localtime(&time), "%Y-%m-%d_%H-%M-%S");
|
datetime << std::put_time(std::localtime(&time), "%Y-%m-%d_%H-%M-%S");
|
||||||
|
|
||||||
if (name_breaks_backups(mServer)) {std::cerr << "Error: Server name contains invalid character sequence ( -_- ) that would break backup naming scheme" << std::endl; return 1;}
|
|
||||||
if (name_breaks_backups(mService)) {std::cerr << "Error: Service name contains invalid character sequence ( -_- ) that would break backup naming scheme" << std::endl; return 1;}
|
|
||||||
if (name_breaks_backups(service_info.template_name)) {std::cerr << "Error: Service template name contains invalid character sequence ( -_- ) that would break backup naming scheme" << std::endl; return 1;}
|
|
||||||
|
|
||||||
// Construct backup filename
|
// Construct backup filename
|
||||||
std::string backup_filename = mServer + magic_string + service_info.template_name + magic_string + mService + magic_string + datetime.str() + ".tgz";
|
shared_commands::cBackupFileName backup_filename_construction(mServer, mService, service_info.template_name);
|
||||||
|
if (!backup_filename_construction.is_valid()) {
|
||||||
|
std::cerr << "Invalid backup filename" << std::endl;
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
std::string backup_filename = backup_filename_construction.get_filename();
|
||||||
std::string remote_backup_file_path = remote_backups_dir + "/" + backup_filename;
|
std::string remote_backup_file_path = remote_backups_dir + "/" + backup_filename;
|
||||||
std::string local_backup_file_path = (std::filesystem::path(local_backups_dir) / backup_filename).string();
|
std::string local_backup_file_path = (std::filesystem::path(local_backups_dir) / backup_filename).string();
|
||||||
|
|
||||||
// assert that the backup filename is valid - -_- appears exactly 3 times in local_backup_file_path.
|
// assert that the backup filename is valid - -_- appears exactly 3 times in local_backup_file_path.
|
||||||
ASSERT(3 == count_substring(magic_string, local_backup_file_path), "Invalid backup filename");
|
ASSERT(3 == count_substring("-_-", local_backup_file_path), "Invalid backup filename");
|
||||||
|
|
||||||
{ // Run backup script
|
{ // Run backup script
|
||||||
shared_commands::cRemoteTempFolder remote_temp_folder(mServerEnv);
|
shared_commands::cRemoteTempFolder remote_temp_folder(mServerEnv);
|
||||||
@ -501,7 +495,7 @@ std::string service_runner::get_latest_backup_file(const std::string& server, co
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Build the expected prefix for backup files
|
// Build the expected prefix for backup files
|
||||||
std::string prefix = server + magic_string + info.template_name + magic_string + service + magic_string;
|
std::string prefix = server + "-_-" + info.template_name + "-_-" + service + "-_-";
|
||||||
std::string latest_file;
|
std::string latest_file;
|
||||||
std::string latest_datetime;
|
std::string latest_datetime;
|
||||||
|
|
||||||
|
@ -47,7 +47,7 @@ std::vector<LocalServiceInfo> get_server_services_info(const std::string& server
|
|||||||
if (!service.local_service_path.empty())
|
if (!service.local_service_path.empty())
|
||||||
services.push_back(service);
|
services.push_back(service);
|
||||||
else
|
else
|
||||||
std::cerr << "Warning: Failed to get service info for " << dirname << " on server " << server_name << std::endl;
|
warning << "Failed to get service info for " << dirname << " on server " << server_name << std::endl;
|
||||||
}
|
}
|
||||||
} // end of for
|
} // end of for
|
||||||
}
|
}
|
||||||
@ -176,6 +176,7 @@ bool get_all_service_env_vars(const std::string &server_name, const std::string
|
|||||||
all_env_vars["SERVICE"] = service_name;
|
all_env_vars["SERVICE"] = service_name;
|
||||||
all_env_vars["AGENT_PATH"] = remotepath::agent(server_name);
|
all_env_vars["AGENT_PATH"] = remotepath::agent(server_name);
|
||||||
all_env_vars["HOST_NAME"] = server_info.ssh_host;
|
all_env_vars["HOST_NAME"] = server_info.ssh_host;
|
||||||
|
all_env_vars["DOCKER_CLI_HINTS"] = "false"; // turn off docker junk.
|
||||||
|
|
||||||
// Lambda function to load environment variables from a file
|
// Lambda function to load environment variables from a file
|
||||||
auto load_env_file = [&all_env_vars](const std::string& file) {
|
auto load_env_file = [&all_env_vars](const std::string& file) {
|
||||||
@ -187,7 +188,7 @@ bool get_all_service_env_vars(const std::string &server_name, const std::string
|
|||||||
all_env_vars.merge(env_vars);
|
all_env_vars.merge(env_vars);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
std::cout << "Warning: Expected environment file not found: " << file << std::endl;
|
warning << "Expected environment file not found: " << file << std::endl;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Load environment files
|
// Load environment files
|
||||||
@ -197,11 +198,10 @@ bool get_all_service_env_vars(const std::string &server_name, const std::string
|
|||||||
// determine template name.
|
// determine template name.
|
||||||
auto it = all_env_vars.find("TEMPLATE");
|
auto it = all_env_vars.find("TEMPLATE");
|
||||||
if (it == all_env_vars.end()) {
|
if (it == all_env_vars.end()) {
|
||||||
std::cerr << std::endl << std::endl;
|
error << "TEMPLATE variable not defined in service " << service_name << " on server " << server_name << std::endl;
|
||||||
std::cerr << "Error: TEMPLATE variable not defined in service " << service_name << " on server " << server_name << std::endl;
|
info << "The TEMPLATE variable is required to determine the template name." << std::endl;
|
||||||
std::cerr << "The TEMPLATE variable is required to determine the template name." << std::endl;
|
info << "Please check the service.env file and the .template_info.env file in:" << std::endl;
|
||||||
std::cerr << "Please check the service.env file and the .template_info.env file in:" << std::endl;
|
info << " " << localpath::service(server_name, service_name) << std::endl << std::endl;
|
||||||
std::cerr << " " << localpath::service(server_name, service_name) << std::endl << std::endl;
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
template_info tinfo = gTemplateManager().get_template_info(it->second);
|
template_info tinfo = gTemplateManager().get_template_info(it->second);
|
||||||
|
@ -1,10 +1,11 @@
|
|||||||
#ifndef ASSERT_HPP
|
#ifndef ASSERT_HPP
|
||||||
#define ASSERT_HPP
|
#define ASSERT_HPP
|
||||||
|
|
||||||
|
#include "output.hpp"
|
||||||
|
|
||||||
#define ASSERT(condition, message) \
|
#define ASSERT(condition, message) \
|
||||||
if (!(condition)) { \
|
if (!(condition)) { \
|
||||||
std::cerr << "Assertion failed: " << message << std::endl; \
|
dropshell::error << "Assertion failed: " << message << std::endl; \
|
||||||
std::exit(1); \
|
std::exit(1); \
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1,10 +1,13 @@
|
|||||||
#include "directories.hpp"
|
#include "directories.hpp"
|
||||||
#include "config.hpp"
|
#include "config.hpp"
|
||||||
#include "server_env_manager.hpp"
|
#include "server_env_manager.hpp"
|
||||||
|
#include "output.hpp"
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
|
|
||||||
|
|
||||||
namespace fs = std::filesystem;
|
namespace fs = std::filesystem;
|
||||||
|
|
||||||
namespace dropshell {
|
namespace dropshell {
|
||||||
@ -78,7 +81,7 @@ namespace localpath {
|
|||||||
std::filesystem::path homedir_path(homedir);
|
std::filesystem::path homedir_path(homedir);
|
||||||
return fs::canonical(homedir_path).string();
|
return fs::canonical(homedir_path).string();
|
||||||
}
|
}
|
||||||
std::cerr << "Warning: Couldn't determine user directory" << std::endl;
|
warning << "Couldn't determine user directory" << std::endl;
|
||||||
return std::string();
|
return std::string();
|
||||||
}
|
}
|
||||||
} // namespace localpath
|
} // namespace localpath
|
||||||
|
@ -13,6 +13,7 @@
|
|||||||
#include "utils/b64ed.hpp"
|
#include "utils/b64ed.hpp"
|
||||||
#include "config.hpp"
|
#include "config.hpp"
|
||||||
#include "utils/directories.hpp"
|
#include "utils/directories.hpp"
|
||||||
|
#include "utils/output.hpp"
|
||||||
|
|
||||||
namespace dropshell
|
namespace dropshell
|
||||||
{
|
{
|
||||||
@ -21,11 +22,14 @@ namespace dropshell
|
|||||||
return (ret != -1 && WIFEXITED(ret) && (WEXITSTATUS(ret) == 0)); // ret is -1 if the command failed to execute.
|
return (ret != -1 && WIFEXITED(ret) && (WEXITSTATUS(ret) == 0)); // ret is -1 if the command failed to execute.
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ----------------------------------------------------------------------------------------------------------
|
||||||
|
// execute_local_command_interactive
|
||||||
|
// ----------------------------------------------------------------------------------------------------------
|
||||||
bool execute_local_command_interactive(const sCommand &command)
|
bool execute_local_command_interactive(const sCommand &command)
|
||||||
{
|
{
|
||||||
if (command.get_command_to_run().empty())
|
if (command.get_command_to_run().empty())
|
||||||
return false;
|
return false;
|
||||||
std::string full_command = command.construct_cmd(localpath::agent()); // Get the command string
|
std::string full_command = command.construct_cmd(localpath::agent()+"/bb64"); // Get the command string
|
||||||
|
|
||||||
pid_t pid = fork();
|
pid_t pid = fork();
|
||||||
|
|
||||||
@ -51,30 +55,62 @@ namespace dropshell
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool execute_local_command_and_capture_output(const sCommand &command, std::string *output)
|
// ----------------------------------------------------------------------------------------------------------
|
||||||
|
// execute_local_command
|
||||||
|
// ----------------------------------------------------------------------------------------------------------
|
||||||
|
|
||||||
|
class fancypinter
|
||||||
{
|
{
|
||||||
ASSERT(output != nullptr, "Output string must be provided");
|
public:
|
||||||
if (command.get_command_to_run().empty())
|
fancypinter(sColour startColour) : startColour_(startColour), currentColour_(startColour) {}
|
||||||
return false;
|
|
||||||
std::string full_cmd = command.construct_cmd(localpath::agent()) + " 2>&1";
|
void print_chunk(std::string chunk)
|
||||||
FILE *pipe = popen(full_cmd.c_str(), "r");
|
|
||||||
if (!pipe)
|
|
||||||
{
|
{
|
||||||
return false;
|
if (chunk.empty())
|
||||||
|
return;
|
||||||
|
|
||||||
|
if (newline_)
|
||||||
|
{
|
||||||
|
// sniff the mode... if the string starts with warning or warning: then set mode to WARNING. etc.
|
||||||
|
if (chunk.find("warning") == 0)
|
||||||
|
currentColour_ = sColour::WARNING;
|
||||||
|
else if (chunk.find("error") == 0)
|
||||||
|
currentColour_ = sColour::ERROR;
|
||||||
|
else if (chunk.find("debug") == 0)
|
||||||
|
currentColour_ = sColour::DEBUG;
|
||||||
|
else if (chunk.find("info") == 0)
|
||||||
|
currentColour_ = sColour::INFO;
|
||||||
|
else
|
||||||
|
currentColour_ = startColour_;
|
||||||
}
|
}
|
||||||
char buffer[128];
|
colourstream(currentColour_) << chunk;
|
||||||
while (fgets(buffer, sizeof(buffer), pipe) != nullptr)
|
newline_ = (chunk[chunk.size() - 1] == '\n');
|
||||||
{
|
|
||||||
(*output) += buffer;
|
|
||||||
}
|
|
||||||
int ret = pclose(pipe);
|
|
||||||
return EXITSTATUSCHECK(ret);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
bool execute_local_command(std::string command, std::string *output, cMode mode)
|
void print(const std::string &buffer)
|
||||||
{
|
{
|
||||||
return execute_local_command("", command, {}, output, mode);
|
size_t start = 0;
|
||||||
|
while (start < buffer.size())
|
||||||
|
{
|
||||||
|
size_t newline_pos = buffer.find('\n', start);
|
||||||
|
if (newline_pos == std::string::npos)
|
||||||
|
{
|
||||||
|
if (start < buffer.size())
|
||||||
|
{
|
||||||
|
print_chunk(buffer.substr(start));
|
||||||
}
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
print_chunk(buffer.substr(start, newline_pos - start + 1)); // include the newline
|
||||||
|
start = newline_pos + 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
bool newline_ = true;
|
||||||
|
sColour startColour_;
|
||||||
|
sColour currentColour_;
|
||||||
|
};
|
||||||
|
|
||||||
bool execute_local_command(std::string directory_to_run_in, std::string command_to_run, const std::map<std::string, std::string> &env_vars, std::string *output, cMode mode)
|
bool execute_local_command(std::string directory_to_run_in, std::string command_to_run, const std::map<std::string, std::string> &env_vars, std::string *output, cMode mode)
|
||||||
{
|
{
|
||||||
@ -82,51 +118,64 @@ namespace dropshell
|
|||||||
|
|
||||||
if (hasFlag(mode, cMode::Interactive))
|
if (hasFlag(mode, cMode::Interactive))
|
||||||
{
|
{
|
||||||
ASSERT(!hasFlag(mode, cMode::CaptureOutput), "Interactive mode and capture output mode cannot be used together");
|
|
||||||
ASSERT(output == nullptr, "Interactive mode and an output string cannot be used together");
|
ASSERT(output == nullptr, "Interactive mode and an output string cannot be used together");
|
||||||
|
|
||||||
return execute_local_command_interactive(command);
|
return execute_local_command_interactive(command);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (hasFlag(mode, cMode::CaptureOutput))
|
|
||||||
{
|
|
||||||
ASSERT(output != nullptr, "Capture output mode requires an output string to be provided");
|
|
||||||
ASSERT(!hasFlag(mode, cMode::Silent), "Silent mode is not allowed with capture output mode");
|
|
||||||
|
|
||||||
return execute_local_command_and_capture_output(command, output);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (command.get_command_to_run().empty())
|
if (command.get_command_to_run().empty())
|
||||||
return false;
|
return false;
|
||||||
|
|
||||||
bool silent = hasFlag(mode, cMode::Silent);
|
bool silent = hasFlag(mode, cMode::Silent);
|
||||||
std::string full_cmd = command.construct_cmd(localpath::agent()) + " 2>&1" + (silent ? " > /dev/null" : "");
|
|
||||||
int ret = system(full_cmd.c_str());
|
|
||||||
|
|
||||||
bool ok = EXITSTATUSCHECK(ret);
|
std::string full_cmd;
|
||||||
if (!ok && !silent)
|
if (!hasFlag(mode, cMode::NoBB64))
|
||||||
|
full_cmd = command.construct_cmd(localpath::agent()+"/bb64");
|
||||||
|
else
|
||||||
|
full_cmd = command.construct_cmd("");
|
||||||
|
|
||||||
|
if (output != nullptr)
|
||||||
|
full_cmd += " 2>&1"; // capture both stdout and stderr
|
||||||
|
|
||||||
|
FILE *pipe = popen(full_cmd.c_str(), "r");
|
||||||
|
if (!pipe)
|
||||||
{
|
{
|
||||||
std::cerr << "Error: Failed to execute command: " << std::endl;
|
return false;
|
||||||
std::cerr << full_cmd << std::endl;
|
|
||||||
}
|
}
|
||||||
return ok;
|
char buffer[128];
|
||||||
|
fancypinter fancyprint(sColour::DEBUG);
|
||||||
|
while (fgets(buffer, sizeof(buffer), pipe) != nullptr)
|
||||||
|
{
|
||||||
|
if (output != nullptr)
|
||||||
|
(*output) += buffer;
|
||||||
|
|
||||||
|
if (!silent)
|
||||||
|
fancyprint.print(buffer);
|
||||||
|
}
|
||||||
|
int ret = pclose(pipe);
|
||||||
|
return EXITSTATUSCHECK(ret);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ----------------------------------------------------------------------------------------------------------
|
||||||
|
// execute_ssh_command
|
||||||
|
// ----------------------------------------------------------------------------------------------------------
|
||||||
bool execute_ssh_command(const sSSHInfo &ssh_info, const sCommand &remote_command, cMode mode, std::string *output)
|
bool execute_ssh_command(const sSSHInfo &ssh_info, const sCommand &remote_command, cMode mode, std::string *output)
|
||||||
{
|
{
|
||||||
if (remote_command.get_command_to_run().empty())
|
if (remote_command.get_command_to_run().empty())
|
||||||
return false;
|
return false;
|
||||||
|
|
||||||
ASSERT(!(hasFlag(mode, cMode::CaptureOutput) && output == nullptr), "Capture output mode must be used with an output string");
|
|
||||||
|
|
||||||
std::stringstream ssh_cmd;
|
std::stringstream ssh_cmd;
|
||||||
ssh_cmd << "ssh -p " << ssh_info.port << " " << (hasFlag(mode, cMode::Interactive) ? "-tt " : "")
|
ssh_cmd << "ssh -p " << ssh_info.port << " " << (hasFlag(mode, cMode::Interactive) ? "-tt " : "")
|
||||||
<< ssh_info.user << "@" << ssh_info.host;
|
<< ssh_info.user << "@" << ssh_info.host;
|
||||||
|
|
||||||
std::string remote_agent_path = remotepath::agent(ssh_info.server_ID);
|
std::string remote_bb64_path;
|
||||||
|
|
||||||
|
if (!hasFlag(mode, cMode::NoBB64))
|
||||||
|
remote_bb64_path = remotepath::agent(ssh_info.server_ID) + "/bb64";
|
||||||
|
|
||||||
bool rval = execute_local_command(
|
bool rval = execute_local_command(
|
||||||
"", // directory to run in
|
"", // local directory to run in
|
||||||
ssh_cmd.str() + " " + remote_command.construct_cmd(remote_agent_path), // local command to run
|
ssh_cmd.str() + " " + remote_command.construct_cmd(remote_bb64_path), // local command to run
|
||||||
{}, // environment variables
|
{}, // environment variables
|
||||||
output, // output string
|
output, // output string
|
||||||
mode // mode
|
mode // mode
|
||||||
@ -134,26 +183,28 @@ namespace dropshell
|
|||||||
|
|
||||||
if (!rval && !hasFlag(mode, cMode::Silent))
|
if (!rval && !hasFlag(mode, cMode::Silent))
|
||||||
{
|
{
|
||||||
std::cerr << std::endl
|
error << "Error: Failed to execute ssh command:" << std::endl;
|
||||||
<< std::endl;
|
debug << ssh_cmd.str() + " " + remote_command.construct_cmd(remote_bb64_path) << std::endl;
|
||||||
std::cerr << "Error: Failed to execute ssh command:" << std::endl;
|
|
||||||
std::cerr << "\033[90m" << ssh_cmd.str() + " " + remote_command.construct_cmd(remote_agent_path) << "\033[0m" << std::endl;
|
|
||||||
std::cerr << std::endl
|
|
||||||
<< std::endl;
|
|
||||||
}
|
}
|
||||||
return rval;
|
return rval;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string sCommand::makesafecmd(std::string agent_path, const std::string &command) const
|
// ----------------------------------------------------------------------------------------------------------
|
||||||
|
// makesafecmd
|
||||||
|
// ----------------------------------------------------------------------------------------------------------
|
||||||
|
std::string sCommand::makesafecmd(std::string bb64path, const std::string &command) const
|
||||||
{
|
{
|
||||||
if (command.empty())
|
if (command.empty())
|
||||||
return "";
|
return "";
|
||||||
std::string encoded = base64_encode(dequote(trim(command)));
|
std::string encoded = base64_encode(dequote(trim(command)));
|
||||||
std::string commandstr = agent_path + "/bb64 " + encoded;
|
std::string commandstr = bb64path + " " + encoded;
|
||||||
return commandstr;
|
return commandstr;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::string sCommand::construct_cmd(std::string agent_path) const
|
// ----------------------------------------------------------------------------------------------------------
|
||||||
|
// construct_cmd
|
||||||
|
// ----------------------------------------------------------------------------------------------------------
|
||||||
|
std::string sCommand::construct_cmd(std::string bb64path) const
|
||||||
{
|
{
|
||||||
if (mCmd.empty())
|
if (mCmd.empty())
|
||||||
return "";
|
return "";
|
||||||
@ -161,6 +212,8 @@ namespace dropshell
|
|||||||
// need to construct to change directory and set environment variables
|
// need to construct to change directory and set environment variables
|
||||||
std::string cmdstr;
|
std::string cmdstr;
|
||||||
|
|
||||||
|
if (!bb64path.empty())
|
||||||
|
{
|
||||||
if (!mDir.empty())
|
if (!mDir.empty())
|
||||||
cmdstr += "cd " + quote(mDir) + " && ";
|
cmdstr += "cd " + quote(mDir) + " && ";
|
||||||
|
|
||||||
@ -170,8 +223,16 @@ namespace dropshell
|
|||||||
|
|
||||||
cmdstr += mCmd;
|
cmdstr += mCmd;
|
||||||
|
|
||||||
if (!agent_path.empty())
|
cmdstr = makesafecmd(bb64path, cmdstr);
|
||||||
cmdstr = makesafecmd(agent_path, cmdstr);
|
}
|
||||||
|
else
|
||||||
|
{ // raw! bootstrapping only.
|
||||||
|
ASSERT(mVars.empty(), "Bootstrapping command must not have environment variables");
|
||||||
|
if (!mDir.empty())
|
||||||
|
cmdstr += mDir + "/" + mCmd;
|
||||||
|
else
|
||||||
|
cmdstr += mCmd;
|
||||||
|
}
|
||||||
|
|
||||||
return cmdstr;
|
return cmdstr;
|
||||||
}
|
}
|
||||||
|
@ -13,7 +13,7 @@ enum class cMode {
|
|||||||
Defaults = 0,
|
Defaults = 0,
|
||||||
Interactive = 1,
|
Interactive = 1,
|
||||||
Silent = 2,
|
Silent = 2,
|
||||||
CaptureOutput = 4
|
NoBB64 = 4
|
||||||
};
|
};
|
||||||
|
|
||||||
inline cMode operator&(cMode lhs, cMode rhs) {return static_cast<cMode>(static_cast<int>(lhs) & static_cast<int>(rhs));}
|
inline cMode operator&(cMode lhs, cMode rhs) {return static_cast<cMode>(static_cast<int>(lhs) & static_cast<int>(rhs));}
|
||||||
@ -31,7 +31,6 @@ typedef struct sSSHInfo {
|
|||||||
std::string server_ID; // dropshell name for server.
|
std::string server_ID; // dropshell name for server.
|
||||||
} sSSHInfo;
|
} sSSHInfo;
|
||||||
|
|
||||||
bool execute_local_command(std::string command, std::string * output = nullptr, cMode mode = cMode::Defaults);
|
|
||||||
bool execute_local_command(std::string directory_to_run_in, std::string command_to_run, const std::map<std::string, std::string> & env_vars, std::string * output = nullptr, cMode mode = cMode::Defaults);
|
bool execute_local_command(std::string directory_to_run_in, std::string command_to_run, const std::map<std::string, std::string> & env_vars, std::string * output = nullptr, cMode mode = cMode::Defaults);
|
||||||
bool execute_ssh_command(const sSSHInfo & ssh_info, const sCommand & remote_command, cMode mode = cMode::Defaults, std::string * output = nullptr);
|
bool execute_ssh_command(const sSSHInfo & ssh_info, const sCommand & remote_command, cMode mode = cMode::Defaults, std::string * output = nullptr);
|
||||||
|
|
||||||
@ -53,10 +52,10 @@ class sCommand {
|
|||||||
|
|
||||||
bool empty() const { return mCmd.empty(); }
|
bool empty() const { return mCmd.empty(); }
|
||||||
|
|
||||||
std::string construct_cmd(std::string agent_path) const;
|
std::string construct_cmd(std::string bb64path) const;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::string makesafecmd(std::string agent_path, const std::string& command) const;
|
std::string makesafecmd(std::string bb64path, const std::string& command) const;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
std::string mDir;
|
std::string mDir;
|
||||||
|
133
source/src/utils/output.cpp
Normal file
133
source/src/utils/output.cpp
Normal file
@ -0,0 +1,133 @@
|
|||||||
|
#include "output.hpp"
|
||||||
|
#include <iostream>
|
||||||
|
#include <mutex>
|
||||||
|
|
||||||
|
namespace dropshell
|
||||||
|
{
|
||||||
|
// Mutex to synchronize output
|
||||||
|
std::mutex output_mutex;
|
||||||
|
|
||||||
|
// ANSI colour codes
|
||||||
|
constexpr const char *GREY = "\033[90m";
|
||||||
|
constexpr const char *RESET = "\033[0m";
|
||||||
|
constexpr const char *DEBUG_COLOUR = "\033[36m"; // Cyan
|
||||||
|
// constexpr const char *INFO_COLOUR = "\033[32m"; // Green
|
||||||
|
constexpr const char *INFO_COLOUR = "\033[37m"; // White
|
||||||
|
constexpr const char *WARNING_COLOUR = "\033[33m"; // Yellow
|
||||||
|
constexpr const char *ERROR_COLOUR = "\033[31m"; // Red
|
||||||
|
|
||||||
|
// Tag and colour for each stream
|
||||||
|
struct StreamInfo
|
||||||
|
{
|
||||||
|
const char *tag;
|
||||||
|
const char *colour;
|
||||||
|
};
|
||||||
|
|
||||||
|
const StreamInfo stream_infos[] = {
|
||||||
|
{"[DBG]", DEBUG_COLOUR},
|
||||||
|
{"[INF]", INFO_COLOUR},
|
||||||
|
{"[WRN]", WARNING_COLOUR},
|
||||||
|
{"[ERR]", ERROR_COLOUR}};
|
||||||
|
|
||||||
|
// Custom streambuf to prefix and colour each line
|
||||||
|
class PrefixStreambuf : public std::streambuf
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
PrefixStreambuf(std::ostream &dest, const char *tag, const char *colour)
|
||||||
|
: dest_(dest), tag_(tag), colour_(colour), at_line_start_(true) {}
|
||||||
|
|
||||||
|
protected:
|
||||||
|
int overflow(int c) override
|
||||||
|
{
|
||||||
|
std::lock_guard<std::mutex> lock(output_mutex);
|
||||||
|
if (c == EOF)
|
||||||
|
return !EOF;
|
||||||
|
if (at_line_start_ && c != '\n')
|
||||||
|
{
|
||||||
|
dest_ << GREY << tag_ << RESET << ' ' << colour_;
|
||||||
|
at_line_start_ = false;
|
||||||
|
}
|
||||||
|
dest_.put(static_cast<char>(c));
|
||||||
|
if (c == '\n')
|
||||||
|
{
|
||||||
|
dest_ << RESET;
|
||||||
|
at_line_start_ = true;
|
||||||
|
}
|
||||||
|
return c;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::ostream &dest_;
|
||||||
|
const char *tag_;
|
||||||
|
const char *colour_;
|
||||||
|
bool at_line_start_;
|
||||||
|
};
|
||||||
|
|
||||||
|
PrefixStreambuf debug_buf(std::clog, stream_infos[0].tag, stream_infos[0].colour);
|
||||||
|
PrefixStreambuf info_buf(std::clog, stream_infos[1].tag, stream_infos[1].colour);
|
||||||
|
PrefixStreambuf warning_buf(std::clog, stream_infos[2].tag, stream_infos[2].colour);
|
||||||
|
PrefixStreambuf error_buf(std::cerr, stream_infos[3].tag, stream_infos[3].colour);
|
||||||
|
|
||||||
|
std::ostream debug_stream(&debug_buf);
|
||||||
|
std::ostream info_stream(&info_buf);
|
||||||
|
std::ostream warning_stream(&warning_buf);
|
||||||
|
std::ostream error_stream(&error_buf);
|
||||||
|
|
||||||
|
std::ostream &debug = debug_stream;
|
||||||
|
std::ostream &info = info_stream;
|
||||||
|
std::ostream &warning = warning_stream;
|
||||||
|
std::ostream &error = error_stream;
|
||||||
|
|
||||||
|
std::ostream &rawout = std::cout;
|
||||||
|
std::ostream &rawerr = std::cerr;
|
||||||
|
|
||||||
|
std::ostream &colourstream(sColour colour)
|
||||||
|
{
|
||||||
|
switch (colour)
|
||||||
|
{
|
||||||
|
case sColour::DEBUG:
|
||||||
|
return debug_stream;
|
||||||
|
case sColour::INFO:
|
||||||
|
return info_stream;
|
||||||
|
case sColour::WARNING:
|
||||||
|
return warning_stream;
|
||||||
|
case sColour::ERROR:
|
||||||
|
return error_stream;
|
||||||
|
default:
|
||||||
|
return info_stream;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void SetColour(sColour colour, std::ostream &os)
|
||||||
|
{
|
||||||
|
switch (colour)
|
||||||
|
{
|
||||||
|
case sColour::RESET:
|
||||||
|
os << RESET;
|
||||||
|
break;
|
||||||
|
case sColour::DEBUG:
|
||||||
|
os << DEBUG_COLOUR;
|
||||||
|
break;
|
||||||
|
case sColour::INFO:
|
||||||
|
os << INFO_COLOUR;
|
||||||
|
break;
|
||||||
|
case sColour::WARNING:
|
||||||
|
os << WARNING_COLOUR;
|
||||||
|
break;
|
||||||
|
case sColour::ERROR:
|
||||||
|
os << ERROR_COLOUR;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
SwitchColour::SwitchColour(sColour colour, std::ostream &os) : os_(os), colour_(colour)
|
||||||
|
{
|
||||||
|
SetColour(colour_, os_);
|
||||||
|
}
|
||||||
|
|
||||||
|
SwitchColour::~SwitchColour()
|
||||||
|
{
|
||||||
|
SetColour(sColour::RESET, os_);
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace dropshell
|
87
source/src/utils/output.hpp
Normal file
87
source/src/utils/output.hpp
Normal file
@ -0,0 +1,87 @@
|
|||||||
|
#ifndef OUTPUT_HPP
|
||||||
|
#define OUTPUT_HPP
|
||||||
|
|
||||||
|
#include <iostream>
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
#include <ostream>
|
||||||
|
|
||||||
|
namespace dropshell {
|
||||||
|
|
||||||
|
/*
|
||||||
|
|
||||||
|
output.hpp and output.cpp - simple output helpers.
|
||||||
|
|
||||||
|
Defines ostreams:
|
||||||
|
|
||||||
|
debug, info, warning, error.
|
||||||
|
|
||||||
|
These ostreams can be used with C++23 print and println, e.g.
|
||||||
|
std::println(debug, "funny variable: {}={}","my_var",my_var);
|
||||||
|
|
||||||
|
Also defines a few helper functions:
|
||||||
|
|
||||||
|
PrintDebug(const std::string& msg); // equivalent to std::println(debug, msg);
|
||||||
|
PrintInfo(const std::string& msg); // equivalent to std::println(info, msg);
|
||||||
|
PrintWarning(const std::string& msg); // equivalent to std::println(warning, msg);
|
||||||
|
PrintError(const std::string& msg); // equivalent to std::println(error, msg);
|
||||||
|
|
||||||
|
Output for these streams for each line is formatted as:
|
||||||
|
[DBG] <message>
|
||||||
|
[INF] <message>
|
||||||
|
[WRN] <message>
|
||||||
|
[ERR] <message>
|
||||||
|
|
||||||
|
The output is coloured, and the tag is printed in grey.
|
||||||
|
|
||||||
|
In addition, when not using any of the above, helper routines for coloring the output of cout and cerr are provided.
|
||||||
|
|
||||||
|
SetColour(std::ostream& os, sColour colour);
|
||||||
|
|
||||||
|
Where sColour is an enum:
|
||||||
|
enum class sColour {
|
||||||
|
RESET,
|
||||||
|
DEBUG,
|
||||||
|
INFO,
|
||||||
|
WARNING,
|
||||||
|
ERROR
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
*/
|
||||||
|
|
||||||
|
// Output streams for different log levels
|
||||||
|
extern std::ostream& debug;
|
||||||
|
extern std::ostream& info;
|
||||||
|
extern std::ostream& warning;
|
||||||
|
extern std::ostream& error;
|
||||||
|
|
||||||
|
extern std::ostream& rawout;
|
||||||
|
extern std::ostream& rawerr;
|
||||||
|
|
||||||
|
// Enum for colours
|
||||||
|
enum class sColour {
|
||||||
|
RESET,
|
||||||
|
DEBUG,
|
||||||
|
INFO,
|
||||||
|
WARNING,
|
||||||
|
ERROR
|
||||||
|
};
|
||||||
|
std::ostream& colourstream(sColour colour);
|
||||||
|
|
||||||
|
// Set colour for a stream
|
||||||
|
void SetColour(sColour colour, std::ostream& os = std::cerr);
|
||||||
|
|
||||||
|
class SwitchColour
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
SwitchColour(sColour colour, std::ostream& os = std::cerr);
|
||||||
|
~SwitchColour();
|
||||||
|
private:
|
||||||
|
std::ostream& os_;
|
||||||
|
sColour colour_;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace dropshell
|
||||||
|
|
||||||
|
#endif // OUTPUT_HPP
|
@ -9,6 +9,8 @@
|
|||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <map>
|
#include <map>
|
||||||
|
|
||||||
|
#include "output.hpp"
|
||||||
|
|
||||||
enum kTextColors {
|
enum kTextColors {
|
||||||
kTextColor_Default,
|
kTextColor_Default,
|
||||||
kTextColor_Red,
|
kTextColor_Red,
|
||||||
@ -32,7 +34,7 @@ const std::map<std::string, coloredText> kReplacements = {
|
|||||||
{":tick:", {"+", kTextColor_Green}},
|
{":tick:", {"+", kTextColor_Green}},
|
||||||
{":cross:", {"x", kTextColor_Red}},
|
{":cross:", {"x", kTextColor_Red}},
|
||||||
{":warning:", {"!", kTextColor_Yellow}},
|
{":warning:", {"!", kTextColor_Yellow}},
|
||||||
{":info:", {"i", kTextColor_Blue}},
|
{":dropshell::info:", {"i", kTextColor_Blue}},
|
||||||
{":check:", {"+", kTextColor_Green}},
|
{":check:", {"+", kTextColor_Green}},
|
||||||
{":x:", {"x", kTextColor_Red}},
|
{":x:", {"x", kTextColor_Red}},
|
||||||
{":error:", {"!", kTextColor_Red}},
|
{":error:", {"!", kTextColor_Red}},
|
||||||
@ -115,12 +117,12 @@ std::string width_print_centered(std::string str,int width, std::string rowcolor
|
|||||||
oss << rowcolor << std::string(lpad, ' ') << process_cell(str, rowcolor) <<
|
oss << rowcolor << std::string(lpad, ' ') << process_cell(str, rowcolor) <<
|
||||||
std::string(rpad, ' ') << "\033[0m";
|
std::string(rpad, ' ') << "\033[0m";
|
||||||
|
|
||||||
// std::cout << "str = "<<str <<std::endl;
|
// dropshell::info << "str = "<<str <<std::endl;
|
||||||
// std::cout << "width = "<<width <<std::endl;
|
// dropshell::info << "width = "<<width <<std::endl;
|
||||||
// std::cout << "padding = "<<padding <<std::endl;
|
// dropshell::info << "padding = "<<padding <<std::endl;
|
||||||
// std::cout << "get_visible_length(str) = "<<get_visible_length(str) <<std::endl;
|
// dropshell::info << "get_visible_length(str) = "<<get_visible_length(str) <<std::endl;
|
||||||
// std::cout << "get_codepoints(str) = "<<get_codepoints(str) <<std::endl;
|
// dropshell::info << "get_codepoints(str) = "<<get_codepoints(str) <<std::endl;
|
||||||
// std::cout << "oss.str() = ["<<oss.str() <<"]"<<std::endl;
|
// dropshell::info << "oss.str() = ["<<oss.str() <<"]"<<std::endl;
|
||||||
|
|
||||||
return oss.str();
|
return oss.str();
|
||||||
}
|
}
|
||||||
@ -195,98 +197,105 @@ void tableprint::print() {
|
|||||||
|
|
||||||
// Print title if it exists
|
// Print title if it exists
|
||||||
if (!title.empty()) {
|
if (!title.empty()) {
|
||||||
std::cout << "\033[90m"; // Dark grey color for borders
|
dropshell::info << "\033[90m"; // Dark grey color for borders
|
||||||
std::cout << "+";
|
dropshell::info << "+";
|
||||||
for (size_t i = 0; i < rows[0].size(); ++i) {
|
for (size_t i = 0; i < rows[0].size(); ++i) {
|
||||||
std::cout << std::string(col_widths[i] + 2, '-');
|
dropshell::info << std::string(col_widths[i] + 2, '-');
|
||||||
if (i < rows[0].size() - 1) std::cout << "-";
|
if (i < rows[0].size() - 1) dropshell::info << "-";
|
||||||
}
|
}
|
||||||
std::cout << "+" << std::endl;
|
dropshell::info << "+" << std::endl;
|
||||||
|
|
||||||
std::cout << "|"; // White color for title
|
dropshell::info << "\033[90m"; // Dark grey color for borders
|
||||||
|
dropshell::info << "|"; // White color for title
|
||||||
size_t title_width = 0;
|
size_t title_width = 0;
|
||||||
for (size_t width : col_widths) {
|
for (size_t width : col_widths) {
|
||||||
title_width += width + 2; // +2 for padding
|
title_width += width + 2; // +2 for padding
|
||||||
}
|
}
|
||||||
title_width += col_widths.size() - 1; // Add space for vertical borders
|
title_width += col_widths.size() - 1; // Add space for vertical borders
|
||||||
|
|
||||||
std::cout << width_print_centered(title,title_width,"\033[1;37m");
|
dropshell::info << width_print_centered(title,title_width,"\033[1;37m");
|
||||||
std::cout << "\033[90m|" << std::endl;
|
dropshell::info << "\033[90m|" << std::endl;
|
||||||
|
|
||||||
// Use └─┴─┘ for bottom of title box to connect with table
|
// Use └─┴─┘ for bottom of title box to connect with table
|
||||||
std::cout << "+";
|
dropshell::info << "\033[90m"; // Dark grey color for borders
|
||||||
|
dropshell::info << "+";
|
||||||
for (size_t i = 0; i < rows[0].size(); ++i) {
|
for (size_t i = 0; i < rows[0].size(); ++i) {
|
||||||
std::cout << std::string(col_widths[i] + 2, '-');
|
dropshell::info << std::string(col_widths[i] + 2, '-');
|
||||||
if (i < rows[0].size() - 1) std::cout << "-";
|
if (i < rows[0].size() - 1) dropshell::info << "-";
|
||||||
}
|
}
|
||||||
std::cout << "+" << std::endl;
|
dropshell::info << "+" << std::endl;
|
||||||
} else {
|
} else {
|
||||||
// Print top border if no title
|
// Print top border if no title
|
||||||
std::cout << "\033[90m"; // Dark grey color for borders
|
dropshell::info << "\033[90m"; // Dark grey color for borders
|
||||||
std::cout << "+";
|
dropshell::info << "+";
|
||||||
for (size_t i = 0; i < rows[0].size(); ++i) {
|
for (size_t i = 0; i < rows[0].size(); ++i) {
|
||||||
std::cout << std::string(col_widths[i] + 2, '-');
|
dropshell::info << std::string(col_widths[i] + 2, '-');
|
||||||
if (i < rows[0].size() - 1) std::cout << "+";
|
if (i < rows[0].size() - 1) dropshell::info << "+";
|
||||||
}
|
}
|
||||||
std::cout << "+" << std::endl;
|
dropshell::info << "+" << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Print header
|
// Print header
|
||||||
std::cout << "|";
|
dropshell::info << "\033[90m"; // Dark grey color for borders
|
||||||
|
dropshell::info << "|";
|
||||||
for (size_t i = 0; i < rows[0].size(); ++i) {
|
for (size_t i = 0; i < rows[0].size(); ++i) {
|
||||||
std::cout << width_print_centered(rows[0][i],col_widths[i]+2,"\033[1;36m");
|
dropshell::info << width_print_centered(rows[0][i],col_widths[i]+2,"\033[1;36m");
|
||||||
if (i < rows[0].size() - 1) {
|
if (i < rows[0].size() - 1) {
|
||||||
std::cout << "\033[90m|\033[1;36m"; // Border color then back to cyan
|
dropshell::info << "\033[90m|\033[1;36m"; // Border color then back to cyan
|
||||||
} else {
|
} else {
|
||||||
std::cout << "\033[90m|"; // Just border color for last column
|
dropshell::info << "\033[90m|"; // Just border color for last column
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
std::cout << std::endl;
|
dropshell::info << std::endl;
|
||||||
|
|
||||||
// Print header separator
|
// Print header separator
|
||||||
if (!mCompact) {
|
if (!mCompact) {
|
||||||
std::cout << "+";
|
dropshell::info << "\033[90m"; // Dark grey color for borders
|
||||||
|
dropshell::info << "+";
|
||||||
for (size_t i = 0; i < rows[0].size(); ++i) {
|
for (size_t i = 0; i < rows[0].size(); ++i) {
|
||||||
for (size_t j = 0; j < col_widths[i] + 2; ++j) {
|
for (size_t j = 0; j < col_widths[i] + 2; ++j) {
|
||||||
std::cout << "-";
|
dropshell::info << "-";
|
||||||
}
|
}
|
||||||
if (i < rows[0].size() - 1) std::cout << "+";
|
if (i < rows[0].size() - 1) dropshell::info << "+";
|
||||||
}
|
}
|
||||||
std::cout << "+" << std::endl;
|
dropshell::info << "+" << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Print rows
|
// Print rows
|
||||||
for (size_t row_idx = 1; row_idx < rows.size(); ++row_idx) {
|
for (size_t row_idx = 1; row_idx < rows.size(); ++row_idx) {
|
||||||
const auto& row = rows[row_idx];
|
const auto& row = rows[row_idx];
|
||||||
std::cout << "|";
|
dropshell::info << "\033[90m"; // Dark grey color for borders
|
||||||
|
dropshell::info << "|";
|
||||||
for (size_t i = 0; i < row.size(); ++i) {
|
for (size_t i = 0; i < row.size(); ++i) {
|
||||||
// Set the appropriate color for the row
|
// Set the appropriate color for the row
|
||||||
std::string rowcolor = (row_idx % 2 == 1) ? "\033[38;5;142m" : "\033[38;5;250m";
|
std::string rowcolor = (row_idx % 2 == 1) ? "\033[38;5;142m" : "\033[38;5;250m";
|
||||||
std::cout << width_print_left(row[i],col_widths[i]+2,rowcolor);
|
dropshell::info << width_print_left(row[i],col_widths[i]+2,rowcolor);
|
||||||
std::cout << "\033[90m" << "|";
|
dropshell::info << "\033[90m" << "|";
|
||||||
}
|
}
|
||||||
std::cout << std::endl;
|
dropshell::info << std::endl;
|
||||||
|
|
||||||
// Print row separator if not the last row
|
// Print row separator if not the last row
|
||||||
if (row_idx < rows.size() - 1 && !mCompact) {
|
if (row_idx < rows.size() - 1 && !mCompact) {
|
||||||
std::cout << "+";
|
dropshell::info << "\033[90m"; // Dark grey color for borders
|
||||||
|
dropshell::info << "+";
|
||||||
for (size_t i = 0; i < rows[0].size(); ++i) {
|
for (size_t i = 0; i < rows[0].size(); ++i) {
|
||||||
for (size_t j = 0; j < col_widths[i] + 2; ++j) {
|
for (size_t j = 0; j < col_widths[i] + 2; ++j) {
|
||||||
std::cout << "-";
|
dropshell::info << "-";
|
||||||
}
|
}
|
||||||
if (i < rows[0].size() - 1) std::cout << "+";
|
if (i < rows[0].size() - 1) dropshell::info << "+";
|
||||||
}
|
}
|
||||||
std::cout << "+" << std::endl;
|
dropshell::info << "+" << std::endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Print bottom border
|
// Print bottom border
|
||||||
std::cout << "+";
|
dropshell::info << "\033[90m"; // Dark grey color for borders
|
||||||
|
dropshell::info << "+";
|
||||||
for (size_t i = 0; i < rows[0].size(); ++i) {
|
for (size_t i = 0; i < rows[0].size(); ++i) {
|
||||||
for (size_t j = 0; j < col_widths[i] + 2; ++j) {
|
for (size_t j = 0; j < col_widths[i] + 2; ++j) {
|
||||||
std::cout << "-";
|
dropshell::info << "-";
|
||||||
}
|
}
|
||||||
if (i < rows[0].size() - 1) std::cout << "+";
|
if (i < rows[0].size() - 1) dropshell::info << "+";
|
||||||
}
|
}
|
||||||
std::cout << "+" << "\033[0m" << std::endl; // Reset color
|
dropshell::info << "+" << "\033[0m" << std::endl; // Reset color
|
||||||
}
|
}
|
||||||
|
@ -10,10 +10,20 @@
|
|||||||
|
|
||||||
namespace dropshell {
|
namespace dropshell {
|
||||||
|
|
||||||
void maketitle(const std::string& title) {
|
std::string magic_string() {
|
||||||
std::cout << std::string(title.length() + 4, '-') << std::endl;
|
return "-_-";
|
||||||
std::cout << "| " << title << " |" << std::endl;
|
}
|
||||||
std::cout << std::string(title.length() + 4, '-') << std::endl;
|
|
||||||
|
bool has_magic_string(std::string name)
|
||||||
|
{
|
||||||
|
return name.find(magic_string()) != std::string::npos;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void maketitle(const std::string& title, sColour colour) {
|
||||||
|
colourstream(colour) << std::string(title.length() + 4, '-') << std::endl;
|
||||||
|
colourstream(colour) << "| " << title << " |" << std::endl;
|
||||||
|
colourstream(colour) << std::string(title.length() + 4, '-') << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool replace_line_in_file(const std::string& file_path, const std::string& search_string, const std::string& replacement_line) {
|
bool replace_line_in_file(const std::string& file_path, const std::string& search_string, const std::string& replacement_line) {
|
||||||
@ -334,20 +344,6 @@ std::string safearg(int argc, char *argv[], int index)
|
|||||||
return argv[index];
|
return argv[index];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void print_left_aligned(const std::string & str, int width) {
|
|
||||||
std::cout << left_align(str, width);
|
|
||||||
}
|
|
||||||
|
|
||||||
void print_centered(const std::string & str, int width) {
|
|
||||||
std::cout << center_align(str, width);
|
|
||||||
}
|
|
||||||
|
|
||||||
void print_right_aligned(const std::string & str, int width) {
|
|
||||||
std::cout << right_align(str, width);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
std::string left_align(const std::string & str, int width) {
|
std::string left_align(const std::string & str, int width) {
|
||||||
if (static_cast<int>(str.size()) >= width)
|
if (static_cast<int>(str.size()) >= width)
|
||||||
return str;
|
return str;
|
||||||
|
@ -3,6 +3,8 @@
|
|||||||
#include <string>
|
#include <string>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
|
#include "output.hpp"
|
||||||
|
|
||||||
namespace dropshell {
|
namespace dropshell {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -10,10 +12,12 @@ namespace dropshell {
|
|||||||
*
|
*
|
||||||
* @param title The title string to display
|
* @param title The title string to display
|
||||||
*/
|
*/
|
||||||
void maketitle(const std::string& title);
|
void maketitle(const std::string& title, sColour colour=sColour::INFO);
|
||||||
|
|
||||||
bool replace_line_in_file(const std::string& file_path, const std::string& search_string, const std::string& replacement_line);
|
bool replace_line_in_file(const std::string& file_path, const std::string& search_string, const std::string& replacement_line);
|
||||||
|
|
||||||
|
std::string magic_string();
|
||||||
|
bool has_magic_string(std::string name);
|
||||||
|
|
||||||
// utility functions
|
// utility functions
|
||||||
std::string trim(std::string str);
|
std::string trim(std::string str);
|
||||||
@ -45,10 +49,6 @@ int die(const std::string & msg);
|
|||||||
std::string safearg(int argc, char *argv[], int index);
|
std::string safearg(int argc, char *argv[], int index);
|
||||||
std::string safearg(const std::vector<std::string> & args, int index);
|
std::string safearg(const std::vector<std::string> & args, int index);
|
||||||
|
|
||||||
void print_left_aligned(const std::string & str, int width);
|
|
||||||
void print_centered(const std::string & str, int width);
|
|
||||||
void print_right_aligned(const std::string & str, int width);
|
|
||||||
|
|
||||||
std::string left_align(const std::string & str, int width);
|
std::string left_align(const std::string & str, int width);
|
||||||
std::string right_align(const std::string & str, int width);
|
std::string right_align(const std::string & str, int width);
|
||||||
std::string center_align(const std::string & str, int width);
|
std::string center_align(const std::string & str, int width);
|
||||||
|
Reference in New Issue
Block a user