X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=main.go;h=8c9c16da1430a761779856688da8382b82a98e18;hb=ca5290c19761bc1806b98d54d966a9cbbfeabd15;hp=734239b682d5bdac63e48e5afe955509349be6b1;hpb=cd29a67e26ef05f0ffadd83448f09886edef4111;p=goredo.git

diff --git a/main.go b/main.go
index 734239b..8c9c16d 100644
--- a/main.go
+++ b/main.go
@@ -18,6 +18,8 @@ along with this program.  If not, see <http://www.gnu.org/licenses/>.
 package main
 
 import (
+	"bufio"
+	"bytes"
 	"crypto/rand"
 	"flag"
 	"fmt"
@@ -27,9 +29,12 @@ import (
 	"os"
 	"path"
 	"path/filepath"
+	"runtime"
+	"sort"
 	"strconv"
 
 	"go.cypherpunks.ru/recfile"
+	"golang.org/x/sys/unix"
 )
 
 var (
@@ -60,27 +65,31 @@ func main() {
 	warranty := flag.Bool("warranty", false, "print warranty information")
 	symlinks := flag.Bool("symlinks", false, "create necessary symlinks in current directory")
 
-	flag.Usage = usage
+	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 {
 		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",
 		} {
 			fmt.Println(os.Args[0], "<-", cmdName)
@@ -96,7 +105,7 @@ func main() {
 	var err error
 	Cwd, err = os.Getwd()
 	if err != nil {
-		panic(err)
+		log.Fatalln(err)
 	}
 
 	NoColor = os.Getenv(EnvNoColor) != ""
@@ -152,13 +161,54 @@ func main() {
 	if *flagTraceAll {
 		mustSetenv(EnvTrace, "1")
 	}
-	if *flagTrace {
+	if os.Getenv(EnvTrace) == "1" {
+		TracedAll = true
 		traced = true
 	} else {
-		traced = os.Getenv(EnvTrace) == "1"
+		traced = *flagTrace
 	}
 
 	// Those are internal envs
+	FdOODTgts, err = ioutil.TempFile("", "ood-tgts")
+	if err != nil {
+		panic(err)
+	}
+	if err = os.Remove(FdOODTgts.Name()); err != nil {
+		panic(err)
+	}
+	FdOODTgtsLock, err = ioutil.TempFile("", "ood-tgts.lock")
+	if err != nil {
+		panic(err)
+	}
+	if err = os.Remove(FdOODTgtsLock.Name()); err != nil {
+		panic(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)
+		}
+		if _, err = fd.Seek(0, io.SeekStart); err != nil {
+			panic(err)
+		}
+		tgtsRaw, err := ioutil.ReadAll(bufio.NewReader(fd))
+		if err != nil {
+			panic(err)
+		}
+		unix.Flock(int(fdLock.Fd()), unix.LOCK_UN)
+		OODTgts = map[string]struct{}{}
+		for _, tgtRaw := range bytes.Split(tgtsRaw, []byte{0}) {
+			t := string(tgtRaw)
+			if t == "" {
+				continue
+			}
+			OODTgts[t] = struct{}{}
+			trace(CDebug, "ood: known to be: %s", t)
+		}
+	}
+
 	StderrPrefix = os.Getenv(EnvStderrPrefix)
 	if v := os.Getenv(EnvLevel); v != "" {
 		Level, err = strconv.Atoi(v)
@@ -169,26 +219,30 @@ func main() {
 			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 == "" {
 		raw := new([16]byte)
 		if _, err = io.ReadFull(rand.Reader, raw[:]); err != nil {
-			panic(err)
+			log.Fatalln(err)
 		}
 		raw[6] = (raw[6] & 0x0F) | uint8(4<<4) // version 4
 		BuildUUID = fmt.Sprintf(
 			"%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()
 
 	for i, tgt := range tgts {
@@ -264,17 +318,19 @@ CmdSwitch:
 			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
 		}
@@ -304,6 +360,75 @@ CmdSwitch:
 			}
 			fmt.Println(rel)
 		}
+	case "redo-targets":
+		if tgtsWasEmpty {
+			tgts = []string{Cwd}
+		}
+		tgts, err = targetsWalker(tgts)
+		if err != nil {
+			break
+		}
+		sort.Strings(tgts)
+		for _, tgt := range tgts {
+			fmt.Println(tgt)
+		}
+	case "redo-affects":
+		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 "redo-ood":
+		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 "redo-sources":
+		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)
+		}
 	default:
 		log.Fatalln("unknown command", cmdName)
 	}