mirror of
https://github.com/cross-the-world/scp-pipeline.git
synced 2025-09-11 12:04:42 +08:00
101 lines
2.7 KiB
Bash
101 lines
2.7 KiB
Bash
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
createKeyFile() {
|
|
local SSH_PATH="$HOME/.ssh"
|
|
|
|
mkdir -p "$SSH_PATH"
|
|
touch "$SSH_PATH/known_hosts"
|
|
|
|
echo "$INPUT_KEY" > "$SSH_PATH/id_rsa"
|
|
|
|
chmod 700 "$SSH_PATH"
|
|
chmod 600 "$SSH_PATH/known_hosts"
|
|
chmod 600 "$SSH_PATH/id_rsa"
|
|
|
|
eval $(ssh-agent)
|
|
ssh-add "$SSH_PATH/id_rsa"
|
|
|
|
ssh-keyscan -t rsa "$INPUT_HOST" >> "$SSH_PATH/known_hosts"
|
|
}
|
|
|
|
check_remote_dir() {
|
|
local USEPASS=$1
|
|
local REMOTE=$2
|
|
CMD="ssh"
|
|
if $USEPASS; then
|
|
CMD="sshpass -p $INPUT_PASS scp"
|
|
fi
|
|
echo "Checking remote directory: '$REMOTE'"
|
|
if $CMD -o StrictHostKeyChecking=no -o ConnectTimeout=${INPUT_CONNECT_TIMEOUT:-30s} -p "${INPUT_PORT:-22}" "$INPUT_USER"@"$INPUT_HOST" "[ ! -d $REMOTE ]"; then
|
|
echo "Creating: '$REMOTE' on '$INPUT_USER'@'$INPUT_HOST'"
|
|
$CMD -o StrictHostKeyChecking=no -o ConnectTimeout=${INPUT_CONNECT_TIMEOUT:-30s} -p "${INPUT_PORT:-22}" "$INPUT_USER"@"$INPUT_HOST" "mkdir -p $REMOTE"
|
|
else
|
|
echo "'$REMOTE' exists [OK]"
|
|
fi
|
|
}
|
|
|
|
executeSCP() {
|
|
local USEPASS=$1
|
|
local LINES=$2
|
|
local COMMAND=
|
|
|
|
CMD="scp"
|
|
if $USEPASS; then
|
|
CMD="sshpass -p $INPUT_PASS scp"
|
|
fi
|
|
|
|
while IFS= read -r LINE; do
|
|
delimiter="=>"
|
|
LINE=$(echo $LINE)
|
|
if [[ -z "${LINE}" ]]; then
|
|
continue
|
|
fi
|
|
s=$LINE$delimiter
|
|
arr=()
|
|
while [[ $s ]]; do
|
|
arr+=( "${s%%"$delimiter"*}" );
|
|
s=${s#*"$delimiter"};
|
|
done;
|
|
LOCAL=$(eval 'echo "${arr[0]}"')
|
|
LOCAL=$(eval echo "$LOCAL")
|
|
REMOTE=$(eval 'echo "${arr[1]}"')
|
|
REMOTE=$(eval echo "$REMOTE")
|
|
|
|
if [[ -z "${LOCAL}" ]] || [[ -z "${REMOTE}" ]]; then
|
|
echo "LOCAL/REMOTE can not be parsed $LINE"
|
|
else
|
|
check_remote_dir $USEPASS $REMOTE
|
|
echo "Copying $LOCAL ---> $REMOTE"
|
|
$CMD -r -o StrictHostKeyChecking=no -o ConnectTimeout=${INPUT_CONNECT_TIMEOUT:-30s} -P "${INPUT_PORT:-22}" $LOCAL "$INPUT_USER"@"$INPUT_HOST":$REMOTE > /dev/stdout
|
|
fi
|
|
done <<< "$LINES"
|
|
}
|
|
|
|
|
|
######################################################################################
|
|
|
|
echo "+++++++++++++++++++STARTING PIPELINES+++++++++++++++++++"
|
|
|
|
USEPASS=true
|
|
if [[ -z "${INPUT_KEY}" ]]; then
|
|
echo "+++++++++++++++++++Use password+++++++++++++++++++"
|
|
else
|
|
echo "+++++++++++++++++++Create Key File+++++++++++++++++++"
|
|
USEPASS=false
|
|
createKeyFile || false
|
|
fi
|
|
|
|
if ! [[ -z "${INPUT_LOCAL}" ]] && ! [[ -z "${INPUT_REMOTE}" ]]; then
|
|
echo "+++++++++++++++++++Pipeline: LOCAL -> REMOTE+++++++++++++++++++"
|
|
executeSCP "$USEPASS" "$INPUT_LOCAL => $INPUT_REMOTE" || false
|
|
fi
|
|
|
|
if ! [[ -z "${INPUT_SCP}" ]]; then
|
|
echo "+++++++++++++++++++Pipeline: RUNNING SCP+++++++++++++++++++"
|
|
executeSCP "$USEPASS" "$INPUT_SCP" || false
|
|
fi
|
|
|
|
echo "+++++++++++++++++++END PIPELINES+++++++++++++++++++"
|