136 lines
3.7 KiB
Go
136 lines
3.7 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"git.towk2.me/towk/makeshift/pkg/client"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var downloadCmd = cobra.Command{
|
|
Use: "download",
|
|
Example: `
|
|
# set up environment
|
|
export MAKESHIFT_HOST=http://localhost:5050
|
|
export MAKESHIFT_PATH=test
|
|
|
|
# download a file or directory (as archive)
|
|
makeshift download
|
|
makeshift download --host http://localhost:5050.com --path test
|
|
|
|
# download a file or directory and run plugins with profile data
|
|
makeshift download --plugins smd,jinja2 --profile compute
|
|
curl $MAKESHIFT_HOST/download/test?plugins=smd,jinja2&profile=test
|
|
|
|
# download directory and extract it's contents automatically
|
|
makeshift download --extract
|
|
`,
|
|
Short: "Download and modify files with plugins",
|
|
PreRun: func(cmd *cobra.Command, args []string) {
|
|
setenv(cmd, "host", "MAKESHIFT_HOST")
|
|
setenv(cmd, "path", "MAKESHIFT_PATH")
|
|
},
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
var (
|
|
host, _ = cmd.Flags().GetString("host")
|
|
path, _ = cmd.Flags().GetString("path")
|
|
outputPath, _ = cmd.Flags().GetString("output")
|
|
pluginNames, _ = cmd.Flags().GetStringSlice("plugins")
|
|
profileIDs, _ = cmd.Flags().GetStringSlice("profiles")
|
|
|
|
c = client.New(host)
|
|
res *http.Response
|
|
query string
|
|
body []byte
|
|
err error
|
|
)
|
|
|
|
// set output path to match path if empty
|
|
if outputPath == "" {
|
|
if path != "." || path != "" {
|
|
outputPath = filepath.Base(path)
|
|
} else {
|
|
outputPath = fmt.Sprintf("%d.file", time.Now().Unix())
|
|
}
|
|
}
|
|
|
|
query = fmt.Sprintf("/download/%s?", path)
|
|
if len(pluginNames) > 0 {
|
|
query += "plugins=" + url.QueryEscape(strings.Join(pluginNames, ","))
|
|
}
|
|
if len(profileIDs) > 0 {
|
|
query += "&profiles=" + url.QueryEscape(strings.Join(profileIDs, ","))
|
|
}
|
|
|
|
log.Debug().
|
|
Str("host", host).
|
|
Str("path", path).
|
|
Str("query", query).
|
|
Str("output", outputPath).
|
|
Strs("profiles", profileIDs).
|
|
Strs("plugins", pluginNames).
|
|
Send()
|
|
|
|
res, body, err = c.MakeRequest(client.HTTPEnvelope{
|
|
Path: query,
|
|
Method: http.MethodGet,
|
|
})
|
|
if err != nil {
|
|
log.Error().Err(err).
|
|
Str("host", host).
|
|
Str("path", path).
|
|
Str("output", outputPath).
|
|
Msg("failed to make request")
|
|
os.Exit(1)
|
|
}
|
|
if res.StatusCode != http.StatusOK {
|
|
log.Error().
|
|
Any("status", map[string]any{
|
|
"code": res.StatusCode,
|
|
"message": res.Status,
|
|
}).
|
|
Str("host", host).
|
|
Str("path", path).
|
|
Str("output", outputPath).
|
|
Msg("response returned bad status")
|
|
os.Exit(1)
|
|
}
|
|
if outputPath != "" {
|
|
err = os.WriteFile(outputPath, body, 0o755)
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to write file(s) from download")
|
|
os.Exit(1)
|
|
}
|
|
} else {
|
|
fmt.Println(string(body))
|
|
}
|
|
|
|
},
|
|
}
|
|
|
|
var downloadProfileCmd = &cobra.Command{
|
|
Use: "profile",
|
|
}
|
|
var downloadPluginCmd = &cobra.Command{
|
|
Use: "plugin",
|
|
}
|
|
|
|
func init() {
|
|
downloadCmd.Flags().String("host", "http://localhost:5050", "Set the makeshift remote host (can be set with MAKESHIFT_HOST)")
|
|
downloadCmd.Flags().StringP("path", "p", ".", "Set the path to list files (can be set with MAKESHIFT_PATH)")
|
|
downloadCmd.Flags().StringP("output", "o", "", "Set the output path to write files")
|
|
downloadCmd.Flags().StringSlice("profiles", []string{}, "Set the profile to use to populate data store")
|
|
downloadCmd.Flags().StringSlice("plugins", []string{}, "Set the plugins to run before downloading files")
|
|
downloadCmd.Flags().Bool("extract", false, "Set whether to extract archive locally after downloading")
|
|
|
|
downloadCmd.AddCommand(downloadProfileCmd, downloadPluginCmd)
|
|
|
|
rootCmd.AddCommand(&downloadCmd)
|
|
}
|