fet: resolve #10 pkgproto

cmd/pkgproto/main.go

@@ -4,21 +4,61 @@ import (
 	"bufio"
 	"flag"
 	"fmt"
+	"io/fs"
 	"log"
 	"os"
+	"path/filepath"
+	"strings"
 
 	"github.com/Projeto-Pindorama/motoko/internal/archivum"
 )
 
-var class *string
+var class string
+var ignoreLinks bool
 
 func main() {
-	class = flag.String("c", "none", "Class")
+	flag.StringVar(&class, "c", "none", "Maps the class of all paths to class.")
+	flag.BoolVar(&ignoreLinks, "i", false, "Ignores symbolic links and records the paths as ftype=f (a file) versus ftype=s (symbolic link).")
+
+	flag.Usage = func() {
+		fmt.Fprintf(os.Stderr, "usage: pkgproto [-i] [-c class] [path ...]\n")
+	}
+
 	flag.Parse()
-	fs := archivum.NewUnixFS("/")
+
+	remainingArgs := flag.Args()
+
+	unixFS := archivum.NewUnixFS("/")
+
+	for _, v := range remainingArgs {
+		dir := strings.SplitN(v, "=", 2)
+
+		filepath.WalkDir(dir[0], func(path string, d fs.DirEntry, err error) error {
+			if err != nil {
+				log.Fatal(err)
+				return err
+			}
+
+			metadata, err := archivum.Scan(unixFS, path)
+			if err != nil {
+				log.Fatal(err)
+				return err
+			}
+
+			fmt.Println(strings.Replace(MetadataToString(metadata), dir[0], dir[1], 1))
+
+			return nil
+		})
+	}
+
+	if len(remainingArgs) > 0 {
+		return
+	}
+
 	readstdin := bufio.NewScanner(os.Stdin)
+
 	for readstdin.Scan() {
-		metadata, err := archivum.Scan(fs, readstdin.Text())
+		metadata, err := archivum.Scan(unixFS, readstdin.Text())
 		if err != nil {
 			log.Fatal(err)
 		}
@@ -28,36 +68,22 @@ func main() {
 }
 
 func MetadataToString(m *archivum.Metadata) string {
-	if m.FType == 's' {
-		return fmt.Sprintf(
-			"%c %s %s=%s",
-			m.FType,
-			*class,
-			m.Path,
-			m.RealPath,
-		)
+	var ret string
+	var fileType rune = m.FType
+
+	if m.FType == 's' && ignoreLinks {
+		fileType = 'f'
+	}
+
+	ret = fmt.Sprintf("%c %s %s", fileType, class, m.Path)
+
+	if fileType == 's' {
+		ret = ret + fmt.Sprintf("=%s", m.RealPath)
 	} else if m.DeviceInfo == nil {
-		return fmt.Sprintf(
-			"%c %s %s %s %s %s",
-			m.FType,
-			*class,
-			m.Path,
-			m.OctalMod,
-			m.Owner,
-			m.Group,
-		)
+		ret = ret + fmt.Sprintf(" %s %s %s", m.OctalMod, m.Owner, m.Group)
 	} else {
-		return fmt.Sprintf(
-			"%c %s %s %d %d %s %s %s",
-			m.FType,
-			*class,
-			m.Path,
-			m.DeviceInfo.Major,
-			m.DeviceInfo.Minor,
-			m.OctalMod,
-			m.Owner,
-			m.Group,
-		)
+		ret = ret + fmt.Sprintf(" %d %d %s %s %s", m.DeviceInfo.Major, m.DeviceInfo.Minor, m.OctalMod, m.Owner, m.Group)
 	}
 
+	return ret
 }