forked from mirrors/ntfy
58 lines
1.5 KiB
Go
58 lines
1.5 KiB
Go
|
package cmd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/urfave/cli/v2"
|
||
|
|
"heckel.io/ntfy/client"
|
||
|
|
"heckel.io/ntfy/util"
|
||
|
|
"log"
|
||
|
|
"os"
|
||
|
|
"os/exec"
|
||
|
|
"os/user"
|
||
|
|
"path/filepath"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
defaultClientRootConfigFile = "/etc/ntfy/client.yml"
|
||
|
|
defaultClientUserConfigFileRelative = "ntfy/client.yml"
|
||
|
|
defaultClientConfigFileDescriptionSuffix = `The default config file for all client commands is /etc/ntfy/client.yml (if root user),
|
||
|
|
or ~/.config/ntfy/client.yml for all other users.`
|
||
|
|
)
|
||
|
|
|
||
|
|
func runCommandInternal(c *cli.Context, command string, m *client.Message) error {
|
||
|
|
scriptFile, err := createTmpScript(command)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
defer os.Remove(scriptFile)
|
||
|
|
verbose := c.Bool("verbose")
|
||
|
|
if verbose {
|
||
|
|
log.Printf("[%s] Executing: %s (for message: %s)", util.ShortTopicURL(m.TopicURL), command, m.Raw)
|
||
|
|
}
|
||
|
|
cmd := exec.Command("sh", "-c", scriptFile)
|
||
|
|
cmd.Stdin = c.App.Reader
|
||
|
|
cmd.Stdout = c.App.Writer
|
||
|
|
cmd.Stderr = c.App.ErrWriter
|
||
|
|
cmd.Env = envVars(m)
|
||
|
|
return cmd.Run()
|
||
|
|
}
|
||
|
|
|
||
|
|
func createTmpScript(command string) (string, error) {
|
||
|
|
scriptFile := fmt.Sprintf("%s/ntfy-subscribe-%s.sh.tmp", os.TempDir(), util.RandomString(10))
|
||
|
|
script := fmt.Sprintf("#!/bin/sh\n%s", command)
|
||
|
|
if err := os.WriteFile(scriptFile, []byte(script), 0700); err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
return scriptFile, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func defaultConfigFile() string {
|
||
|
|
u, _ := user.Current()
|
||
|
|
configFile := defaultClientRootConfigFile
|
||
|
|
if u.Uid != "0" {
|
||
|
|
homeDir, _ := os.UserConfigDir()
|
||
|
|
return filepath.Join(homeDir, defaultClientUserConfigFileRelative)
|
||
|
|
}
|
||
|
|
return configFile
|
||
|
|
}
|