Support copy to clipboard

file_test.go

@@ -1,6 +1,9 @@
 package main
 
-import "testing"
+import (
+	"path/filepath"
+	"testing"
+)
 
 func TestByteCountSI(t *testing.T) {
 	t.Parallel()
@@ -31,5 +34,17 @@ func TestByteCountSI(t *testing.T) {
 			}
 		})
 	}
+}
+
+func TestReadFileInfo(t *testing.T) {
+	testFile := "file.go"
+	abs, _, err := ReadFileInfo(testFile)
+	if err != nil {
+		t.Fatalf("cannot read file info: %v", err)
+	}
 
+	want, _ := filepath.Abs(testFile)
+	if abs != want {
+		t.Fatalf("invalid abs file, want: %v - got: %v", want, abs)
+	}
 }

fp.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"bytes"
 	"crypto/md5"
 	"crypto/sha1"
 	"crypto/sha256"
@@ -11,6 +12,8 @@ import (
 	"io"
 	"log"
 	"os"
+
+	"github.com/atotto/clipboard"
 )
 
 // DigestGroup hashes a file for a given path
@@ -49,7 +52,7 @@ func usage() {
 
 var (
 	flagVerbose = flag.Bool("v", false, `print the file hash in (md5, sha1, sha256)`)
-	flagClip    = flag.Bool("clip", false, "copy the path to the clipboard")
+	flagClip    = flag.Bool("c", false, "copy the path to the clipboard")
 )
 
 func main() {
@@ -68,14 +71,15 @@ func main() {
 		log.Fatal(err)
 	}
 
-	rr := NewRowRender(os.Stdout)
-	if fi.IsDir() {
-		rr.AddRow("DIR", abs)
-	} else {
-		rr.AddRow("FILE", abs)
-	}
+	rr := NewRowRender()
+	rr.AddRow("", abs)
 
 	if *flagVerbose && !fi.IsDir() {
+		if fi.IsDir() {
+			rr.AddRow("DIR", abs)
+		} else {
+			rr.AddRow("FILE", abs)
+		}
 		rr.AddRow("SIZE", ByteCountSI(fi.Size()))
 		hg := make(map[string]hash.Hash)
 		hg["MD5"] = md5.New()
@@ -87,7 +91,13 @@ func main() {
 			log.Fatal("unable to digest group: ", err)
 		}
 		rr.AddRowMap(values)
+		rr.Write(os.Stdout)
 	}
 
-	rr.RenderTo(os.Stdout)
+	if *flagClip {
+		buf := new(bytes.Buffer)
+		rr.Write(buf)
+		clipboard.WriteAll(buf.String())
+		return
+	}
 }

go.mod

@@ -1,3 +1,5 @@
 module github.com/lotusirous/fp
 
 go 1.17
+
+require github.com/atotto/clipboard v0.1.4

go.sum

@@ -0,0 +1,2 @@
+github.com/atotto/clipboard v0.1.4 h1:EH0zSVneZPSuFR11BlR9YppQTVDbh5+16AmcJi4g1z4=
+github.com/atotto/clipboard v0.1.4/go.mod h1:ZY9tmq7sm5xIbd9bOK4onWV4S6X0u6GY7Vn0Yu86PYI=

printer.go

@@ -13,23 +13,26 @@ type RowRenderer interface {
 	AddRow(field, value string)
 	// AddRowMap appends the row by map input.
 	AddRowMap(values map[string]string)
-	// RenderTo the rows.
-	RenderTo(w io.Writer)
+	// Write the rows.
+	Write(w io.Writer) (n int, err error)
 }
 
 // NewRowRender creates the printer.
-func NewRowRender(w io.Writer) RowRenderer {
-	return &pw{w, make([][]string, 0)}
+func NewRowRender() RowRenderer {
+	return &pw{make([]row, 0)}
 }
 
+type row struct {
+	header string
+	value  string
+}
 type pw struct {
-	w    io.Writer
-	rows [][]string
+	rows []row
 }
 
 // AddRow adds a row to writer.
 func (p *pw) AddRow(field, value string) {
-	r := []string{field, value}
+	r := row{header: field, value: value}
 	p.rows = append(p.rows, r)
 }
 
@@ -42,15 +45,19 @@ func (p *pw) AddRowMap(values map[string]string) {
 	sort.Strings(keys)
 
 	for _, name := range keys {
-		r := []string{name, values[name]}
+		r := row{header: name, value: values[name]}
 		p.rows = append(p.rows, r)
 	}
-
 }
 
 // Render the rows to output
-func (p *pw) RenderTo(w io.Writer) {
-	for _, l := range p.rows {
-		fmt.Fprintf(p.w, "%s: %s\n", strings.ToUpper(l[0]), l[1])
+func (p *pw) Write(w io.Writer) (int, error) {
+	for _, row := range p.rows {
+		if row.header == "" {
+			fmt.Fprintf(w, "%s\n", row.value)
+		} else {
+			fmt.Fprintf(w, "%s: %s\n", strings.ToUpper(row.header), row.value)
+		}
 	}
+	return 0, nil
 }

printer_test.go

@@ -7,9 +7,9 @@ import (
 
 func TestRowPrinter(t *testing.T) {
 	buf := new(bytes.Buffer)
-	pw := &pw{buf, make([][]string, 0)}
+	pw := &pw{make([]row, 0)}
 	pw.AddRow("FILE", "bar")
-	pw.RenderTo(buf)
+	pw.Write(buf)
 
 	got := buf.String()
 	want := "FILE: bar\n"