98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
// Copyright 2019 Drone.IO Inc. All rights reserved.
|
|
// Use of this source code is governed by the Polyform License
|
|
// that can be found in the LICENSE file.
|
|
|
|
// Package shell provides functions for converting shell commands
|
|
// to posix shell scripts.
|
|
package shell
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// Script converts a slice of individual shell commands to
|
|
// a posix-compliant shell script.
|
|
func Script(commands []string) string {
|
|
buf := new(bytes.Buffer)
|
|
fmt.Fprintln(buf)
|
|
fmt.Fprintf(buf, optionScript)
|
|
fmt.Fprintf(buf, tmateScript)
|
|
fmt.Fprintln(buf)
|
|
for _, command := range commands {
|
|
escaped := fmt.Sprintf("%q", command)
|
|
escaped = strings.Replace(escaped, "$", `\$`, -1)
|
|
buf.WriteString(fmt.Sprintf(
|
|
traceScript,
|
|
escaped,
|
|
command,
|
|
))
|
|
}
|
|
return buf.String()
|
|
}
|
|
|
|
// optionScript is a helper script this is added to the build
|
|
// to set shell options, in this case, to exit on error.
|
|
const optionScript = `
|
|
if [ ! -z "${DRONE_NETRC_FILE}" ]; then
|
|
echo $DRONE_NETRC_FILE > $HOME/.netrc
|
|
chmod 600 $HOME/.netrc
|
|
fi
|
|
|
|
unset DRONE_SCRIPT
|
|
unset DRONE_NETRC_MACHINE
|
|
unset DRONE_NETRC_USERNAME
|
|
unset DRONE_NETRC_PASSWORD
|
|
unset DRONE_NETRC_FILE
|
|
|
|
set -e
|
|
`
|
|
|
|
// traceScript is a helper script that is added to
|
|
// the build script to trace a command.
|
|
const traceScript = `
|
|
echo + %s
|
|
%s
|
|
`
|
|
|
|
const tmateScript = `
|
|
remote_debug() {
|
|
if [ "$?" -ne "0" ];
|
|
then
|
|
|
|
if command -v apt-get &> /dev/null
|
|
then
|
|
apt-get update -qq
|
|
apt-get install xz-utils --assume-yes -qq
|
|
fi
|
|
|
|
rm -rf tmate-2.4.0-static-linux-amd64.tar.xz
|
|
wget https://github.com/tmate-io/tmate/releases/download/2.4.0/tmate-2.4.0-static-linux-amd64.tar.xz
|
|
tar -xf tmate-2.4.0-static-linux-amd64.tar.xz
|
|
mv tmate-2.4.0-static-linux-amd64/tmate /usr/bin/
|
|
chmod +x /usr/bin/tmate
|
|
rm -rf tmate-2.4.0-static-linux-amd64
|
|
tmate -F
|
|
|
|
fi
|
|
}
|
|
|
|
if [ "${DRONE_BUILD_DEBUG}" = "true" ]; then
|
|
trap remote_debug EXIT
|
|
fi
|
|
`
|
|
|
|
// TODO(bradrydzewski) add timeout 30m to tmate
|
|
|
|
//
|
|
// TODO(bradrydzewski) support custom tmate config options:
|
|
//
|
|
// if [ ! -z "${DRONE_TMATE_SERVER_HOST}" ]; then
|
|
// echo "set -g tmate-server-host \"$DRONE_TMATE_SERVER_HOST\"" >> $HOME/.tmate.conf
|
|
// echo "set -g tmate-server-port $DRONE_TMATE_SERVER_PORT" >> $HOME/.tmate.conf
|
|
// echo "set -g tmate-server-rsa-fingerprint \"$DRONE_TMATE_SERVER_RSA_FINGERPRINT\"" >> $HOME/.tmate.conf
|
|
// echo "set -g tmate-server-ed25519-fingerprint \"$DRONE_TMATE_SERVER_ED25519_FINGERPRINT\"" >> $HOME/.tmate.conf
|
|
// fi
|
|
//
|