diff --git a/admin.go b/admin.go index 8b1b38d..ec482f5 100644 --- a/admin.go +++ b/admin.go @@ -3,7 +3,6 @@ package main import ( "labix.org/v2/mgo/bson" "net/http" - "os" "strings" ) @@ -29,13 +28,7 @@ func deleteHandler(w http.ResponseWriter, r *http.Request) { return } book := books[0] - if book.Cover != "" { - os.RemoveAll(book.Cover[1:]) - } - if book.CoverSmall != "" { - os.RemoveAll(book.CoverSmall[1:]) - } - os.RemoveAll(book.Path) + DeleteBook(book) db.RemoveBook(id) if !book.Active { diff --git a/store.go b/store.go index ffb5ff0..d2aa250 100644 --- a/store.go +++ b/store.go @@ -77,6 +77,16 @@ func StoreBook(book Book) (path string, err error) { return } +func DeleteBook(book Book) { + if book.Cover != "" { + os.RemoveAll(book.Cover[1:]) + } + if book.CoverSmall != "" { + os.RemoveAll(book.CoverSmall[1:]) + } + os.RemoveAll(book.Path) +} + func validFileName(path string, title string, extension string) string { title = strings.Replace(title, "/", "_", -1) title = strings.Replace(title, "?", "_", -1)