1 files changed, 69 insertions, 0 deletions

diff --git a/tools/charcrawler/crawler/anidb.go b/tools/charcrawler/crawler/anidb.go
new file mode 100644
index 0000000..12bf05b
--- /dev/null
+++ b/tools/charcrawler/crawler/anidb.go
@@ -0,0 +1,69 @@
+package crawler
+
+import (
+        "log"
+        "strconv"
+        "strings"
+
+        "github.com/PuerkitoBio/goquery"
+)
+
+type AniDBCrawler struct{}
+
+func (a AniDBCrawler) Name() string {
+        return "AniDB"
+}
+
+func (a AniDBCrawler) Crawl(id int) (CharacterData, error) {
+        c := make(CharacterData)
+        doc, err := goquery.NewDocument("http://anidb.net/perl-bin/animedb.pl?show=character&charid=" + strconv.Itoa(id))
+        if err != nil {
+                log.Println(err)
+                return nil, CrawlError
+        }
+        siteErr, _ := doc.Find(".error p").Html()
+        if strings.Contains(siteErr, "Unknown character id") {
+                return nil, CharacterNotFound
+        } else if strings.Contains(siteErr, "BANNED") {
+                return nil, CrawlError
+        }
+        first := true
+        links := make([]string, 0)
+        doc.Find(".characters").Each(func(i int, s *goquery.Selection) {
+                if first {
+                        s.Find(".g_definitionlist tr").Each(func(i int, s *goquery.Selection) {
+                                keyHtml, _ := s.Find("th").Html()
+                                valueHtml, _ := s.Find("td").Html()
+                                key := strings.TrimSpace(keyHtml)
+                                value := strings.TrimSpace(valueHtml)
+                                c[key] = value
+                        })
+                        desc, _ := s.Find(".desc").Html()
+                        c["__desc"] = strings.TrimSpace(desc)
+                        imgUrl, _ := s.Find(".image img").Attr("src")
+                        c["__img"] = imgUrl
+                        animes := make([]map[string]string, 0)
+                        s.Find(".animelist tr").Each(func(i int, s *goquery.Selection) {
+                                if s.HasClass("header") {
+                                        return
+                                }
+                                anime := make(map[string]string)
+                                relHtml, _ := s.Find(".relation").Html()
+                                nameHtml, _ := s.Find(".name a").Html()
+                                rel := strings.TrimSpace(relHtml)
+                                name := strings.TrimSpace(nameHtml)
+                                anime["rel"] = rel
+                                anime["name"] = name
+                                animes = append(animes, anime)
+                        })
+                        c["__animes"] = animes
+                } else {
+                        link, _ := s.Find(".mainname a").Attr("href")
+                        links = append(links, strings.Replace(link, "http://anidb.net/ch", "", 1))
+                }
+                first = false
+        })
+        c["__links"] = links
+
+        return c, nil
+}