@peaske7/readit 0.2.0 → 0.2.1

package/go/cmd/readit/main.go

@@ -0,0 +1,416 @@
+package main
+
+import (
+	"bytes"
+	"encoding/json"
+	"flag"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"os/exec"
+	"os/signal"
+	"path/filepath"
+	"strings"
+	"syscall"
+	"time"
+
+	"github.com/peaske7/readit/go/internal/server"
+	"github.com/pkg/browser"
+)
+
+const version = "0.3.0"
+
+type serverInfo struct {
+	Port int    `json:"port"`
+	PID  int    `json:"pid"`
+	Host string `json:"host,omitempty"`
+}
+
+func main() {
+	if len(os.Args) < 2 {
+		fmt.Println("Usage: readit <file.md> [flags]")
+		fmt.Println("       readit list")
+		fmt.Println("       readit show <file.md>")
+		fmt.Println("       readit open <file.md>")
+		os.Exit(1)
+	}
+
+	switch os.Args[1] {
+	case "list":
+		cmdList()
+	case "show":
+		cmdShow()
+	case "open":
+		cmdOpen()
+	case "version", "--version", "-v":
+		fmt.Println("readit", version)
+	case "help", "--help", "-h":
+		printHelp()
+	default:
+		cmdServe()
+	}
+}
+
+func cmdServe() {
+	fs := flag.NewFlagSet("serve", flag.ExitOnError)
+	port := fs.Int("port", 0, "port to listen on (default: random)")
+	host := fs.String("host", "127.0.0.1", "host to bind to")
+	noOpen := fs.Bool("no-open", false, "don't open browser")
+	clean := fs.Bool("clean", false, "clear existing comments")
+	assetsDir := fs.String("assets-dir", "", "serve assets from directory instead of embedded")
+	dev := fs.Bool("dev", false, "development mode (proxy to Vite)")
+
+	// Separate file args from flag args. Flags that take values
+	// (--port 3000, --host 0.0.0.0, --assets-dir ./dist) consume the next arg.
+	valueFlags := map[string]bool{
+		"-port": true, "--port": true,
+		"-host": true, "--host": true,
+		"-assets-dir": true, "--assets-dir": true,
+	}
+
+	var fileArgs []string
+	var flagArgs []string
+	args := os.Args[1:]
+	for i := 0; i < len(args); i++ {
+		arg := args[i]
+		if strings.HasPrefix(arg, "-") {
+			flagArgs = append(flagArgs, arg)
+			// If this flag takes a value, consume the next arg too
+			if valueFlags[arg] && i+1 < len(args) {
+				i++
+				flagArgs = append(flagArgs, args[i])
+			}
+		} else {
+			fileArgs = append(fileArgs, arg)
+		}
+	}
+	_ = fs.Parse(flagArgs)
+
+	if len(fileArgs) == 0 {
+		fmt.Fprintln(os.Stderr, "Error: at least one file is required")
+		os.Exit(1)
+	}
+
+	files, err := resolveFiles(fileArgs)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		os.Exit(1)
+	}
+
+	if *port == 0 {
+		*port = 4567
+	}
+
+	// Dev mode: spawn Vite
+	var viteCmd *exec.Cmd
+	if *dev {
+		viteCmd = spawnVite()
+	}
+
+	// Setup signal channel before server so the SSE shutdown callback can use it
+	sig := make(chan os.Signal, 1)
+	signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
+
+	srv, err := server.NewServer(server.Options{
+		Files:     files,
+		Port:      *port,
+		Host:      *host,
+		Clean:     *clean,
+		AssetsDir: *assetsDir,
+		Dev:       *dev,
+		OnShutdown: func() {
+			log.Println("All clients disconnected, shutting down")
+			sig <- syscall.SIGTERM
+		},
+	})
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	actualPort, err := srv.Start(*host, *port)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	writeServerInfo(actualPort, *host)
+
+	url := fmt.Sprintf("http://%s:%d", *host, actualPort)
+	fmt.Printf("readit v%s serving at %s\n", version, url)
+
+	if !*noOpen {
+		_ = browser.OpenURL(url)
+	}
+
+	// Wait for interrupt or SSE-triggered shutdown
+	<-sig
+
+	fmt.Println("\nShutting down...")
+	srv.Stop()
+	if viteCmd != nil && viteCmd.Process != nil {
+		_ = viteCmd.Process.Kill()
+	}
+	removeServerInfo()
+}
+
+func cmdList() {
+	home, _ := os.UserHomeDir()
+	commentsDir := filepath.Join(home, ".readit", "comments")
+
+	_ = filepath.Walk(commentsDir, func(path string, info os.FileInfo, err error) error {
+		if err != nil || info.IsDir() {
+			return nil
+		}
+		if !strings.HasSuffix(path, ".comments.md") {
+			return nil
+		}
+
+		data, err := os.ReadFile(path)
+		if err != nil {
+			return nil
+		}
+
+		cf, err := server.ParseCommentFile(data)
+		if err != nil {
+			return nil
+		}
+
+		fmt.Printf("%s (%d comments)\n", cf.Source, len(cf.Comments))
+		return nil
+	})
+}
+
+func cmdShow() {
+	if len(os.Args) < 3 {
+		fmt.Fprintln(os.Stderr, "Usage: readit show <file.md>")
+		os.Exit(1)
+	}
+
+	filePath, err := filepath.Abs(os.Args[2])
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		os.Exit(1)
+	}
+
+	commentPath, err := server.CommentPath(filePath)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		os.Exit(1)
+	}
+	data, err := os.ReadFile(commentPath)
+	if err != nil {
+		fmt.Println("No comments found for", filePath)
+		return
+	}
+
+	cf, err := server.ParseCommentFile(data)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		os.Exit(1)
+	}
+
+	for i, c := range cf.Comments {
+		text := c.SelectedText
+		if len(text) > 80 {
+			text = text[:80] + "…"
+		}
+		fmt.Printf("[%d] %s\n", i+1, c.LineHint)
+		fmt.Printf("    > %s\n", strings.ReplaceAll(text, "\n", "\n    > "))
+		fmt.Printf("    %s\n", c.Comment)
+		fmt.Printf("    (%s)\n\n", c.CreatedAt)
+	}
+}
+
+func cmdOpen() {
+	if len(os.Args) < 3 {
+		fmt.Fprintln(os.Stderr, "Usage: readit open <file.md>")
+		os.Exit(1)
+	}
+
+	files, err := resolveFiles(os.Args[2:])
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		os.Exit(1)
+	}
+
+	// Try to discover a running server
+	info, err := discoverServer()
+	if err == nil && info != nil {
+		// Attach to existing server
+		for _, f := range files {
+			attachFile(info, f.FilePath)
+		}
+		url := fmt.Sprintf("http://%s:%d", info.resolvedHost(), info.Port)
+		_ = browser.OpenURL(url)
+		return
+	}
+
+	// No running server — start one
+	os.Args = append([]string{os.Args[0]}, os.Args[2:]...)
+	cmdServe()
+}
+
+func resolveFiles(args []string) ([]server.FileEntry, error) {
+	seen := make(map[string]bool)
+	var files []server.FileEntry
+
+	for _, arg := range args {
+		absPath, err := filepath.Abs(arg)
+		if err != nil {
+			return nil, fmt.Errorf("invalid path %s: %w", arg, err)
+		}
+
+		absPath, err = filepath.EvalSymlinks(absPath)
+		if err != nil {
+			return nil, fmt.Errorf("file not found: %s", arg)
+		}
+
+		info, err := os.Stat(absPath)
+		if err != nil {
+			return nil, fmt.Errorf("file not found: %s", arg)
+		}
+
+		if info.IsDir() {
+			// Scan directory for markdown files
+			_ = filepath.Walk(absPath, func(path string, fi os.FileInfo, err error) error {
+				if err != nil || fi.IsDir() {
+					if fi != nil && fi.IsDir() && strings.HasPrefix(fi.Name(), ".") {
+						return filepath.SkipDir
+					}
+					if fi != nil && fi.IsDir() && fi.Name() == "node_modules" {
+						return filepath.SkipDir
+					}
+					return nil
+				}
+				ext := strings.ToLower(filepath.Ext(path))
+				if ext == ".md" || ext == ".markdown" {
+					if !seen[path] {
+						seen[path] = true
+						files = append(files, server.FileEntry{FilePath: path})
+					}
+				}
+				return nil
+			})
+		} else {
+			ext := strings.ToLower(filepath.Ext(absPath))
+			if ext != ".md" && ext != ".markdown" {
+				return nil, fmt.Errorf("unsupported file type: %s (only .md and .markdown)", arg)
+			}
+			if !seen[absPath] {
+				seen[absPath] = true
+				files = append(files, server.FileEntry{FilePath: absPath})
+			}
+		}
+	}
+
+	if len(files) == 0 {
+		return nil, fmt.Errorf("no markdown files found")
+	}
+
+	return files, nil
+}
+
+func serverInfoPath() string {
+	home, _ := os.UserHomeDir()
+	return filepath.Join(home, ".readit", "server.json")
+}
+
+func writeServerInfo(port int, host string) {
+	info := serverInfo{Port: port, PID: os.Getpid(), Host: host}
+	data, _ := json.Marshal(info)
+	path := serverInfoPath()
+	_ = os.MkdirAll(filepath.Dir(path), 0755)
+	_ = os.WriteFile(path, data, 0644)
+}
+
+func removeServerInfo() {
+	_ = os.Remove(serverInfoPath())
+}
+
+func (si *serverInfo) resolvedHost() string {
+	if si.Host != "" {
+		return si.Host
+	}
+	return "127.0.0.1"
+}
+
+func discoverServer() (*serverInfo, error) {
+	data, err := os.ReadFile(serverInfoPath())
+	if err != nil {
+		return nil, err
+	}
+
+	var info serverInfo
+	if err := json.Unmarshal(data, &info); err != nil {
+		return nil, err
+	}
+
+	// Check if process is alive
+	proc, err := os.FindProcess(info.PID)
+	if err != nil {
+		return nil, err
+	}
+	if err := proc.Signal(syscall.Signal(0)); err != nil {
+		return nil, fmt.Errorf("process not alive")
+	}
+
+	// Health check
+	client := &http.Client{Timeout: 2 * time.Second}
+	resp, err := client.Get(fmt.Sprintf("http://%s:%d/api/health", info.resolvedHost(), info.Port))
+	if err != nil {
+		return nil, err
+	}
+	_ = resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("health check failed")
+	}
+
+	return &info, nil
+}
+
+func attachFile(info *serverInfo, filePath string) {
+	body, _ := json.Marshal(map[string]string{"path": filePath})
+	resp, err := http.Post(
+		fmt.Sprintf("http://%s:%d/api/documents", info.resolvedHost(), info.Port),
+		"application/json",
+		bytes.NewReader(body),
+	)
+	if err != nil {
+		log.Printf("Warning: failed to attach %s: %v", filePath, err)
+		return
+	}
+	_ = resp.Body.Close()
+}
+
+func spawnVite() *exec.Cmd {
+	cmd := exec.Command("bunx", "vite", "--port", "24678")
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	if err := cmd.Start(); err != nil {
+		log.Printf("Warning: failed to start Vite: %v", err)
+		return nil
+	}
+	// Give Vite a moment to start
+	time.Sleep(500 * time.Millisecond)
+	return cmd
+}
+
+func printHelp() {
+	fmt.Printf(`readit v%s — Review Markdown documents with inline comments
+
+Usage:
+  readit <file.md> [flags]     Start server and open browser
+  readit list                  List files with comments
+  readit show <file.md>        Show comments for a file
+  readit open <file.md>        Add file to running server
+
+Flags:
+  --port <n>        Port to listen on (default: 4567)
+  --host <addr>     Host to bind to (default: 127.0.0.1)
+  --no-open         Don't auto-open browser
+  --clean           Clear existing comments
+  --assets-dir <d>  Serve assets from directory
+  --dev             Development mode (spawn Vite)
+
+`, version)
+}

