package main import ( "flag" "fmt" "io" "io/ioutil" "log" "net/http" "os" "path" "strings" ) const usage = `Usage: iiifdownloader url Downloads all pages from a IIIF server. Currently supports the following IIIF using services: - BNF's Gallica (any book or page URL should work) ` const bnfPrefix = `https://gallica.bnf.fr/ark:/` func filesAreIdentical(fn1, fn2 string) (bool, error) { f1, err := os.Open(fn1) defer f1.Close() if err != nil { return false, fmt.Errorf("Error opening file %s: %v\n", fn1, err) } b1, err := ioutil.ReadAll(f1) if err != nil { return false, fmt.Errorf("Error reading file %s: %v\n", fn1, err) } f2, err := os.Open(fn2) defer f2.Close() if err != nil { return false, fmt.Errorf("Error opening file %s: %v\n", fn2, err) } b2, err := ioutil.ReadAll(f2) if err != nil { return false, fmt.Errorf("Error reading file %s: %v\n", fn2, err) } for i, _ := range b1 { if b1[i] != b2[i] { return false, nil } } return true, nil } func main() { flag.Usage = func() { fmt.Fprintf(flag.CommandLine.Output(), usage) flag.PrintDefaults() } flag.Parse() if flag.NArg() < 1 { flag.Usage() return } url := flag.Arg(0) var bookdir string var pgurlStart, pgurlEnd string var pgurlAltStart, pgurlAltEnd string //var pgNums []int var noPgNums bool switch { case strings.HasPrefix(url, bnfPrefix): f := strings.Split(url[len(bnfPrefix):], "/") if len(f) < 2 { log.Fatalln("Failed to extract BNF book ID from URL") } var lastpart string dot := strings.Index(f[1], ".") if dot == -1 { lastpart = f[1] } else { lastpart = f[1][0:dot] } bookid := f[0] + "/" + lastpart bookdir = f[0] + "-" + lastpart pgurlStart = "https://gallica.bnf.fr/iiif/ark:/" + bookid + "/f" pgurlEnd = "/full/full/0/native.jpg" noPgNums = true // BNF don't have all pages available from IIIF, but they do have // the missing ones in less good quality from an alternative URL. pgurlAltStart = "https://gallica.bnf.fr/ark:/" + bookid + "/f" pgurlAltEnd = ".highres" default: log.Fatalln("Error: generic IIIF downloading not supported yet") } err := os.MkdirAll(bookdir, 0777) if err != nil { log.Fatalf("Error creating book dir: %v\n", err) } if noPgNums { pgnum := 0 for { pgnum++ fmt.Printf("Downloading page %d\n", pgnum) fn := path.Join(bookdir, fmt.Sprintf("%04d.jpg", pgnum)) _, err = os.Stat(fn) if err == nil || os.IsExist(err) { fmt.Printf("Skipping already present page %d\n", pgnum) continue } u := fmt.Sprintf("%s%d%s", pgurlStart, pgnum, pgurlEnd) resp, err := http.Get(u) if err != nil { log.Fatalf("Error downloading page %d, %s: %v\n", pgnum, u, err) } defer resp.Body.Close() switch { case resp.StatusCode == http.StatusNotFound: fmt.Printf("Got 404, assuming end of pages, for page %d, %s\n", pgnum, u) return case resp.StatusCode != http.StatusOK: fmt.Printf("Error downloading page %d, %s: HTTP Code %s\n", pgnum, u, resp.Status) if pgurlAltStart == "" && pgurlAltEnd == "" { log.Fatalln("No alternative URL to try, book failed (or ended, hopefully)") } fmt.Printf("Trying to redownload page %d at lower quality\n", pgnum) u = fmt.Sprintf("%s%d%s", pgurlAltStart, pgnum, pgurlAltEnd) resp, err = http.Get(u) if err != nil { log.Fatalf("Error downloading page %d, %s: %v\n", pgnum, u, err) } defer resp.Body.Close() if resp.StatusCode != http.StatusOK { log.Fatalf("Error downloading page %d, %s: HTTP Code %s\n", pgnum, u, resp.Status) } } f, err := os.Create(fn) defer f.Close() if err != nil { log.Fatalf("Error creating file %s: %v\n", fn, err) } _, err = io.Copy(f, resp.Body) if err != nil { log.Fatalf("Error writing file %s: %v\n", fn, err) } // Close once finished with, as defer won't trigger until the end of the function resp.Body.Close() f.Close() // Check that the last two downloaded files aren't identical, as this // can happen when there are no more pages to download. if pgnum == 1 { continue } fn2 := path.Join(bookdir, fmt.Sprintf("%04d.jpg", pgnum-1)) identical, err := filesAreIdentical(fn, fn2) if err != nil { log.Fatalf("Error checking for files being identical: %v\n", err) } if identical { fmt.Println("Last 2 pages were identical, looks like it's the end of the book") err = os.Remove(fn) if err != nil { log.Fatalf("Error removing dupilicate page %d: %v", fn, err) } err = os.Remove(fn2) if err != nil { log.Fatalf("Error removing dupilicate page %d: %v", fn2, err) } return } } } }