2 goredo -- djb's redo implementation on pure Go
3 Copyright (C) 2020-2021 Sergey Matveev <stargrave@stargrave.org>
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, version 3 of the License.
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program. If not, see <http://www.gnu.org/licenses/>.
38 "go.cypherpunks.ru/recfile"
39 "golang.org/x/sys/unix"
43 CmdNameGoredo = "goredo"
45 CmdNameRedoAffects = "redo-affects"
46 CmdNameRedoAlways = "redo-always"
47 CmdNameRedoCleanup = "redo-cleanup"
48 CmdNameRedoDot = "redo-dot"
49 CmdNameRedoIfchange = "redo-ifchange"
50 CmdNameRedoIfcreate = "redo-ifcreate"
51 CmdNameRedoLog = "redo-log"
52 CmdNameRedoOOD = "redo-ood"
53 CmdNameRedoSources = "redo-sources"
54 CmdNameRedoStamp = "redo-stamp"
55 CmdNameRedoTargets = "redo-targets"
56 CmdNameRedoWhichdo = "redo-whichdo"
62 IsTopRedo bool // is it the top redo instance
65 func mustSetenv(key, value string) {
66 if err := os.Setenv(key, value); err != nil {
71 func mustParseFd(v, name string) *os.File {
72 ptr, err := strconv.ParseUint(v, 10, 64)
76 fd := os.NewFile(uintptr(ptr), name)
78 panic("can not parse fd: " + name)
83 func CmdName() string {
84 return path.Base(os.Args[0])
88 version := flag.Bool("version", false, "print version")
89 warranty := flag.Bool("warranty", false, "print warranty information")
93 if cmdName == "goredo" {
94 symlinks = flag.Bool("symlinks", false, "create necessary symlinks in current directory")
97 flag.Usage = func() { usage(os.Args[0]) }
100 fmt.Println(Warranty)
104 fmt.Println("goredo", Version, "built with", runtime.Version())
107 if cmdName == CmdNameGoredo && *symlinks {
109 for _, cmdName := range []string{
124 fmt.Println(os.Args[0], "<-", cmdName)
125 if err := os.Symlink(os.Args[0], cmdName); err != nil {
132 log.SetFlags(log.Lshortfile)
135 Cwd, err = os.Getwd()
140 NoColor = os.Getenv(EnvNoColor) != ""
141 NoSync = os.Getenv(EnvNoSync) == "1"
142 InodeTrust = os.Getenv(EnvInodeNoTrust) == ""
144 TopDir = os.Getenv(EnvTopDir)
148 TopDir, err = filepath.Abs(TopDir)
153 DirPrefix = os.Getenv(EnvDirPrefix)
154 DepCwd = os.Getenv(EnvDepCwd)
156 if flagStderrKeep != nil && *flagStderrKeep {
157 mustSetenv(EnvStderrKeep, "1")
159 if flagStderrSilent != nil && *flagStderrSilent {
160 mustSetenv(EnvStderrSilent, "1")
162 if flagNoProgress != nil && *flagNoProgress {
163 mustSetenv(EnvNoProgress, "1")
165 if flagDebug != nil && *flagDebug {
166 mustSetenv(EnvDebug, "1")
168 if flagLogWait != nil && *flagLogWait {
169 mustSetenv(EnvLogWait, "1")
171 if flagLogLock != nil && *flagLogLock {
172 mustSetenv(EnvLogLock, "1")
174 if flagLogPid != nil && *flagLogPid {
175 mustSetenv(EnvLogPid, "1")
177 if flagLogJS != nil && *flagLogJS {
178 mustSetenv(EnvLogJS, "1")
180 StderrKeep = os.Getenv(EnvStderrKeep) == "1"
181 StderrSilent = os.Getenv(EnvStderrSilent) == "1"
182 NoProgress = os.Getenv(EnvNoProgress) == "1"
183 Debug = os.Getenv(EnvDebug) == "1"
184 LogWait = os.Getenv(EnvLogWait) == "1"
185 LogLock = os.Getenv(EnvLogLock) == "1"
186 LogJS = os.Getenv(EnvLogJS) == "1"
187 if Debug || os.Getenv(EnvLogPid) == "1" {
191 if flagTraceAll != nil && *flagTraceAll {
192 mustSetenv(EnvTrace, "1")
194 if os.Getenv(EnvTrace) == "1" {
197 } else if flagTrace != nil {
201 // Those are internal envs
202 FdOODTgts, err = ioutil.TempFile("", "ood-tgts")
206 if err = os.Remove(FdOODTgts.Name()); err != nil {
209 FdOODTgtsLock, err = ioutil.TempFile("", "ood-tgts.lock")
213 if err = os.Remove(FdOODTgtsLock.Name()); err != nil {
217 if v := os.Getenv(EnvOODTgtsFd); v != "" {
218 fd := mustParseFd(v, EnvOODTgtsFd)
219 fdLock := mustParseFd(v, EnvOODTgtsLockFd)
220 if err = unix.Flock(int(fdLock.Fd()), unix.LOCK_EX); err != nil {
223 if _, err = fd.Seek(0, io.SeekStart); err != nil {
226 tgtsRaw, err := ioutil.ReadAll(bufio.NewReader(fd))
230 unix.Flock(int(fdLock.Fd()), unix.LOCK_UN)
231 OODTgts = map[string]struct{}{}
232 for _, tgtRaw := range bytes.Split(tgtsRaw, []byte{0}) {
237 OODTgts[t] = struct{}{}
238 trace(CDebug, "ood: known to be: %s", t)
242 StderrPrefix = os.Getenv(EnvStderrPrefix)
243 if v := os.Getenv(EnvLevel); v != "" {
244 Level, err = strconv.Atoi(v)
249 panic("negative " + EnvLevel)
254 if v := os.Getenv(EnvDepFd); v != "" {
255 fdDep = mustParseFd(v, EnvDepFd)
259 BuildUUID = os.Getenv(EnvBuildUUID)
260 tgtsWasEmpty := len(tgts) == 0
264 if _, err = io.ReadFull(rand.Reader, raw[:]); err != nil {
267 raw[6] = (raw[6] & 0x0F) | uint8(4<<4) // version 4
268 BuildUUID = fmt.Sprintf(
270 raw[0:4], raw[4:6], raw[6:8], raw[8:10], raw[10:],
273 tgts = []string{"all"}
277 if cmdName == CmdNameRedo || cmdName == CmdNameRedoIfchange {
281 for i, tgt := range tgts {
283 tgts[i] = cwdMustRel(tgt)
287 killed := make(chan os.Signal, 0)
288 signal.Notify(killed, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)
291 trace(CDebug, "[%s] killed", BuildUUID)
294 for pid, proc := range RunningProcs {
295 trace(CDebug, "[%s] killing child %d", BuildUUID, pid)
296 proc.Signal(syscall.SIGTERM)
304 CDebug, "[%s] run: %s %s cwd:%s dirprefix:%s",
305 BuildUUID, cmdName, tgts, Cwd, DirPrefix,
311 for _, tgt := range tgts {
312 ok, err = ifchange([]string{tgt}, true, traced)
313 if err != nil || !ok {
317 case CmdNameRedoIfchange:
318 ok, err = ifchange(tgts, false, traced)
320 err = writeDeps(fdDep, tgts)
322 case CmdNameRedoIfcreate:
324 log.Fatalln("no", EnvDepFd)
326 for _, tgt := range tgts {
327 tgtRel, err := filepath.Rel(
328 filepath.Join(Cwd, DirPrefix),
329 filepath.Join(Cwd, tgt),
334 err = ifcreate(fdDep, tgtRel)
339 case CmdNameRedoAlways:
341 log.Fatalln("no", EnvDepFd)
344 case CmdNameRedoCleanup:
345 for _, what := range tgts {
346 err = cleanupWalker(Cwd, what)
353 case CmdNameRedoStamp:
355 log.Fatalln("no", EnvDepFd)
357 err = stamp(fdDep, os.Stdin)
360 log.Fatalln("single target expected")
362 d, t := cwdAndTgt(tgts[0])
363 err = showBuildLog(d, t, nil, 0)
364 case CmdNameRedoWhichdo:
366 log.Fatalln("single target expected")
369 fdTmp, err = ioutil.TempFile("", "whichdo")
373 if err = os.Remove(fdTmp.Name()); err != nil {
376 cwd, tgt := cwdAndTgt(tgts[0])
377 doFile, upLevels, err := findDo(fdTmp, cwd, tgt)
381 _, err = fdTmp.Seek(0, io.SeekStart)
385 r := recfile.NewReader(fdTmp)
387 m, err := r.NextMap()
394 fmt.Println(cwdMustRel(cwd, m["Target"]))
399 p := make([]string, 0, upLevels+2)
401 for i := 0; i < upLevels; i++ {
404 p = append(p, doFile)
405 rel, err := filepath.Rel(Cwd, path.Join(p...))
411 case CmdNameRedoTargets:
415 tgts, err = targetsWalker(tgts)
420 for _, tgt := range tgts {
423 case CmdNameRedoAffects:
425 log.Fatalln("no targets specified")
427 var tgtsKnown []string
428 tgtsKnown, err = targetsWalker([]string{Cwd})
432 deps := map[string]map[string]struct{}{}
433 for _, tgt := range tgtsKnown {
434 collectDeps(Cwd, tgt, 0, deps, true)
436 seen := map[string]struct{}{}
437 for _, tgt := range tgts {
438 collectWholeDeps(deps[tgt], deps, seen)
440 tgts := make([]string, 0, len(seen))
441 for dep := range seen {
442 tgts = append(tgts, dep)
445 for _, dep := range tgts {
450 tgts, err = targetsWalker([]string{Cwd})
457 for _, tgt := range tgts {
458 ood, err = isOOD(Cwd, tgt, 0, nil)
466 case CmdNameRedoSources:
468 tgts, err = targetsWalker([]string{Cwd})
475 srcs, err = sourcesWalker(tgts)
477 for _, src := range srcs {
481 log.Fatalln("unknown command", cmdName)
487 if !ok || err != nil {
490 trace(CDebug, "[%s] finished: %s %s", BuildUUID, cmdName, tgts)