mirror of
https://codeberg.org/frosty/modbot.git
synced 2024-09-19 11:46:34 -04:00
59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
// 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"
|
|
"errors"
|
|
"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
|
|
}
|
|
|
|
if cmd.ProcessState.ExitCode() != 0 {
|
|
return ExecInfo(""), errors.New("returned non-zero exit code")
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|