mirror of
https://github.com/aptly-dev/aptly.git
synced 2026-06-02 04:50:49 +00:00
Simon Aquino
18e3ed5d64
This new aptly command will allow to run multiple commands within a single aptly command, running in a single thread. The commands can be included in a text file or added to the aptly script run command in one line, separated by comas ','. If one command returns an error, then all the subsequent commands will be skipped. Each command output will appear on the console within coloured strings, clearly stating which command is running and the start and the end of the received output.
152 lines
2.9 KiB
Go
152 lines
2.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/mattn/go-shellwords"
|
|
"github.com/smira/commander"
|
|
"github.com/wsxiaoys/terminal/color"
|
|
)
|
|
|
|
func aptlyScriptRun(cmd *commander.Command, args []string) error {
|
|
|
|
var err error
|
|
var cmd_list [][]string
|
|
|
|
if len(args) == 0 {
|
|
var text string
|
|
cmd_args := []string{}
|
|
|
|
fmt.Println("One command per line and press enter when finished.")
|
|
|
|
reader := bufio.NewReader(os.Stdin)
|
|
for {
|
|
fmt.Printf("> ")
|
|
text, _ = reader.ReadString('\n')
|
|
if text == "\n" {
|
|
break
|
|
} else {
|
|
text = strings.TrimSpace(text) + ","
|
|
parsed_args, _ := shellwords.Parse(text)
|
|
cmd_args = append(cmd_args, parsed_args...)
|
|
}
|
|
}
|
|
|
|
if len(cmd_args) == 0 {
|
|
return fmt.Errorf("Nothing entered. Exiting...\n")
|
|
}
|
|
|
|
cmd_list = formatCommands(cmd_args)
|
|
|
|
} else if len(args) == 1 {
|
|
var text string
|
|
cmd_args := []string{}
|
|
|
|
if finfo, err := os.Stat(args[0]); os.IsNotExist(err) || finfo.IsDir() {
|
|
return fmt.Errorf("No such file, %s\n", args[0])
|
|
}
|
|
|
|
fmt.Println("Reading file...\n")
|
|
|
|
file, err := os.Open(args[0])
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
scanner := bufio.NewScanner(file)
|
|
|
|
for scanner.Scan() {
|
|
text = strings.TrimSpace(scanner.Text()) + ","
|
|
parsed_args, _ := shellwords.Parse(text)
|
|
cmd_args = append(cmd_args, parsed_args...)
|
|
}
|
|
|
|
if err = scanner.Err(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(cmd_args) == 0 {
|
|
return fmt.Errorf("The file is empty. Exiting...\n")
|
|
}
|
|
|
|
cmd_list = formatCommands(cmd_args)
|
|
|
|
} else if len(args) > 1 {
|
|
cmd_list = formatCommands(args)
|
|
}
|
|
|
|
switchContext()
|
|
|
|
for i, command := range cmd_list {
|
|
|
|
if !context.panicked {
|
|
color.Printf("@g%d) [Running]: %s@!\n", (i + 1), strings.Join(command, " "))
|
|
color.Println("\n@yBegin command output: ----------------------------\n@!")
|
|
Run(command, false)
|
|
color.Println("\n@yEnd command output: ------------------------------\n@!")
|
|
|
|
} else {
|
|
color.Printf("@r%d) [Skipping]: %s@!\n", (i + 1), strings.Join(command, " "))
|
|
}
|
|
|
|
}
|
|
|
|
if context.panicked {
|
|
err = fmt.Errorf("At least one command has reported an error\n")
|
|
}
|
|
|
|
switchContext()
|
|
|
|
return err
|
|
}
|
|
|
|
func formatCommands(args []string) [][]string {
|
|
|
|
var cmd []string
|
|
var cmd_array [][]string
|
|
|
|
for _, s := range args {
|
|
if s_trimmed := strings.TrimRight(s, ","); s_trimmed != s {
|
|
cmd = append(cmd, s_trimmed)
|
|
cmd_array = append(cmd_array, cmd)
|
|
cmd = []string{}
|
|
} else {
|
|
cmd = append(cmd, s)
|
|
}
|
|
}
|
|
|
|
if len(cmd) > 0 {
|
|
cmd_array = append(cmd_array, cmd)
|
|
}
|
|
|
|
return cmd_array
|
|
}
|
|
|
|
func makeCmdScriptRun() *commander.Command {
|
|
cmd := &commander.Command{
|
|
Run: aptlyScriptRun,
|
|
UsageLine: "run <filename> | <command1>, <command2>, ...",
|
|
Short: "run aptly scripts",
|
|
Long: `
|
|
Command runs multiple aptly commands in one single aptly thread.
|
|
|
|
Example:
|
|
|
|
$ aptly script run
|
|
> repo create local
|
|
> repo add local pkg1
|
|
> publish repo local
|
|
> serve
|
|
>
|
|
|
|
`,
|
|
}
|
|
|
|
return cmd
|
|
}
|