modbot/lib/readers/exec.go

54 lines
1.4 KiB
Go
Raw Normal View History

// modbot is a system information agregator
// Copyright (C) 2024 frosty <inthishouseofcards@gmail.com>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published
// by the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
package readers
import (
"bytes"
"os/exec"
"strings"
)
type ExecInfo string
func readExec(command string) (interface{}, error) {
args := []string{"sh", "-c", command}
var stdout, stderr bytes.Buffer
cmd := exec.Command(args[0], args[1:]...)
cmd.Stdout = &stdout
cmd.Stderr = &stderr
if err := cmd.Run(); err != nil {
return ExecInfo(""), err
}
outputLines := strings.Split(stdout.String(), "\n")
if len(outputLines) == 0 {
return ExecInfo(""), nil
}
outputString := outputLines[0]
return ExecInfo(outputString), nil
}
func ReadExec(command string) func() (interface{}, error) {
return func() (interface{}, error) {
return readExec(command)
}
}