/*
goredo -- djb's redo implementation on pure Go
-Copyright (C) 2020-2021 Sergey Matveev <stargrave@stargrave.org>
+Copyright (C) 2020-2022 Sergey Matveev <stargrave@stargrave.org>
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
package main
import (
+ "bufio"
+ "bytes"
"crypto/rand"
+ "errors"
"flag"
"fmt"
"io"
"io/ioutil"
"log"
"os"
+ "os/signal"
"path"
"path/filepath"
+ "runtime"
+ "sort"
"strconv"
+ "syscall"
"go.cypherpunks.ru/recfile"
+ "golang.org/x/sys/unix"
+)
+
+const (
+ CmdNameGoredo = "goredo"
+ CmdNameRedo = "redo"
+ CmdNameRedoAffects = "redo-affects"
+ CmdNameRedoAlways = "redo-always"
+ CmdNameRedoCleanup = "redo-cleanup"
+ CmdNameRedoDot = "redo-dot"
+ CmdNameRedoIfchange = "redo-ifchange"
+ CmdNameRedoIfcreate = "redo-ifcreate"
+ CmdNameRedoLog = "redo-log"
+ CmdNameRedoOOD = "redo-ood"
+ CmdNameRedoSources = "redo-sources"
+ CmdNameRedoStamp = "redo-stamp"
+ CmdNameRedoTargets = "redo-targets"
+ CmdNameRedoWhichdo = "redo-whichdo"
+ CmdNameRedoDepFix = "redo-depfix"
)
var (
Cwd string
BuildUUID string
+ IsTopRedo bool // is it the top redo instance
)
-func mustSetenv(key, value string) {
- if err := os.Setenv(key, value); err != nil {
+func mustSetenv(key string) {
+ if err := os.Setenv(key, "1"); err != nil {
panic(err)
}
}
return fd
}
+func CmdName() string {
+ return path.Base(os.Args[0])
+}
+
func main() {
version := flag.Bool("version", false, "print version")
warranty := flag.Bool("warranty", false, "print warranty information")
- symlinks := flag.Bool("symlinks", false, "create necessary symlinks in current directory")
- flag.Usage = usage
+ var symlinks *bool
+ cmdName := CmdName()
+ if cmdName == "goredo" {
+ symlinks = flag.Bool("symlinks", false, "create necessary symlinks in current directory")
+ }
+
+ flag.Usage = func() { usage(os.Args[0]) }
flag.Parse()
if *warranty {
fmt.Println(Warranty)
return
}
if *version {
- fmt.Println(versionGet())
+ fmt.Println("goredo", Version, "built with", runtime.Version())
return
}
- if *symlinks {
+ if cmdName == CmdNameGoredo && *symlinks {
rc := 0
for _, cmdName := range []string{
- "redo",
- "redo-always",
- "redo-cleanup",
- "redo-dot",
- "redo-ifchange",
- "redo-ifcreate",
- "redo-log",
- "redo-stamp",
- "redo-whichdo",
+ CmdNameRedo,
+ CmdNameRedoAffects,
+ CmdNameRedoAlways,
+ CmdNameRedoCleanup,
+ CmdNameRedoDepFix,
+ CmdNameRedoDot,
+ CmdNameRedoIfchange,
+ CmdNameRedoIfcreate,
+ CmdNameRedoLog,
+ CmdNameRedoOOD,
+ CmdNameRedoSources,
+ CmdNameRedoStamp,
+ CmdNameRedoTargets,
+ CmdNameRedoWhichdo,
} {
fmt.Println(os.Args[0], "<-", cmdName)
if err := os.Symlink(os.Args[0], cmdName); err != nil {
}
os.Exit(rc)
}
- log.SetFlags(0)
+ log.SetFlags(log.Lshortfile)
var err error
Cwd, err = os.Getwd()
log.Fatalln(err)
}
- NoColor = os.Getenv(EnvNoColor) != ""
- NoSync = os.Getenv(EnvNoSync) == "1"
- InodeTrust = os.Getenv(EnvInodeNoTrust) == ""
-
TopDir = os.Getenv(EnvTopDir)
if TopDir == "" {
TopDir = "/"
}
}
DirPrefix = os.Getenv(EnvDirPrefix)
+ DepCwd = os.Getenv(EnvDepCwd)
- if *flagStderrKeep {
- mustSetenv(EnvStderrKeep, "1")
+ if flagStderrKeep != nil && *flagStderrKeep {
+ mustSetenv(EnvStderrKeep)
}
- if *flagStderrSilent {
- mustSetenv(EnvStderrSilent, "1")
+ if flagStderrSilent != nil && *flagStderrSilent {
+ mustSetenv(EnvStderrSilent)
}
- if *flagNoProgress {
- mustSetenv(EnvNoProgress, "1")
+ if flagNoProgress != nil && *flagNoProgress {
+ mustSetenv(EnvNoProgress)
}
- if *flagDebug {
- mustSetenv(EnvDebug, "1")
+ if flagDebug != nil && *flagDebug {
+ mustSetenv(EnvDebug)
}
- if *flagLogWait {
- mustSetenv(EnvLogWait, "1")
+ if flagLogWait != nil && *flagLogWait {
+ mustSetenv(EnvLogWait)
}
- if *flagLogLock {
- mustSetenv(EnvLogLock, "1")
+ if flagLogLock != nil && *flagLogLock {
+ mustSetenv(EnvLogLock)
}
- if *flagLogPid {
- mustSetenv(EnvLogPid, "1")
+ if flagLogPid != nil && *flagLogPid {
+ mustSetenv(EnvLogPid)
}
- if *flagLogJS {
- mustSetenv(EnvLogJS, "1")
+ if flagLogJS != nil && *flagLogJS {
+ mustSetenv(EnvLogJS)
}
StderrKeep = os.Getenv(EnvStderrKeep) == "1"
StderrSilent = os.Getenv(EnvStderrSilent) == "1"
MyPid = os.Getpid()
}
var traced bool
- if *flagTraceAll {
- mustSetenv(EnvTrace, "1")
+ if flagTraceAll != nil && *flagTraceAll {
+ mustSetenv(EnvTrace)
}
- if *flagTrace {
+ if os.Getenv(EnvTrace) == "1" {
+ TracedAll = true
traced = true
- } else {
- traced = os.Getenv(EnvTrace) == "1"
+ } else if flagTrace != nil {
+ traced = *flagTrace
+ }
+ NoColor = os.Getenv(EnvNoColor) != ""
+ NoSync = os.Getenv(EnvNoSync) == "1"
+ switch s := os.Getenv(EnvInodeTrust); s {
+ case "none":
+ InodeTrust = InodeTrustNone
+ case "", "ctime":
+ InodeTrust = InodeTrustCtime
+ case "mtime":
+ InodeTrust = InodeTrustMtime
+ default:
+ log.Fatalln("unknown", EnvInodeTrust, "value")
}
+ tracef(CDebug, "inode-trust: %s", InodeTrust)
// Those are internal envs
+ FdOODTgts, err = ioutil.TempFile("", "ood-tgts")
+ if err != nil {
+ log.Fatalln(err)
+ }
+ if err = os.Remove(FdOODTgts.Name()); err != nil {
+ log.Fatalln(err)
+ }
+ FdOODTgtsLock, err = ioutil.TempFile("", "ood-tgts.lock")
+ if err != nil {
+ log.Fatalln(err)
+ }
+ if err = os.Remove(FdOODTgtsLock.Name()); err != nil {
+ log.Fatalln(err)
+ }
+
+ if v := os.Getenv(EnvOODTgtsFd); v != "" {
+ fd := mustParseFd(v, EnvOODTgtsFd)
+ fdLock := mustParseFd(v, EnvOODTgtsLockFd)
+ flock := unix.Flock_t{
+ Type: unix.F_WRLCK,
+ Whence: io.SeekStart,
+ }
+ if err = unix.FcntlFlock(fdLock.Fd(), unix.F_SETLKW, &flock); err != nil {
+ log.Fatalln(err)
+ }
+ if _, err = fd.Seek(0, io.SeekStart); err != nil {
+ log.Fatalln(err)
+ }
+ tgtsRaw, err := ioutil.ReadAll(bufio.NewReader(fd))
+ if err != nil {
+ log.Fatalln(err)
+ }
+ flock.Type = unix.F_UNLCK
+ if err = unix.FcntlFlock(fdLock.Fd(), unix.F_SETLK, &flock); err != nil {
+ log.Fatalln(err)
+ }
+ OODTgts = map[string]struct{}{}
+ for _, tgtRaw := range bytes.Split(tgtsRaw, []byte{0}) {
+ t := string(tgtRaw)
+ if t == "" {
+ continue
+ }
+ OODTgts[t] = struct{}{}
+ tracef(CDebug, "ood: known to be: %s", t)
+ }
+ }
+
StderrPrefix = os.Getenv(EnvStderrPrefix)
if v := os.Getenv(EnvLevel); v != "" {
Level, err = strconv.Atoi(v)
panic("negative " + EnvLevel)
}
}
+
var fdDep *os.File
if v := os.Getenv(EnvDepFd); v != "" {
fdDep = mustParseFd(v, EnvDepFd)
}
- BuildUUID = os.Getenv(EnvBuildUUID)
+
tgts := flag.Args()
+ BuildUUID = os.Getenv(EnvBuildUUID)
+ tgtsWasEmpty := len(tgts) == 0
if BuildUUID == "" {
+ IsTopRedo = true
raw := new([16]byte)
if _, err = io.ReadFull(rand.Reader, raw[:]); err != nil {
log.Fatalln(err)
"%x-%x-%x-%x-%x",
raw[0:4], raw[4:6], raw[6:8], raw[8:10], raw[10:],
)
- if len(tgts) == 0 {
+ if tgtsWasEmpty {
tgts = []string{"all"}
}
}
- statusInit()
+
+ if cmdName == CmdNameRedo || cmdName == CmdNameRedoIfchange {
+ statusInit()
+ }
for i, tgt := range tgts {
if path.IsAbs(tgt) {
}
}
+ killed := make(chan os.Signal, 1)
+ signal.Notify(killed, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)
+ go func() {
+ <-killed
+ tracef(CDebug, "[%s] killed", BuildUUID)
+ jsReleaseAll()
+ RunningProcsM.Lock()
+ for pid, proc := range RunningProcs {
+ tracef(CDebug, "[%s] killing child %d", BuildUUID, pid)
+ _ = proc.Signal(syscall.SIGTERM)
+ }
+ os.Exit(1)
+ }()
+
ok := true
err = nil
- cmdName := path.Base(os.Args[0])
- trace(
+ tracef(
CDebug, "[%s] run: %s %s cwd:%s dirprefix:%s",
BuildUUID, cmdName, tgts, Cwd, DirPrefix,
)
CmdSwitch:
switch cmdName {
- case "redo":
+ case CmdNameRedo:
for _, tgt := range tgts {
ok, err = ifchange([]string{tgt}, true, traced)
if err != nil || !ok {
break
}
}
- case "redo-ifchange":
+ case CmdNameRedoIfchange:
ok, err = ifchange(tgts, false, traced)
if err == nil {
- err = writeDeps(fdDep, tgts)
+ err = depsWrite(fdDep, tgts)
}
- case "redo-ifcreate":
+ case CmdNameRedoIfcreate:
if fdDep == nil {
log.Fatalln("no", EnvDepFd)
}
for _, tgt := range tgts {
- err = ifcreate(fdDep, tgt)
+ tgtRel, err := filepath.Rel(
+ path.Join(Cwd, DirPrefix),
+ path.Join(Cwd, tgt),
+ )
+ if err != nil {
+ break
+ }
+ err = ifcreate(fdDep, tgtRel)
if err != nil {
break
}
}
- case "redo-always":
+ case CmdNameRedoAlways:
if fdDep == nil {
log.Fatalln("no", EnvDepFd)
}
err = always(fdDep)
- case "redo-cleanup":
+ case CmdNameRedoCleanup:
for _, what := range tgts {
err = cleanupWalker(Cwd, what)
if err != nil {
break
}
}
- case "redo-dot":
+ case CmdNameRedoDot:
err = dotPrint(tgts)
- case "redo-stamp":
+ case CmdNameRedoStamp:
if fdDep == nil {
log.Fatalln("no", EnvDepFd)
}
err = stamp(fdDep, os.Stdin)
- case "redo-log":
+ case CmdNameRedoLog:
if len(tgts) != 1 {
log.Fatalln("single target expected")
}
d, t := cwdAndTgt(tgts[0])
- var fd *os.File
- fd, err = os.Open(path.Join(d, RedoDir, t+LogSuffix))
- if err != nil {
- break
- }
- _, err = io.Copy(os.Stdout, fd)
- case "redo-whichdo":
+ err = showBuildLog(d, t, nil, 0)
+ case CmdNameRedoWhichdo:
if len(tgts) != 1 {
log.Fatalln("single target expected")
}
var fdTmp *os.File
- fdTmp, err = ioutil.TempFile("", "")
+ fdTmp, err = ioutil.TempFile("", "whichdo")
if err != nil {
break
}
- os.Remove(fdTmp.Name())
+ if err = os.Remove(fdTmp.Name()); err != nil {
+ break
+ }
cwd, tgt := cwdAndTgt(tgts[0])
doFile, upLevels, err := findDo(fdTmp, cwd, tgt)
if err != nil {
break
}
- _, err = fdTmp.Seek(0, 0)
+ _, err = fdTmp.Seek(0, io.SeekStart)
if err != nil {
break
}
for {
m, err := r.NextMap()
if err != nil {
- if err == io.EOF {
+ if errors.Is(err, io.EOF) {
break
}
break CmdSwitch
}
fmt.Println(rel)
}
+ case CmdNameRedoTargets:
+ if tgtsWasEmpty {
+ tgts = []string{Cwd}
+ }
+ tgts, err = targetsWalker(tgts)
+ if err != nil {
+ break
+ }
+ sort.Strings(tgts)
+ for _, tgt := range tgts {
+ fmt.Println(tgt)
+ }
+ case CmdNameRedoAffects:
+ if tgtsWasEmpty {
+ log.Fatalln("no targets specified")
+ }
+ var tgtsKnown []string
+ tgtsKnown, err = targetsWalker([]string{Cwd})
+ if err != nil {
+ break
+ }
+ deps := map[string]map[string]struct{}{}
+ for _, tgt := range tgtsKnown {
+ collectDeps(Cwd, tgt, 0, deps, true)
+ }
+ seen := map[string]struct{}{}
+ for _, tgt := range tgts {
+ collectWholeDeps(deps[tgt], deps, seen)
+ }
+ tgts := make([]string, 0, len(seen))
+ for dep := range seen {
+ tgts = append(tgts, dep)
+ }
+ sort.Strings(tgts)
+ for _, dep := range tgts {
+ fmt.Println(dep)
+ }
+ case CmdNameRedoOOD:
+ if tgtsWasEmpty {
+ tgts, err = targetsWalker([]string{Cwd})
+ if err != nil {
+ break
+ }
+ }
+ sort.Strings(tgts)
+ var ood bool
+ for _, tgt := range tgts {
+ ood, err = isOOD(Cwd, tgt, 0, nil)
+ if err != nil {
+ break
+ }
+ if ood {
+ fmt.Println(tgt)
+ }
+ }
+ case CmdNameRedoSources:
+ if tgtsWasEmpty {
+ tgts, err = targetsWalker([]string{Cwd})
+ if err != nil {
+ break
+ }
+ }
+ sort.Strings(tgts)
+ var srcs []string
+ srcs, err = sourcesWalker(tgts)
+ sort.Strings(srcs)
+ for _, src := range srcs {
+ fmt.Println(src)
+ }
+ case CmdNameRedoDepFix:
+ err = depFix(Cwd)
default:
log.Fatalln("unknown command", cmdName)
}
if !ok || err != nil {
rc = 1
}
- trace(CDebug, "[%s] finished: %s %s", BuildUUID, cmdName, tgts)
+ tracef(CDebug, "[%s] finished: %s %s", BuildUUID, cmdName, tgts)
os.Exit(rc)
}