bshchk/main.go

81 lines
2.0 KiB
Go
Raw Normal View History

2024-06-02 11:32:17 +02:00
package main
import (
"bufio"
"fmt"
"os"
"strings"
"github.com/alexflint/go-arg"
)
var version string // to be set by compiler
var args struct {
File string `arg:"positional" help:"if not specified, will read from stdin" default:""`
Outfile string `arg:"positional" help:"if not specified, will emit to stdout" default:""`
Version bool `arg:"-v" help:"print version and exit"`
YieldDepsOnly bool `arg:"--yield-deps-only" help:"print dependencies as a JSON array and exit" default:false`
ExposeDeps bool `arg:"--expose-deps" help:"expose dependencies to program" default:false`
DepsVarName string `arg:"--deps-var-name" help:"override deps variable name" default:"deps"`
IgnoreShebang bool `arg:"--ignore-shebang" help:"ignore shebang requirement" default:false`
}
func main() {
arg.MustParse(&args)
if args.Version {
fmt.Println(version)
os.Exit(0)
}
var file []byte
if args.File != "" {
f, err := os.ReadFile(args.File)
if err != nil {
if os.IsNotExist(err) {
fmt.Fprintf(os.Stderr, "File %s does not exist!", args.File)
os.Exit(1)
}
panic(err)
}
file = f
} else {
scanner := bufio.NewScanner(os.Stdin)
for scanner.Scan() {
file = scanner.Bytes()
}
if scanner.Err() != nil {
panic(scanner.Err())
}
}
code := string(file)
found, err := find(code)
if err != nil {
panic(err)
}
codelines := strings.Split(code, "\n")
if len(codelines) < 2 {
fmt.Fprintf(os.Stderr, "The code must have at least two lines!\n")
os.Exit(3)
}
shebang := codelines[0]
if !((shebang == "#!/bin/bash") || (shebang == "#!/usr/bin/env bash")) && (!args.IgnoreShebang) {
fmt.Fprintf(os.Stderr, "The code must start with one of those shebangs: #!/bin/bash OR #!/usr/bin/env bash\n")
os.Exit(2)
}
gen := shebang + "\n\n" + gencode(found) + "\n" + strings.Join(codelines[1:], "\n")
if args.Outfile == "" {
fmt.Printf("%s", gen)
} else {
os.WriteFile(args.Outfile, []byte(gen), os.FileMode(0o755))
}
}