cli/cmd/upload.go

78 lines
1.4 KiB
Go
Raw Normal View History

2021-03-27 03:47:13 +00:00
package cmd
import (
2021-03-27 04:07:32 +00:00
"errors"
"github.com/spf13/cobra"
"github.com/spf13/viper"
"io/ioutil"
"math"
"os"
"paste.ee/go"
"path"
2021-03-27 03:47:13 +00:00
)
// uploadCommandHandler handles the root command from Cobra
// The program takes two inputs, an arg list (files), or stdin.
// If args is empty, stdin will be used.
func uploadCommandHandler(cmd *cobra.Command, args []string) {
2021-03-27 04:07:32 +00:00
client := pastee.New(viper.GetString("apiKey"))
paste := &pastee.Paste{
Encrypted: viper.GetBool("encrypt"),
}
if len(args) > 0 && args[0] != "-" {
paste.Sections = make([]*pastee.Section, int(math.Min(float64(len(args)), 10)))
// File list
for i, filePath := range args {
if i >= 10 {
break
}
bytes, err := ioutil.ReadFile(filePath)
if err != nil {
cmd.PrintErrln("Unable to read file:", err)
os.Exit(1)
}
paste.Sections[i] = &pastee.Section{
Name: path.Base(filePath),
Contents: string(bytes),
}
}
} else {
// Stdin
stdinContents, err := ioutil.ReadAll(os.Stdin)
if err != nil {
cmd.PrintErrln("Error reading from stdin:", err)
os.Exit(1)
}
paste.Sections = []*pastee.Section{
{Contents: string(stdinContents)},
}
}
res, err := client.Submit(paste)
if err != nil || res.Link == "" {
if err == nil {
err = errors.New("unknown error")
}
cmd.PrintErrln("Unable to upload paste:", err)
os.Exit(1)
}
link := res.Link
if res.Key != "" {
link += "#" + res.Key
}
cmd.Println(link)
}