137 lines
2.9 KiB
Go
137 lines
2.9 KiB
Go
//go:build plus
|
||
|
||
package mediasenders
|
||
|
||
import (
|
||
"bytes"
|
||
"errors"
|
||
"github.com/iwind/TeaGo/Tea"
|
||
"github.com/iwind/TeaGo/files"
|
||
"github.com/iwind/TeaGo/logs"
|
||
"github.com/iwind/TeaGo/utils/string"
|
||
"os"
|
||
"os/exec"
|
||
"runtime"
|
||
"strings"
|
||
)
|
||
|
||
// ScriptMedia 脚本媒介
|
||
type ScriptMedia struct {
|
||
Path string `yaml:"path" json:"path"`
|
||
ScriptType string `yaml:"scriptType" json:"scriptType"` // 脚本类型，可以为path, code
|
||
ScriptLang string `yaml:"scriptLang" json:"scriptLang"` // 脚本语言
|
||
Script string `yaml:"script" json:"script"` // 脚本代码
|
||
Cwd string `yaml:"cwd" json:"cwd"`
|
||
Env []*Variable `yaml:"env" json:"env"`
|
||
}
|
||
|
||
// NewScriptMedia 获取新对象
|
||
func NewScriptMedia() *ScriptMedia {
|
||
return &ScriptMedia{}
|
||
}
|
||
|
||
// AddEnv 添加环境变量
|
||
func (this *ScriptMedia) AddEnv(name, value string) {
|
||
this.Env = append(this.Env, &Variable{
|
||
Name: name,
|
||
Value: value,
|
||
})
|
||
}
|
||
|
||
// FormattedScript 格式化脚本
|
||
func (this *ScriptMedia) FormattedScript() string {
|
||
script := this.Script
|
||
script = strings.Replace(script, "\r", "", -1)
|
||
return script
|
||
}
|
||
|
||
// Generate 保存到本地
|
||
func (this *ScriptMedia) Generate(id string) (path string, err error) {
|
||
var tmpDir = os.TempDir()
|
||
if runtime.GOOS == "windows" {
|
||
path = tmpDir + Tea.DS + "edge.script." + id + ".bat"
|
||
} else {
|
||
path = tmpDir + "/edge.script." + id + ".script"
|
||
}
|
||
shFile := files.NewFile(path)
|
||
if !shFile.Exists() {
|
||
err = shFile.WriteString(this.FormattedScript())
|
||
if err != nil {
|
||
return
|
||
}
|
||
err = shFile.Chmod(0777)
|
||
if err != nil {
|
||
return
|
||
}
|
||
}
|
||
return
|
||
}
|
||
|
||
// Send 发送
|
||
func (this *ScriptMedia) Send(user string, subject string, body string, productName string, datetime string) (resp []byte, err error) {
|
||
// 脚本
|
||
if this.ScriptType == "code" {
|
||
path, err := this.Generate(stringutil.Rand(16))
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
this.Path = path
|
||
|
||
defer func() {
|
||
f := files.NewFile(this.Path)
|
||
if f.Exists() {
|
||
err := f.Delete()
|
||
if err != nil {
|
||
logs.Error(err)
|
||
}
|
||
}
|
||
}()
|
||
}
|
||
|
||
if len(this.Path) == 0 {
|
||
return nil, errors.New("'path' should be specified")
|
||
}
|
||
|
||
cmd := exec.Command(this.Path)
|
||
|
||
if len(this.Env) > 0 {
|
||
for _, env := range this.Env {
|
||
cmd.Env = append(cmd.Env, env.Name+"="+env.Value)
|
||
}
|
||
}
|
||
cmd.Env = append(cmd.Env, "MessageUser="+user)
|
||
cmd.Env = append(cmd.Env, "MessageSubject="+subject)
|
||
cmd.Env = append(cmd.Env, "MessageBody="+body)
|
||
|
||
if len(this.Cwd) > 0 {
|
||
cmd.Dir = this.Cwd
|
||
}
|
||
|
||
stdout := bytes.NewBuffer([]byte{})
|
||
stderr := bytes.NewBuffer([]byte{})
|
||
|
||
cmd.Stdout = stdout
|
||
cmd.Stderr = stderr
|
||
|
||
err = cmd.Start()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
err = cmd.Wait()
|
||
if err != nil {
|
||
// do nothing
|
||
}
|
||
|
||
if stderr.Len() > 0 {
|
||
return stdout.Bytes(), errors.New(stderr.String())
|
||
}
|
||
|
||
return stdout.Bytes(), nil
|
||
}
|
||
|
||
// RequireUser 是否需要用户标识
|
||
func (this *ScriptMedia) RequireUser() bool {
|
||
return false
|
||
}
|