This repository has been archived on 2025-03-01. You can view files and clone it, but cannot push or open issues or pull requests.
trantor/cover.go

203 lines
4.6 KiB
Go
Raw Normal View History

package main
import log "github.com/cihub/seelog"
2013-04-25 14:33:50 +02:00
import _ "image/png"
import _ "image/jpeg"
import _ "image/gif"
import (
"bytes"
2013-04-01 14:05:04 +02:00
"git.gitorious.org/go-pkg/epubgo.git"
2013-04-15 22:10:48 +02:00
"github.com/gorilla/mux"
"github.com/nfnt/resize"
"image"
"image/jpeg"
2013-04-01 14:05:04 +02:00
"io"
"io/ioutil"
2013-04-15 22:10:48 +02:00
"labix.org/v2/mgo"
"labix.org/v2/mgo/bson"
"regexp"
"strings"
)
func coverHandler(h handler) {
vars := mux.Vars(h.r)
2013-05-09 09:42:58 +02:00
if !bson.IsObjectIdHex(vars["id"]) {
notFound(h)
2013-05-09 09:42:58 +02:00
return
}
2013-04-15 22:10:48 +02:00
id := bson.ObjectIdHex(vars["id"])
books, _, err := h.db.GetBooks(bson.M{"_id": id})
2013-04-15 22:10:48 +02:00
if err != nil || len(books) == 0 {
notFound(h)
2013-04-15 22:10:48 +02:00
return
}
book := books[0]
if !book.Active {
if !h.sess.IsAdmin() {
notFound(h)
2013-04-15 22:10:48 +02:00
return
}
}
fs := h.db.GetFS(FS_IMGS)
2013-04-15 22:10:48 +02:00
var f *mgo.GridFile
if vars["size"] == "small" {
f, err = fs.OpenId(book.CoverSmall)
} else {
f, err = fs.OpenId(book.Cover)
}
if err != nil {
2014-02-11 13:13:43 +01:00
log.Error("Error while opening image: ", err)
notFound(h)
2013-04-15 22:10:48 +02:00
return
}
defer f.Close()
headers := h.w.Header()
2013-04-15 22:10:48 +02:00
headers["Content-Type"] = []string{"image/jpeg"}
io.Copy(h.w, f)
2013-04-15 22:10:48 +02:00
}
func GetCover(e *epubgo.Epub, title string, db *DB) (bson.ObjectId, bson.ObjectId) {
imgId, smallId := coverFromMetadata(e, title, db)
2013-07-18 21:08:22 +02:00
if imgId != "" {
return imgId, smallId
}
imgId, smallId = searchCommonCoverNames(e, title, db)
2013-04-23 01:04:47 +02:00
if imgId != "" {
return imgId, smallId
}
/* search for img on the text */
2013-07-18 20:28:35 +02:00
exp, _ := regexp.Compile("<.*ima?g.*[(src)(href)]=[\"']([^\"']*(\\.[^\\.\"']*))[\"']")
2013-04-01 14:05:04 +02:00
it, errNext := e.Spine()
for errNext == nil {
file, err := it.Open()
if err != nil {
break
}
defer file.Close()
txt, err := ioutil.ReadAll(file)
if err != nil {
break
}
res := exp.FindSubmatch(txt)
if res != nil {
2013-04-01 14:05:04 +02:00
href := string(res[1])
2013-06-04 21:30:37 +02:00
urlPart := strings.Split(it.URL(), "/")
url := strings.Join(urlPart[:len(urlPart)-1], "/")
2013-04-01 14:05:04 +02:00
if href[:3] == "../" {
href = href[3:]
url = strings.Join(urlPart[:len(urlPart)-2], "/")
}
2013-04-01 14:05:04 +02:00
href = strings.Replace(href, "%20", " ", -1)
href = strings.Replace(href, "%27", "'", -1)
href = strings.Replace(href, "%28", "(", -1)
href = strings.Replace(href, "%29", ")", -1)
if url == "" {
2013-04-01 14:05:04 +02:00
url = href
} else {
2013-04-01 14:05:04 +02:00
url = url + "/" + href
}
2013-04-01 14:05:04 +02:00
img, err := e.OpenFile(url)
if err == nil {
defer img.Close()
return storeImg(img, title, db)
}
}
2013-04-01 14:05:04 +02:00
errNext = it.Next()
}
return "", ""
}
func coverFromMetadata(e *epubgo.Epub, title string, db *DB) (bson.ObjectId, bson.ObjectId) {
2013-07-18 21:08:22 +02:00
metaList, _ := e.MetadataAttr("meta")
for _, meta := range metaList {
if meta["name"] == "cover" {
img, err := e.OpenFileId(meta["content"])
if err == nil {
defer img.Close()
return storeImg(img, title, db)
2013-07-18 21:08:22 +02:00
}
}
}
return "", ""
}
func searchCommonCoverNames(e *epubgo.Epub, title string, db *DB) (bson.ObjectId, bson.ObjectId) {
2013-07-18 21:08:22 +02:00
for _, p := range []string{"cover.jpg", "Images/cover.jpg", "images/cover.jpg", "cover.jpeg", "cover1.jpg", "cover1.jpeg"} {
2013-04-01 14:05:04 +02:00
img, err := e.OpenFile(p)
if err == nil {
defer img.Close()
return storeImg(img, title, db)
2013-04-01 14:05:04 +02:00
}
}
return "", ""
}
func storeImg(img io.Reader, title string, db *DB) (bson.ObjectId, bson.ObjectId) {
/* open the files */
fBig, err := createCoverFile(title, db)
if err != nil {
2014-02-11 13:13:43 +01:00
log.Error("Error creating ", title, ": ", err.Error())
return "", ""
}
defer fBig.Close()
fSmall, err := createCoverFile(title+"_small", db)
if err != nil {
2014-02-11 13:13:43 +01:00
log.Error("Error creating ", title+"_small", ": ", err.Error())
return "", ""
}
defer fSmall.Close()
/* resize img */
2013-04-01 14:05:04 +02:00
var img2 bytes.Buffer
img1 := io.TeeReader(img, &img2)
jpgOptions := jpeg.Options{IMG_QUALITY}
2013-04-01 14:05:04 +02:00
imgResized, err := resizeImg(img1, IMG_WIDTH_BIG)
if err != nil {
2014-02-11 13:13:43 +01:00
log.Error("Error resizing big image: ", err.Error())
return "", ""
}
err = jpeg.Encode(fBig, imgResized, &jpgOptions)
if err != nil {
2014-02-11 13:13:43 +01:00
log.Error("Error encoding big image: ", err.Error())
return "", ""
}
2013-04-01 14:05:04 +02:00
imgSmallResized, err := resizeImg(&img2, IMG_WIDTH_SMALL)
if err != nil {
2014-02-11 13:13:43 +01:00
log.Error("Error resizing small image: ", err.Error())
return "", ""
}
err = jpeg.Encode(fSmall, imgSmallResized, &jpgOptions)
if err != nil {
2014-02-11 13:13:43 +01:00
log.Error("Error encoding small image: ", err.Error())
return "", ""
}
2013-04-15 22:10:48 +02:00
idBig, _ := fBig.Id().(bson.ObjectId)
idSmall, _ := fSmall.Id().(bson.ObjectId)
return idBig, idSmall
}
func createCoverFile(title string, db *DB) (*mgo.GridFile, error) {
2013-04-15 22:10:48 +02:00
fs := db.GetFS(FS_IMGS)
2013-04-23 01:04:47 +02:00
return fs.Create(title + ".jpg")
}
2013-04-01 14:05:04 +02:00
func resizeImg(imgReader io.Reader, width uint) (image.Image, error) {
img, _, err := image.Decode(imgReader)
if err != nil {
return nil, err
}
return resize.Resize(width, 0, img, resize.NearestNeighbor), nil
}