package buildfile

import (
	"bytes"
	"fmt"
)

type Buildfile struct {
	bytes.Buffer
}

func New() *Buildfile {
	b := Buildfile{}
	b.WriteString(base)
	return &b
}

// WriteCmd writes a command to the build file. The
// command will be echoed back as a base16 encoded
// command so that it can be parsed and appended to
// the build output
func (b *Buildfile) WriteCmd(command string) {
	// echo the command as an encoded value
	b.WriteString(fmt.Sprintf("echo '#DRONE:%x'\n", command))
	// and then run the command
	b.WriteString(fmt.Sprintf("%s\n", command))
}

// WriteCmdSilent writes a command to the build file
// but does not echo the command.
func (b *Buildfile) WriteCmdSilent(command string) {
	b.WriteString(fmt.Sprintf("%s\n", command))
}

// WriteComment adds a comment to the build file. This
// is really only used internally for debugging purposes.
func (b *Buildfile) WriteComment(comment string) {
	b.WriteString(fmt.Sprintf("#%s\n", comment))
}

// WriteEnv exports the environment variable as
// part of the script. The environment variables
// are not echoed back to the console, and are
// kept private by default.
func (b *Buildfile) WriteEnv(key, value string) {
	b.WriteString(fmt.Sprintf("export %s=%q\n", key, value))
}

// WriteHost adds an entry to the /etc/hosts file.
func (b *Buildfile) WriteHost(mapping string) {
	b.WriteCmdSilent(fmt.Sprintf("[ -f /usr/bin/sudo ] || echo %q | tee -a /etc/hosts", mapping))
	b.WriteCmdSilent(fmt.Sprintf("[ -f /usr/bin/sudo ] && echo %q | sudo tee -a /etc/hosts", mapping))
}

// WriteFile add files as part of the script.
func (b *Buildfile) WriteFile(path string, file []byte, i int) {
	b.WriteString(fmt.Sprintf("echo '%s' | tee %s > /dev/null\n", string(file), path))
	b.WriteCmdSilent(fmt.Sprintf("chmod %d %s", i, path))
}

// every build script starts with the following
// code at the start.
var base = `
#!/bin/bash
set +e

# drone configuration files are stored in /etc/drone.d
# execute these files prior to our build to set global
# environment variables and initialize programs (like rbenv)
if [ -d /etc/drone.d ]; then
  for i in /etc/drone.d/*.sh; do
    if [ -r $i ]; then
      . $i
    fi
  done
  unset i
fi

if [ ! -d $HOME/.ssh ]; then
  mkdir -p $HOME/.ssh
fi

chmod 0700 $HOME/.ssh
echo 'StrictHostKeyChecking no' | tee $HOME/.ssh/config > /dev/null

# be sure to exit on error and print out
# our bash commands, so we can see which commands
# are executing and troubleshoot failures.
set -e

# user-defined commands below ##############################
`