]> Cypherpunks.ru repositories - goredo.git/blobdiff - dep.go
Less excess file remove calls
[goredo.git] / dep.go
diff --git a/dep.go b/dep.go
index 38d8bb79c200ede2acedeb9ce357fe313211d7c2..e587e6aa88ced9193b503b94ed2302fb459fba46 100644 (file)
--- a/dep.go
+++ b/dep.go
@@ -1,6 +1,6 @@
 /*
-goredo -- redo implementation on pure Go
-Copyright (C) 2020-2021 Sergey Matveev <stargrave@stargrave.org>
+goredo -- djb's redo implementation on pure Go
+Copyright (C) 2020-2023 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
@@ -21,20 +21,26 @@ package main
 
 import (
        "bufio"
+       "bytes"
        "encoding/hex"
        "errors"
-       "fmt"
        "io"
+       "log"
        "os"
        "path"
-       "strings"
 
        "go.cypherpunks.ru/recfile"
-       "golang.org/x/crypto/blake2b"
-       "golang.org/x/sys/unix"
+       "lukechampine.com/blake3"
 )
 
-func recfileWrite(fdDep *os.File, fields ...recfile.Field) error {
+var (
+       DirPrefix string
+       DepCwd    string
+
+       ErrBadRecFormat = errors.New("invalid format of .rec")
+)
+
+func recfileWrite(fdDep io.StringWriter, fields ...recfile.Field) error {
        w := recfile.NewWriter(fdDep)
        if _, err := w.RecordStart(); err != nil {
                return err
@@ -46,7 +52,7 @@ func recfileWrite(fdDep *os.File, fields ...recfile.Field) error {
 }
 
 func ifcreate(fdDep *os.File, tgt string) error {
-       trace(CDebug, "ifcreate: %s <- %s", fdDep.Name(), tgt)
+       tracef(CDebug, "ifcreate: %s <- %s", fdDep.Name(), tgt)
        return recfileWrite(
                fdDep,
                recfile.Field{Name: "Type", Value: DepTypeIfcreate},
@@ -55,120 +61,137 @@ func ifcreate(fdDep *os.File, tgt string) error {
 }
 
 func always(fdDep *os.File) error {
-       trace(CDebug, "always: %s", fdDep.Name())
+       tracef(CDebug, "always: %s", fdDep.Name())
        return recfileWrite(fdDep, recfile.Field{Name: "Type", Value: DepTypeAlways})
 }
 
 func stamp(fdDep, src *os.File) error {
-       var hsh string
-       hsh, err := fileHash(os.Stdin)
+       hsh, err := fileHash(src)
        if err != nil {
                return err
        }
-       trace(CDebug, "stamp: %s <- %s", fdDep.Name(), hsh)
+       tracef(CDebug, "stamp: %s <- %s", fdDep.Name(), hex.EncodeToString(hsh))
        return recfileWrite(
                fdDep,
                recfile.Field{Name: "Type", Value: DepTypeStamp},
-               recfile.Field{Name: "Hash", Value: hsh},
+               recfile.Field{Name: "Hash", Value: hex.EncodeToString(hsh)},
        )
 }
 
-func fileCtime(fd *os.File) (string, error) {
-       var stat unix.Stat_t
-       if err := unix.Fstat(int(fd.Fd()), &stat); err != nil {
-               return "", err
-       }
-       sec, nsec := stat.Ctim.Unix()
-       return fmt.Sprintf("%d.%d", sec, nsec), nil
-}
-
-func fileHash(fd *os.File) (string, error) {
-       h, err := blake2b.New256(nil)
-       if err != nil {
-               panic(err)
-       }
-       if _, err = io.Copy(h, bufio.NewReader(fd)); err != nil {
-               return "", err
+func fileHash(fd *os.File) ([]byte, error) {
+       h := blake3.New(32, nil)
+       if _, err := io.Copy(h, bufio.NewReader(fd)); err != nil {
+               return nil, err
        }
-       return hex.EncodeToString(h.Sum(nil)), nil
+       return h.Sum(nil), nil
 }
 
-func writeDep(fdDep *os.File, cwd, tgt string) error {
-       trace(CDebug, "ifchange: %s <- %s", fdDep.Name(), tgt)
-       fd, err := os.Open(path.Join(cwd, tgt))
+func depWrite(fdDep *os.File, cwd string, tgt *Tgt, hsh []byte) error {
+       tracef(CDebug, "ifchange: %s <- %s", fdDep.Name(), tgt)
+       fd, err := os.Open(tgt.a)
        if err != nil {
-               return err
+               return ErrLine(err)
        }
        defer fd.Close()
-       fi, err := fd.Stat()
+       inode, isDir, err := inodeFromFileByFd(fd)
        if err != nil {
-               return err
+               return ErrLine(err)
        }
-       if fi.IsDir() {
+       if isDir {
                return nil
        }
-       ts, err := fileCtime(fd)
-       if err != nil {
-               return err
+       if hsh == nil {
+               hsh, err = fileHash(fd)
+               if err != nil {
+                       return ErrLine(err)
+               }
        }
-       hsh, err := fileHash(fd)
-       if err != nil {
-               return err
+       fields := []recfile.Field{
+               {Name: "Type", Value: DepTypeIfchange},
+               {Name: "Target", Value: tgt.RelTo(cwd)},
+               {Name: "Hash", Value: hex.EncodeToString(hsh)},
        }
-       return recfileWrite(
-               fdDep,
-               recfile.Field{Name: "Type", Value: DepTypeIfchange},
-               recfile.Field{Name: "Target", Value: tgt},
-               recfile.Field{Name: "Ctime", Value: ts},
-               recfile.Field{Name: "Hash", Value: hsh},
-       )
+       fields = append(fields, inode.RecfileFields()...)
+       return recfileWrite(fdDep, fields...)
 }
 
-func writeDeps(fdDep *os.File, tgts []string) error {
+func depsWrite(fdDep *os.File, tgts []*Tgt) error {
        if fdDep == nil {
-               trace(CDebug, "no opened fdDep: %s", tgts)
+               tracef(CDebug, "no opened fdDep: %s", tgts)
                return nil
        }
-       ups := []string{}
-       upLevels := strings.Count(os.Getenv(EnvDirPrefix), "/")
-       for i := 0; i < upLevels; i++ {
-               ups = append(ups, "..")
-       }
-       up := path.Join(ups...)
+       var err error
+       var cwd string
        for _, tgt := range tgts {
-               if _, err := os.Stat(tgt); err == nil {
-                       if err = writeDep(fdDep, Cwd, path.Join(up, tgt)); err != nil {
-                               return err
+               cwd = Cwd
+               if DepCwd != "" && Cwd != DepCwd {
+                       cwd = DepCwd
+               }
+               tgtDir := path.Join(cwd, DirPrefix)
+               if _, errStat := os.Stat(tgt.a); errStat == nil {
+                       err = ErrLine(depWrite(fdDep, tgtDir, tgt, nil))
+               } else {
+                       tgtRel := tgt.RelTo(tgtDir)
+                       tracef(CDebug, "ifchange: %s <- %s (non-existing)", fdDep.Name(), tgtRel)
+                       fields := []recfile.Field{
+                               {Name: "Type", Value: DepTypeIfchange},
+                               {Name: "Target", Value: tgtRel},
                        }
+                       inodeDummy := Inode{}
+                       fields = append(fields, inodeDummy.RecfileFields()...)
+                       err = ErrLine(recfileWrite(fdDep, fields...))
+               }
+               if err != nil {
+                       return err
                }
        }
        return nil
 }
 
+type DepInfoIfchange struct {
+       tgt   *Tgt
+       inode *Inode
+       hash  []byte
+}
+
 type DepInfo struct {
        build     string
        always    bool
-       stamp     string
-       ifcreates []string
-       ifchanges []map[string]string
+       stamp     []byte
+       ifcreates []*Tgt
+       ifchanges []DepInfoIfchange
+}
+
+func mustHexDecode(s string) []byte {
+       b, err := hex.DecodeString(s)
+       if err != nil {
+               log.Fatal(err)
+       }
+       return b
 }
 
-func depRead(fdDep *os.File) (*DepInfo, error) {
-       r := recfile.NewReader(fdDep)
+var missingBuild = errors.New(".rec missing Build:")
+
+func depRead(tgt *Tgt) (*DepInfo, error) {
+       data, err := os.ReadFile(tgt.Dep())
+       if err != nil {
+               return nil, err
+       }
+       r := recfile.NewReader(bytes.NewReader(data))
        m, err := r.NextMap()
        if err != nil {
                return nil, err
        }
        depInfo := DepInfo{}
-       if b := m["Build"]; b == "" {
-               return nil, errors.New(".dep missing Build:")
-       } else {
-               depInfo.build = b
+       b := m["Build"]
+       if b == "" {
+               return nil, missingBuild
        }
+       depInfo.build = b
        for {
                m, err := r.NextMap()
                if err != nil {
-                       if err == io.EOF {
+                       if errors.Is(err, io.EOF) {
                                break
                        }
                        return nil, err
@@ -179,21 +202,74 @@ func depRead(fdDep *os.File) (*DepInfo, error) {
                case DepTypeIfcreate:
                        dep := m["Target"]
                        if dep == "" {
-                               return nil, errors.New("invalid format of .dep")
+                               return nil, ErrBadRecFormat
                        }
-                       depInfo.ifcreates = append(depInfo.ifcreates, dep)
+                       depInfo.ifcreates = append(depInfo.ifcreates,
+                               NewTgt(path.Join(tgt.h, dep)))
                case DepTypeIfchange:
-                       delete(m, "Type")
-                       depInfo.ifchanges = append(depInfo.ifchanges, m)
+                       dep := m["Target"]
+                       if dep == "" {
+                               return nil, ErrBadRecFormat
+                       }
+                       inode, err := inodeFromRec(m)
+                       if err != nil {
+                               log.Print(err)
+                               return nil, ErrBadRecFormat
+                       }
+                       hsh := mustHexDecode(m["Hash"])
+                       depInfo.ifchanges = append(depInfo.ifchanges, DepInfoIfchange{
+                               tgt: NewTgt(path.Join(tgt.h, dep)), inode: inode, hash: hsh,
+                       })
                case DepTypeStamp:
                        hsh := m["Hash"]
                        if hsh == "" {
-                               return nil, errors.New("invalid format of .dep")
+                               return nil, ErrBadRecFormat
                        }
-                       depInfo.stamp = hsh
+                       depInfo.stamp = mustHexDecode(hsh)
                default:
-                       return nil, errors.New("invalid format of .dep")
+                       return nil, ErrBadRecFormat
                }
        }
        return &depInfo, nil
 }
+
+func depReadOnlyIfchanges(pth string) (ifchanges []string, err error) {
+       data, err := os.ReadFile(pth)
+       if err != nil {
+               return
+       }
+       r := recfile.NewReader(bytes.NewReader(data))
+       var m map[string]string
+       for {
+               m, err = r.NextMap()
+               if err != nil {
+                       if errors.Is(err, io.EOF) {
+                               err = nil
+                               break
+                       }
+                       return
+               }
+               if m["Type"] == DepTypeIfchange {
+                       ifchanges = append(ifchanges, m["Target"])
+               }
+       }
+       return
+}
+
+func depReadBuild(pth string) (string, error) {
+       fd, err := os.Open(pth)
+       if err != nil {
+               return "", err
+       }
+       r := recfile.NewReader(fd)
+       m, err := r.NextMap()
+       fd.Close()
+       if err != nil {
+               return "", err
+       }
+       build := m["Build"]
+       if build == "" {
+               err = missingBuild
+       }
+       return build, err
+}