package/go/go.mod

@@ -0,0 +1,20 @@
+module github.com/peaske7/readit/go
+
+go 1.24
+
+require (
+	github.com/alecthomas/chroma/v2 v2.23.1
+	github.com/fsnotify/fsnotify v1.9.0
+	github.com/microcosm-cc/bluemonday v1.0.27
+	github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c
+	github.com/yuin/goldmark v1.8.2
+	github.com/yuin/goldmark-highlighting/v2 v2.0.0-20230729083705-37449abec8cc
+)
+
+require (
+	github.com/aymerick/douceur v0.2.0 // indirect
+	github.com/dlclark/regexp2 v1.11.5 // indirect
+	github.com/gorilla/css v1.0.1 // indirect
+	golang.org/x/net v0.26.0 // indirect
+	golang.org/x/sys v0.21.0 // indirect
+)

package/go/go.sum

@@ -0,0 +1,41 @@
+github.com/alecthomas/assert/v2 v2.11.0 h1:2Q9r3ki8+JYXvGsDyBXwH3LcJ+WK5D0gc5E8vS6K3D0=
+github.com/alecthomas/assert/v2 v2.11.0/go.mod h1:Bze95FyfUr7x34QZrjL+XP+0qgp/zg8yS+TtBj1WA3k=
+github.com/alecthomas/chroma/v2 v2.2.0/go.mod h1:vf4zrexSH54oEjJ7EdB65tGNHmH3pGZmVkgTP5RHvAs=
+github.com/alecthomas/chroma/v2 v2.23.1 h1:nv2AVZdTyClGbVQkIzlDm/rnhk1E9bU9nXwmZ/Vk/iY=
+github.com/alecthomas/chroma/v2 v2.23.1/go.mod h1:NqVhfBR0lte5Ouh3DcthuUCTUpDC9cxBOfyMbMQPs3o=
+github.com/alecthomas/repr v0.0.0-20220113201626-b1b626ac65ae/go.mod h1:2kn6fqh/zIyPLmm3ugklbEi5hg5wS435eygvNfaDQL8=
+github.com/alecthomas/repr v0.5.2 h1:SU73FTI9D1P5UNtvseffFSGmdNci/O6RsqzeXJtP0Qs=
+github.com/alecthomas/repr v0.5.2/go.mod h1:Fr0507jx4eOXV7AlPV6AVZLYrLIuIeSOWtW57eE/O/4=
+github.com/aymerick/douceur v0.2.0 h1:Mv+mAeH1Q+n9Fr+oyamOlAkUNPWPlA8PPGR0QAaYuPk=
+github.com/aymerick/douceur v0.2.0/go.mod h1:wlT5vV2O3h55X9m7iVYN0TBM0NH/MmbLnd30/FjWUq4=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/dlclark/regexp2 v1.4.0/go.mod h1:2pZnwuY/m+8K6iRw6wQdMtk+rH5tNGR1i55kozfMjCc=
+github.com/dlclark/regexp2 v1.7.0/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
+github.com/dlclark/regexp2 v1.11.5 h1:Q/sSnsKerHeCkc/jSTNq1oCm7KiVgUMZRDUoRu0JQZQ=
+github.com/dlclark/regexp2 v1.11.5/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
+github.com/fsnotify/fsnotify v1.9.0 h1:2Ml+OJNzbYCTzsxtv8vKSFD9PbJjmhYF14k/jKC7S9k=
+github.com/fsnotify/fsnotify v1.9.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0=
+github.com/gorilla/css v1.0.1 h1:ntNaBIghp6JmvWnxbZKANoLyuXTPZ4cAMlo6RyhlbO8=
+github.com/gorilla/css v1.0.1/go.mod h1:BvnYkspnSzMmwRK+b8/xgNPLiIuNZr6vbZBTPQ2A3b0=
+github.com/hexops/gotextdiff v1.0.3 h1:gitA9+qJrrTCsiCl7+kh75nPqQt1cx4ZkudSTLoUqJM=
+github.com/hexops/gotextdiff v1.0.3/go.mod h1:pSWU5MAI3yDq+fZBTazCSJysOMbxWL1BSow5/V2vxeg=
+github.com/microcosm-cc/bluemonday v1.0.27 h1:MpEUotklkwCSLeH+Qdx1VJgNqLlpY2KXwXFM08ygZfk=
+github.com/microcosm-cc/bluemonday v1.0.27/go.mod h1:jFi9vgW+H7c3V0lb6nR74Ib/DIB5OBs92Dimizgw2cA=
+github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c h1:+mdjkGKdHQG3305AYmdv1U2eRNDiU2ErMBj1gwrq8eQ=
+github.com/pkg/browser v0.0.0-20240102092130-5ac0b6a4141c/go.mod h1:7rwL4CYBLnjLxUqIJNnCWiEdr3bn6IUYi15bNlnbCCU=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/yuin/goldmark v1.4.15/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
+github.com/yuin/goldmark v1.8.2 h1:kEGpgqJXdgbkhcOgBxkC0X0PmoPG1ZyoZ117rDVp4zE=
+github.com/yuin/goldmark v1.8.2/go.mod h1:ip/1k0VRfGynBgxOz0yCqHrbZXhcjxyuS66Brc7iBKg=
+github.com/yuin/goldmark-highlighting/v2 v2.0.0-20230729083705-37449abec8cc h1:+IAOyRda+RLrxa1WC7umKOZRsGq4QrFFMYApOeHzQwQ=
+github.com/yuin/goldmark-highlighting/v2 v2.0.0-20230729083705-37449abec8cc/go.mod h1:ovIvrum6DQJA4QsJSovrkC4saKHQVs7TvcaeO8AIl5I=
+golang.org/x/net v0.26.0 h1:soB7SVo0PWrY4vPW/+ay0jKDNScG2X9wFeYlXIvJsOQ=
+golang.org/x/net v0.26.0/go.mod h1:5YKkiSynbBIh3p6iOc/vibscux0x38BZDkn8sCUPxHE=
+golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.21.0 h1:rF+pYz3DAGSQAxAu1CbC7catZg4ebC4UIeIhKxBZvws=
+golang.org/x/sys v0.21.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=