From 77d9aa6414691e9d5ee79e4b4f4bbe0d1a703db0 Mon Sep 17 00:00:00 2001
From: David Howden <dhowden@gmail.com>
Date: Sun, 24 May 2015 09:01:33 +1000
Subject: [PATCH] Added tool for checking tags over iTunes Library or path

---
 check/main.go | 183 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 183 insertions(+)
 create mode 100644 check/main.go

diff --git a/check/main.go b/check/main.go
new file mode 100644
index 0000000..b199763
--- /dev/null
+++ b/check/main.go
@@ -0,0 +1,183 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"net/url"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/dhowden/itl"
+	"github.com/dhowden/tag"
+)
+
+var itlXML, path string
+var sum bool
+
+func init() {
+	flag.StringVar(&itlXML, "itlXML", "", "iTunes Library Path")
+	flag.StringVar(&path, "path", "", "path to directory containing audio files")
+	flag.BoolVar(&sum, "sum", false, "compute the checksum of the audio file (doesn't work for .flac or .ogg yet)")
+}
+
+func decodeLocation(l string) (string, error) {
+	u, err := url.ParseRequestURI(l)
+	if err != nil {
+		return "", err
+	}
+	// Annoyingly this doesn't replace &#38; (&)
+	path := strings.Replace(u.Path, "&#38;", "&", -1)
+	return path, nil
+}
+
+func main() {
+	flag.Parse()
+
+	if itlXML == "" && path == "" || itlXML != "" && path != "" {
+		fmt.Println("you must specify one of -itlXML or -path")
+		os.Exit(1)
+	}
+
+	var paths <-chan string
+	if itlXML != "" {
+		var err error
+		paths, err = walkLibrary(itlXML)
+		if err != nil {
+			fmt.Println(err)
+			os.Exit(1)
+		}
+	}
+
+	if path != "" {
+		paths = walkPath(path)
+	}
+
+	p := &processor{
+		decodingErrors: make(map[string]int),
+		hashErrors:     make(map[string]int),
+		hashes:         make(map[string]int),
+	}
+
+	done := make(chan bool)
+	go func() {
+		p.do(paths)
+		fmt.Println(p)
+		close(done)
+	}()
+	<-done
+}
+
+func walkPath(root string) <-chan string {
+	ch := make(chan string)
+	fn := func(path string, info os.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if info.IsDir() {
+			return nil
+		}
+		ch <- path
+		return nil
+	}
+
+	go func() {
+		err := filepath.Walk(root, fn)
+		if err != nil {
+			fmt.Println(err)
+		}
+		close(ch)
+	}()
+	return ch
+}
+
+func walkLibrary(path string) (<-chan string, error) {
+	f, err := os.Open(itlXML)
+	if err != nil {
+		return nil, err
+	}
+	defer f.Close()
+
+	l, err := itl.ReadFromXML(f)
+	if err != nil {
+		return nil, err
+	}
+
+	paths := make(chan string)
+	go func() {
+		for _, t := range l.Tracks {
+			loc, err := decodeLocation(t.Location)
+			if err != nil {
+				fmt.Println(err)
+				break
+			}
+			paths <- loc
+		}
+		close(paths)
+	}()
+	return paths, nil
+}
+
+type processor struct {
+	decodingErrors map[string]int
+	hashErrors     map[string]int
+	hashes         map[string]int
+}
+
+func (p *processor) String() string {
+	result := ""
+	for k, v := range p.decodingErrors {
+		result += fmt.Sprintf("%v : %v\n", k, v)
+	}
+
+	for k, v := range p.hashErrors {
+		result += fmt.Sprintf("%v : %v\n", k, v)
+	}
+
+	for k, v := range p.hashErrors {
+		if v > 1 {
+			result += fmt.Sprintf("%v : %v\n", k, v)
+		}
+	}
+	return result
+}
+
+func (p *processor) do(ch <-chan string) {
+	for path := range ch {
+		func() {
+			defer func() {
+				if p := recover(); p != nil {
+					fmt.Printf("Panicing at: %v", path)
+					panic(p)
+				}
+			}()
+			tf, err := os.Open(path)
+			if err != nil {
+				p.decodingErrors["error opening file"]++
+				return
+			}
+			defer tf.Close()
+
+			_, err = tag.ReadFrom(tf)
+			if err != nil {
+				fmt.Println("READFROM:", path, err.Error())
+				p.decodingErrors[err.Error()]++
+			}
+
+			if sum {
+				_, err = tf.Seek(0, os.SEEK_SET)
+				if err != nil {
+					fmt.Println("DIED:", path, "error seeking back to 0:", err)
+					return
+				}
+
+				h, err := tag.Sum(tf)
+				if err != nil {
+					fmt.Println("SUM:", path, err.Error())
+					p.hashErrors[err.Error()]++
+				}
+				p.hashes[h]++
+			}
+		}()
+	}
+}