mirror of
https://github.com/charmbracelet/glow
synced 2024-12-14 14:12:27 +00:00
133 lines
2.8 KiB
Go
133 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/charmbracelet/charm"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var (
|
|
identityFile string
|
|
forceKey bool
|
|
memo string
|
|
stashCmd = &cobra.Command{
|
|
Use: "stash SOURCE",
|
|
Hidden: false,
|
|
Short: "stash a markdown",
|
|
Long: "",
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
cc := initCharmClient()
|
|
f, err := os.Open(args[0])
|
|
if err != nil {
|
|
return fmt.Errorf("bad filename")
|
|
}
|
|
defer f.Close()
|
|
b, err := ioutil.ReadAll(f)
|
|
if err != nil {
|
|
return fmt.Errorf("error reading file")
|
|
}
|
|
err = cc.StashMarkdown(memo, string(b))
|
|
if err != nil {
|
|
return fmt.Errorf("error stashing markdown")
|
|
}
|
|
fmt.Println("Stashed!")
|
|
return nil
|
|
},
|
|
}
|
|
|
|
stashListCmd = &cobra.Command{
|
|
Use: "stash-list",
|
|
Hidden: false,
|
|
Short: "list your stashed markdowns",
|
|
Long: "",
|
|
Args: cobra.NoArgs,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
cc := initCharmClient()
|
|
mds, err := cc.GetStash(1)
|
|
if err != nil {
|
|
return fmt.Errorf("error getting stash")
|
|
}
|
|
fmt.Println("ID\tNote")
|
|
fmt.Println("--\t----")
|
|
for _, md := range mds {
|
|
fmt.Printf("%d\t%s\n", md.ID, md.Note)
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
|
|
stashGetCmd = &cobra.Command{
|
|
Use: "stash-get",
|
|
Hidden: false,
|
|
Short: "get a stashed markdown by id",
|
|
Long: "",
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
id, err := strconv.Atoi(args[0])
|
|
if err != nil {
|
|
return fmt.Errorf("invalid markdown id")
|
|
}
|
|
cc := initCharmClient()
|
|
md, err := cc.GetStashMarkdown(id)
|
|
if err != nil {
|
|
return fmt.Errorf("error getting markdown")
|
|
}
|
|
fmt.Println(md.Body)
|
|
return nil
|
|
},
|
|
}
|
|
|
|
stashDeleteCmd = &cobra.Command{
|
|
Use: "stash-delete",
|
|
Hidden: false,
|
|
Short: "get a stashed markdown by id",
|
|
Long: "",
|
|
Args: cobra.ExactArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
id, err := strconv.Atoi(args[0])
|
|
if err != nil {
|
|
return fmt.Errorf("invalid markdown id")
|
|
}
|
|
cc := initCharmClient()
|
|
err = cc.DeleteMarkdown(id)
|
|
if err != nil {
|
|
return fmt.Errorf("error deleting markdown")
|
|
}
|
|
fmt.Println("Deleted!")
|
|
return nil
|
|
},
|
|
}
|
|
)
|
|
|
|
func getCharmConfig() *charm.Config {
|
|
cfg, err := charm.ConfigFromEnv()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if identityFile != "" {
|
|
cfg.SSHKeyPath = identityFile
|
|
cfg.ForceKey = true
|
|
}
|
|
if forceKey {
|
|
cfg.ForceKey = true
|
|
}
|
|
return cfg
|
|
}
|
|
|
|
func initCharmClient() *charm.Client {
|
|
cfg := getCharmConfig()
|
|
cc, err := charm.NewClient(cfg)
|
|
if err == charm.ErrMissingSSHAuth {
|
|
log.Fatal("Missing ssh key. Run `ssh-keygen` to make one or set the `CHARM_SSH_KEY_PATH` env var to your private key path.")
|
|
} else if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
return cc
|
|
}
|