2024-08-20 17:58:40 -04:00
|
|
|
// 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
|
2024-08-25 02:27:55 -04:00
|
|
|
// it under the terms of the GNU General Public License as published by
|
|
|
|
// the Free Software Foundation, either version 3 of the License, or
|
2024-08-20 17:58:40 -04:00
|
|
|
// (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
|
2024-08-25 02:27:55 -04:00
|
|
|
// GNU General Public License for more details.
|
2024-08-20 17:58:40 -04:00
|
|
|
//
|
2024-08-25 02:27:55 -04:00
|
|
|
// You should have received a copy of the GNU General Public License
|
2024-08-20 17:58:40 -04:00
|
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
package readers
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
2024-08-20 19:01:05 -04:00
|
|
|
"errors"
|
2024-08-20 17:58:40 -04:00
|
|
|
"os/exec"
|
|
|
|
"strings"
|
|
|
|
)
|
|
|
|
|
|
|
|
type ExecInfo string
|
|
|
|
|
|
|
|
func ReadExec(command string) func() (interface{}, error) {
|
|
|
|
return func() (interface{}, error) {
|
2024-08-24 03:13:04 -04:00
|
|
|
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")
|
|
|
|
}
|
|
|
|
if len(stderr.String()) > 0 {
|
|
|
|
return ExecInfo(""), errors.New("data in stderr found")
|
|
|
|
}
|
|
|
|
|
|
|
|
outputLines := strings.Split(stdout.String(), "\n")
|
|
|
|
if len(outputLines) == 0 {
|
|
|
|
return ExecInfo(""), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
outputString := outputLines[0]
|
|
|
|
return ExecInfo(outputString), nil
|
2024-08-20 17:58:40 -04:00
|
|
|
}
|
|
|
|
}
|