build/treble_build/local/cmd.go

// Copyright 2022 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package local

import (
	"bufio"
	"bytes"
	"context"
	"errors"
	"io"
	"os/exec"
	"time"
)

// Run the input command via pipe with given arguments, stdout of the pipe is passed to input parser
// argument.
func runPipe(ctx context.Context, timeout time.Duration, cmdName string, args []string, parser func(r io.Reader)) (err error, stdErr string) {
	ctx, cancel := context.WithTimeout(ctx, timeout)
	defer cancel()

	cmd := exec.CommandContext(ctx, cmdName, args[0:]...)
	errorBuf := bytes.Buffer{}
	cmd.Stderr = &errorBuf
	stdout, err := cmd.StdoutPipe()
	if err != nil {
		return err, errorBuf.String()
	}

	if err = cmd.Start(); err != nil {
		return err, errorBuf.String()
	}
	parser(stdout)
	if err = cmd.Wait(); err != nil {
		return err, errorBuf.String()
	}
	return nil, ""
}

// Run input command, stdout is passed via out parameter to user, if error the stderr is provided via
// stdErr string to the user.
func run(ctx context.Context, timeout time.Duration, cmdName string, args []string) (out *bytes.Buffer, err error, stdErr string) {
	ctx, cancel := context.WithTimeout(ctx, timeout)
	defer cancel()

	cmd := exec.CommandContext(ctx, cmdName, args[0:]...)
	errorBuf := bytes.Buffer{}
	outputBuf := bytes.Buffer{}
	cmd.Stderr = &errorBuf
	cmd.Stdout = &outputBuf
	if err = cmd.Run(); err != nil {
		return nil, err, errorBuf.String()
	}

	return &outputBuf, nil, ""
}

// lineScanner
//
//  Map output lines to strings, with expected number of
// lines
type lineScanner struct {
	Lines []string
}

// Parse into lines
func (l *lineScanner) Parse(s *bufio.Scanner) error {
	i := 0
	for s.Scan() {
		if i < len(l.Lines) {
			l.Lines[i] = s.Text()
		} else {
			i++
			break
		}
		i++
	}
	if i != len(l.Lines) {
		return errors.New("cmd: incorrect number of lines")
	}
	return nil
}

func newLineScanner(numLines int) *lineScanner {
	out := &lineScanner{Lines: make([]string, numLines)}
	return (out)
}