package main

import (
	"bufio"
	"flag"
	"io/fs"
	"log"
	"os"
	"path/filepath"
)

func main() {
	var (
		vaultDir string
	)

	flag.StringVar(&vaultDir, "vault", "./vault", "path to Obsidian vault")

	flag.Parse()

	absVaultPath, err := filepath.Abs(vaultDir)
	if err != nil {
		log.Fatalf("error generating absolute path: %s\n", err)
	}
	log.Println("absVaultPath:", absVaultPath)

	vaultRoot := os.DirFS(absVaultPath)

	noteLengthsChar := make(map[string]int)
	noteLengthsWord := make(map[string]int)

	fs.WalkDir(vaultRoot, ".", func(path string, d fs.DirEntry, err error) error {
		absFilePath := filepath.Join(vaultDir, path)
		f, err := os.Open(absFilePath)
		if err != nil {
			log.Printf("error opening note to calculate stats: %s\n", err)
			return err
		}
		fStat, err := f.Stat()
		if err != nil {
			log.Printf("error calling stat() on file: %s\n", err)
			return err
		}
		basePath := filepath.Base(path)
		noteLengthsChar[basePath] = int(fStat.Size())

		r := bufio.NewScanner(f)
		for r.Scan() {
			noteLengthsWord[basePath] += 1
		}
		return nil
	})

	log.Printf("Average character count: %.2f\n", averageMap(noteLengthsChar))
	log.Printf("Average word count: %.2f\n", averageMap(noteLengthsWord))

}

func averageMap(m map[string]int) float64 {
	sum := 0.0
	for _, v := range m {
		sum += float64(v)
	}
	return sum / float64(len(m))

}