From: Diogo Franco (Kovensky) <diogomfranco@gmail.com>
Date: Fri, 26 Jul 2013 05:22:40 +0000 (-0300)
Subject: mylistadd: Add a simple mylist adder as an example program
X-Git-Url: https://git.lizzy.rs/?a=commitdiff_plain;h=3c5d74fe9479763780e121df543ec32da2f06047;p=go-anidb.git

mylistadd: Add a simple mylist adder as an example program

There's no guarantee that the API is stable yet.
---

diff --git a/examples/mylistadd/main.go b/examples/mylistadd/main.go
new file mode 100644
index 0000000..458200f
--- /dev/null
+++ b/examples/mylistadd/main.go
@@ -0,0 +1,115 @@
+package main
+
+import (
+	"encoding/hex"
+	"flag"
+	"fmt"
+	"github.com/Kovensky/go-anidb"
+	ed2khash "github.com/Kovensky/go-ed2k"
+	"io"
+	"os"
+)
+
+var (
+	username = flag.String("username", "", "AniDB Username")
+	password = flag.String("password", "", "AniDB Password")
+	apikey   = flag.String("apikey", "", "UDP API key (optional)")
+)
+
+type ProgressReader struct {
+	io.Reader
+
+	Prefix  string
+	Size    int64
+	pos     int64
+	prevpos int64
+}
+
+func (r *ProgressReader) Read(p []byte) (n int, err error) {
+	n, err = r.Reader.Read(p)
+
+	if r.pos-512*1024 > r.prevpos || r.prevpos == 0 {
+		// only every 512KB
+		fmt.Printf("%s%.2f%%\r", r.Prefix, float64(r.pos)*100/float64(r.Size))
+		r.prevpos = r.pos
+	}
+	r.pos += int64(n)
+	return
+}
+
+func (r *ProgressReader) Close() (err error) {
+	fmt.Printf("%s%.2f%%\n", r.Prefix, float64(r.pos)*100/float64(r.Size))
+	return nil
+}
+
+func hashFile(path string) (ed2k string, size int64) {
+	fh, err := os.Open(path)
+	if err != nil {
+		return
+	}
+	defer fh.Close()
+
+	stat, err := fh.Stat()
+	if err != nil {
+		return
+	}
+	size = stat.Size()
+
+	rd := ProgressReader{
+		Reader: fh,
+		Prefix: fmt.Sprintf("Hashing %s: ", path),
+		Size:   size,
+	}
+	defer rd.Close()
+
+	hash := ed2khash.New(true)
+	_, err = io.Copy(hash, &rd)
+	if err != nil {
+		return
+	}
+
+	ed2k = hex.EncodeToString(hash.Sum(nil))
+	return
+}
+
+func main() {
+	flag.Parse()
+
+	if *username == "" || *password == "" {
+		fmt.Println("Username and password must be supplied")
+		os.Exit(1)
+	}
+
+	adb := anidb.NewAniDB()
+	adb.SetCredentials(*username, *password, *apikey)
+	defer adb.Logout()
+
+	max := len(flag.Args())
+	done := make(chan bool, max)
+
+	for _, path := range flag.Args() {
+		ed2k, size := hashFile(path)
+		if ed2k != "" {
+			go func() {
+				f := <-adb.FileByEd2kSize(ed2k, size)
+				state := anidb.MyListStateHDD
+				done <- <-adb.MyListAdd(f, &anidb.MyListSet{State: &state}) != 0
+			}()
+		} else {
+			go func() { done <- false }()
+		}
+	}
+
+	count := 0
+	for ok := range done {
+		if ok {
+			count++
+		}
+		max--
+		if max == 0 {
+			break
+		}
+	}
+
+	fmt.Println("Added", count, "files to mylist")
+}