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/>.
39 "go.cypherpunks.ru/recfile"
40 "golang.org/x/sys/unix"
44 CmdNameGoredo = "goredo"
46 CmdNameRedoAffects = "redo-affects"
47 CmdNameRedoAlways = "redo-always"
48 CmdNameRedoCleanup = "redo-cleanup"
49 CmdNameRedoDot = "redo-dot"
50 CmdNameRedoIfchange = "redo-ifchange"
51 CmdNameRedoIfcreate = "redo-ifcreate"
52 CmdNameRedoLog = "redo-log"
53 CmdNameRedoOOD = "redo-ood"
54 CmdNameRedoSources = "redo-sources"
55 CmdNameRedoStamp = "redo-stamp"
56 CmdNameRedoTargets = "redo-targets"
57 CmdNameRedoWhichdo = "redo-whichdo"
63 IsTopRedo bool // is it the top redo instance
66 func mustSetenv(key string) {
67 if err := os.Setenv(key, "1"); err != nil {
72 func mustParseFd(v, name string) *os.File {
73 ptr, err := strconv.ParseUint(v, 10, 64)
77 fd := os.NewFile(uintptr(ptr), name)
79 panic("can not parse fd: " + name)
84 func CmdName() string {
85 return path.Base(os.Args[0])
89 version := flag.Bool("version", false, "print version")
90 warranty := flag.Bool("warranty", false, "print warranty information")
94 if cmdName == "goredo" {
95 symlinks = flag.Bool("symlinks", false, "create necessary symlinks in current directory")
98 flag.Usage = func() { usage(os.Args[0]) }
101 fmt.Println(Warranty)
105 fmt.Println("goredo", Version, "built with", runtime.Version())
108 if cmdName == CmdNameGoredo && *symlinks {
110 for _, cmdName := range []string{
125 fmt.Println(os.Args[0], "<-", cmdName)
126 if err := os.Symlink(os.Args[0], cmdName); err != nil {
133 log.SetFlags(log.Lshortfile)
136 Cwd, err = os.Getwd()
141 TopDir = os.Getenv(EnvTopDir)
145 TopDir, err = filepath.Abs(TopDir)
150 DirPrefix = os.Getenv(EnvDirPrefix)
151 DepCwd = os.Getenv(EnvDepCwd)
153 if flagStderrKeep != nil && *flagStderrKeep {
154 mustSetenv(EnvStderrKeep)
156 if flagStderrSilent != nil && *flagStderrSilent {
157 mustSetenv(EnvStderrSilent)
159 if flagNoProgress != nil && *flagNoProgress {
160 mustSetenv(EnvNoProgress)
162 if flagDebug != nil && *flagDebug {
165 if flagLogWait != nil && *flagLogWait {
166 mustSetenv(EnvLogWait)
168 if flagLogLock != nil && *flagLogLock {
169 mustSetenv(EnvLogLock)
171 if flagLogPid != nil && *flagLogPid {
172 mustSetenv(EnvLogPid)
174 if flagLogJS != nil && *flagLogJS {
177 StderrKeep = os.Getenv(EnvStderrKeep) == "1"
178 StderrSilent = os.Getenv(EnvStderrSilent) == "1"
179 NoProgress = os.Getenv(EnvNoProgress) == "1"
180 Debug = os.Getenv(EnvDebug) == "1"
181 LogWait = os.Getenv(EnvLogWait) == "1"
182 LogLock = os.Getenv(EnvLogLock) == "1"
183 LogJS = os.Getenv(EnvLogJS) == "1"
184 if Debug || os.Getenv(EnvLogPid) == "1" {
188 if flagTraceAll != nil && *flagTraceAll {
191 if os.Getenv(EnvTrace) == "1" {
194 } else if flagTrace != nil {
197 NoColor = os.Getenv(EnvNoColor) != ""
198 NoSync = os.Getenv(EnvNoSync) == "1"
199 switch s := os.Getenv(EnvInodeTrust); s {
201 InodeTrust = InodeTrustNone
203 InodeTrust = InodeTrustCtime
205 InodeTrust = InodeTrustMtime
207 log.Fatalln("unknown", EnvInodeTrust, "value")
209 tracef(CDebug, "inode-trust: %s", InodeTrust)
211 // Those are internal envs
212 FdOODTgts, err = ioutil.TempFile("", "ood-tgts")
216 if err = os.Remove(FdOODTgts.Name()); err != nil {
219 FdOODTgtsLock, err = ioutil.TempFile("", "ood-tgts.lock")
223 if err = os.Remove(FdOODTgtsLock.Name()); err != nil {
227 if v := os.Getenv(EnvOODTgtsFd); v != "" {
228 fd := mustParseFd(v, EnvOODTgtsFd)
229 fdLock := mustParseFd(v, EnvOODTgtsLockFd)
230 if err = unix.Flock(int(fdLock.Fd()), unix.LOCK_EX); err != nil {
233 if _, err = fd.Seek(0, io.SeekStart); err != nil {
236 tgtsRaw, err := ioutil.ReadAll(bufio.NewReader(fd))
240 if err = unix.Flock(int(fdLock.Fd()), unix.LOCK_UN); err != nil {
243 OODTgts = map[string]struct{}{}
244 for _, tgtRaw := range bytes.Split(tgtsRaw, []byte{0}) {
249 OODTgts[t] = struct{}{}
250 tracef(CDebug, "ood: known to be: %s", t)
254 StderrPrefix = os.Getenv(EnvStderrPrefix)
255 if v := os.Getenv(EnvLevel); v != "" {
256 Level, err = strconv.Atoi(v)
261 panic("negative " + EnvLevel)
266 if v := os.Getenv(EnvDepFd); v != "" {
267 fdDep = mustParseFd(v, EnvDepFd)
271 BuildUUID = os.Getenv(EnvBuildUUID)
272 tgtsWasEmpty := len(tgts) == 0
276 if _, err = io.ReadFull(rand.Reader, raw[:]); err != nil {
279 raw[6] = (raw[6] & 0x0F) | uint8(4<<4) // version 4
280 BuildUUID = fmt.Sprintf(
282 raw[0:4], raw[4:6], raw[6:8], raw[8:10], raw[10:],
285 tgts = []string{"all"}
289 if cmdName == CmdNameRedo || cmdName == CmdNameRedoIfchange {
293 for i, tgt := range tgts {
295 tgts[i] = cwdMustRel(tgt)
299 killed := make(chan os.Signal, 1)
300 signal.Notify(killed, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)
303 tracef(CDebug, "[%s] killed", BuildUUID)
306 for pid, proc := range RunningProcs {
307 tracef(CDebug, "[%s] killing child %d", BuildUUID, pid)
308 _ = proc.Signal(syscall.SIGTERM)
316 CDebug, "[%s] run: %s %s cwd:%s dirprefix:%s",
317 BuildUUID, cmdName, tgts, Cwd, DirPrefix,
323 for _, tgt := range tgts {
324 ok, err = ifchange([]string{tgt}, true, traced)
325 if err != nil || !ok {
329 case CmdNameRedoIfchange:
330 ok, err = ifchange(tgts, false, traced)
332 err = depsWrite(fdDep, tgts)
334 case CmdNameRedoIfcreate:
336 log.Fatalln("no", EnvDepFd)
338 for _, tgt := range tgts {
339 tgtRel, err := filepath.Rel(
340 filepath.Join(Cwd, DirPrefix),
341 filepath.Join(Cwd, tgt),
346 err = ifcreate(fdDep, tgtRel)
351 case CmdNameRedoAlways:
353 log.Fatalln("no", EnvDepFd)
356 case CmdNameRedoCleanup:
357 for _, what := range tgts {
358 err = cleanupWalker(Cwd, what)
365 case CmdNameRedoStamp:
367 log.Fatalln("no", EnvDepFd)
369 err = stamp(fdDep, os.Stdin)
372 log.Fatalln("single target expected")
374 d, t := cwdAndTgt(tgts[0])
375 err = showBuildLog(d, t, nil, 0)
376 case CmdNameRedoWhichdo:
378 log.Fatalln("single target expected")
381 fdTmp, err = ioutil.TempFile("", "whichdo")
385 if err = os.Remove(fdTmp.Name()); err != nil {
388 cwd, tgt := cwdAndTgt(tgts[0])
389 doFile, upLevels, err := findDo(fdTmp, cwd, tgt)
393 _, err = fdTmp.Seek(0, io.SeekStart)
397 r := recfile.NewReader(fdTmp)
399 m, err := r.NextMap()
401 if errors.Is(err, io.EOF) {
406 fmt.Println(cwdMustRel(cwd, m["Target"]))
411 p := make([]string, 0, upLevels+2)
413 for i := 0; i < upLevels; i++ {
416 p = append(p, doFile)
417 rel, err := filepath.Rel(Cwd, path.Join(p...))
423 case CmdNameRedoTargets:
427 tgts, err = targetsWalker(tgts)
432 for _, tgt := range tgts {
435 case CmdNameRedoAffects:
437 log.Fatalln("no targets specified")
439 var tgtsKnown []string
440 tgtsKnown, err = targetsWalker([]string{Cwd})
444 deps := map[string]map[string]struct{}{}
445 for _, tgt := range tgtsKnown {
446 collectDeps(Cwd, tgt, 0, deps, true)
448 seen := map[string]struct{}{}
449 for _, tgt := range tgts {
450 collectWholeDeps(deps[tgt], deps, seen)
452 tgts := make([]string, 0, len(seen))
453 for dep := range seen {
454 tgts = append(tgts, dep)
457 for _, dep := range tgts {
462 tgts, err = targetsWalker([]string{Cwd})
469 for _, tgt := range tgts {
470 ood, err = isOOD(Cwd, tgt, 0, nil)
478 case CmdNameRedoSources:
480 tgts, err = targetsWalker([]string{Cwd})
487 srcs, err = sourcesWalker(tgts)
489 for _, src := range srcs {
493 log.Fatalln("unknown command", cmdName)
499 if !ok || err != nil {
502 tracef(CDebug, "[%s] finished: %s %s", BuildUUID, cmdName, tgts)