blob: a4a1bf57d10ab395b107e0385159058cedd74783 [file] [log] [blame] [edit]
// Copyright 2019 The SwiftShader Authors. All Rights Reserved.
//
// 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.
// strip_unneeded is a tool that attempts to remove unnecessary lines from a
// file by running a test script after each marked line is removed.
//
// strip_unneeded will scan the file specified by the --file argument for lines
// that contain the substring specified by the --marker argument. One-by-one
// those marked lines will be removed from the file, after which the test script
// specified by --test will be run. If the test passes (the process completes
// with a 0 return code), then the line will remain removed, otherwise it is
// restored. This will repeat for every line in the file containing the marker,
// until all lines are tested.
package main
import (
"flag"
"fmt"
"io/ioutil"
"os"
"os/exec"
"strings"
)
var (
file = flag.String("file", "", "file to modify")
marker = flag.String("marker", "CHECK_NEEDED", "line token")
test = flag.String("test", "", "test script to run with each change")
)
func main() {
if err := run(); err != nil {
fmt.Println(err.Error())
os.Exit(1)
}
}
func run() error {
flag.Parse()
if *file == "" {
return fmt.Errorf("Missing --file argument")
}
if *marker == "" {
return fmt.Errorf("Missing --marker argument")
}
if *test == "" {
return fmt.Errorf("Missing --test argument")
}
// make sure the test passes with no modifications
if err := runTest(); err != nil {
return fmt.Errorf("Test fails with no modifications.\n%v", err)
}
// load the test file
body, err := ioutil.ReadFile(*file)
if err != nil {
return fmt.Errorf("Couldn't load file '%v'", *file)
}
// gather all the lines
allLines := strings.Split(string(body), "\n")
// find all the lines with the marker
markerLines := make([]int, 0, len(allLines))
for i, l := range allLines {
if strings.Contains(l, *marker) {
markerLines = append(markerLines, i)
}
}
omit := map[int]bool{}
save := func() error {
f, err := os.Create(*file)
if err != nil {
return err
}
defer f.Close()
for i, l := range allLines {
if !omit[i] {
f.WriteString(l)
f.WriteString("\n")
}
}
return nil
}
for i, l := range markerLines {
omit[l] = true
if err := save(); err != nil {
return err
}
if err := runTest(); err != nil {
omit[l] = false
fmt.Printf("%d/%d: Test fails when removing line %v: %v\n", i, len(markerLines), l, allLines[l])
} else {
fmt.Printf("%d/%d: Test passes when removing line %v: %v\n", i, len(markerLines), l, allLines[l])
}
}
return save()
}
func runTest() error {
cmd := exec.Command("sh", "-c", *test)
out, err := cmd.CombinedOutput()
if err != nil {
return fmt.Errorf("Test failed with error: %v\n%v", err, string(out))
}
return nil
}