modbot/lib/readers/exec.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 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 General Public License for more details.
//
// You should have received a copy of the GNU 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) func() (interface{}, error) {
	return func() (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")
		}
		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
	}
}