]> Cypherpunks.ru repositories - goredo.git/blobdiff - main.go
Unify dep*Read/Write name
[goredo.git] / main.go
diff --git a/main.go b/main.go
index 3a7088d16431492773e78a07781e54ff85eeef19..0b893dac172389ca199dcc70a08a2358d88c502e 100644 (file)
--- a/main.go
+++ b/main.go
@@ -21,30 +21,50 @@ 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"
+)
+
 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)
        }
 }
@@ -61,10 +81,19 @@ func mustParseFd(v, name string) *os.File {
        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")
+
+       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()
@@ -76,22 +105,22 @@ func main() {
                fmt.Println("goredo", Version, "built with", runtime.Version())
                return
        }
-       if *symlinks {
+       if cmdName == CmdNameGoredo && *symlinks {
                rc := 0
                for _, cmdName := range []string{
-                       "redo",
-                       "redo-affects",
-                       "redo-always",
-                       "redo-cleanup",
-                       "redo-dot",
-                       "redo-ifchange",
-                       "redo-ifcreate",
-                       "redo-log",
-                       "redo-ood",
-                       "redo-sources",
-                       "redo-stamp",
-                       "redo-targets",
-                       "redo-whichdo",
+                       CmdNameRedo,
+                       CmdNameRedoAffects,
+                       CmdNameRedoAlways,
+                       CmdNameRedoCleanup,
+                       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 {
@@ -101,7 +130,7 @@ func main() {
                }
                os.Exit(rc)
        }
-       log.SetFlags(0)
+       log.SetFlags(log.Lshortfile)
 
        var err error
        Cwd, err = os.Getwd()
@@ -125,29 +154,29 @@ func main() {
        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"
@@ -160,46 +189,48 @@ func main() {
                MyPid = os.Getpid()
        }
        var traced bool
-       if *flagTraceAll {
-               mustSetenv(EnvTrace, "1")
+       if flagTraceAll != nil && *flagTraceAll {
+               mustSetenv(EnvTrace)
        }
        if os.Getenv(EnvTrace) == "1" {
                TracedAll = true
                traced = true
-       } else {
+       } else if flagTrace != nil {
                traced = *flagTrace
        }
 
        // Those are internal envs
        FdOODTgts, err = ioutil.TempFile("", "ood-tgts")
        if err != nil {
-               panic(err)
+               log.Fatalln(err)
        }
        if err = os.Remove(FdOODTgts.Name()); err != nil {
-               panic(err)
+               log.Fatalln(err)
        }
        FdOODTgtsLock, err = ioutil.TempFile("", "ood-tgts.lock")
        if err != nil {
-               panic(err)
+               log.Fatalln(err)
        }
        if err = os.Remove(FdOODTgtsLock.Name()); err != nil {
-               panic(err)
+               log.Fatalln(err)
        }
 
        if v := os.Getenv(EnvOODTgtsFd); v != "" {
                fd := mustParseFd(v, EnvOODTgtsFd)
                fdLock := mustParseFd(v, EnvOODTgtsLockFd)
                if err = unix.Flock(int(fdLock.Fd()), unix.LOCK_EX); err != nil {
-                       panic(err)
+                       log.Fatalln(err)
                }
                if _, err = fd.Seek(0, io.SeekStart); err != nil {
-                       panic(err)
+                       log.Fatalln(err)
                }
                tgtsRaw, err := ioutil.ReadAll(bufio.NewReader(fd))
                if err != nil {
-                       panic(err)
+                       log.Fatalln(err)
+               }
+               if err = unix.Flock(int(fdLock.Fd()), unix.LOCK_UN); err != nil {
+                       log.Fatalln(err)
                }
-               unix.Flock(int(fdLock.Fd()), unix.LOCK_UN)
                OODTgts = map[string]struct{}{}
                for _, tgtRaw := range bytes.Split(tgtsRaw, []byte{0}) {
                        t := string(tgtRaw)
@@ -207,7 +238,7 @@ func main() {
                                continue
                        }
                        OODTgts[t] = struct{}{}
-                       trace(CDebug, "ood: known to be: %s", t)
+                       tracef(CDebug, "ood: known to be: %s", t)
                }
        }
 
@@ -246,7 +277,9 @@ func main() {
                }
        }
 
-       statusInit()
+       if cmdName == CmdNameRedo || cmdName == CmdNameRedoIfchange {
+               statusInit()
+       }
 
        for i, tgt := range tgts {
                if path.IsAbs(tgt) {
@@ -254,69 +287,84 @@ func main() {
                }
        }
 
+       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(
+                               filepath.Join(Cwd, DirPrefix),
+                               filepath.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")
                }
@@ -341,7 +389,7 @@ CmdSwitch:
                for {
                        m, err := r.NextMap()
                        if err != nil {
-                               if err == io.EOF {
+                               if errors.Is(err, io.EOF) {
                                        break
                                }
                                break CmdSwitch
@@ -363,7 +411,7 @@ CmdSwitch:
                        }
                        fmt.Println(rel)
                }
-       case "redo-targets":
+       case CmdNameRedoTargets:
                if tgtsWasEmpty {
                        tgts = []string{Cwd}
                }
@@ -375,7 +423,7 @@ CmdSwitch:
                for _, tgt := range tgts {
                        fmt.Println(tgt)
                }
-       case "redo-affects":
+       case CmdNameRedoAffects:
                if tgtsWasEmpty {
                        log.Fatalln("no targets specified")
                }
@@ -400,7 +448,7 @@ CmdSwitch:
                for _, dep := range tgts {
                        fmt.Println(dep)
                }
-       case "redo-ood":
+       case CmdNameRedoOOD:
                if tgtsWasEmpty {
                        tgts, err = targetsWalker([]string{Cwd})
                        if err != nil {
@@ -418,7 +466,7 @@ CmdSwitch:
                                fmt.Println(tgt)
                        }
                }
-       case "redo-sources":
+       case CmdNameRedoSources:
                if tgtsWasEmpty {
                        tgts, err = targetsWalker([]string{Cwd})
                        if err != nil {
@@ -442,6 +490,6 @@ CmdSwitch:
        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)
 }