func ListDB(args martini.Params, su models.User, r render.Render, p *models.Page, dbh *db.Dbh) {
images := dbh.GetAllImages()
var album models.Album
for _, image := range images {
album = dbh.GetAlbum(image.Album)
image.AlbumId = album.Id
nImage := dbh.UpdateImage(image)
log.Println(nImage)
}
return
}
func AlbumMove(req *http.Request, r render.Render, su models.User, dbh *db.Dbh, p *models.Page) {
// Default status
var status string = "Permission Denied"
var code int = 401
var reader io.Reader = req.Body
b, e := ioutil.ReadAll(reader)
if e != nil {
log.Println(e)
}
var f Data
err := json.Unmarshal(b, &f)
if err != nil {
log.Println(err)
}
log.Println(f)
albumId := f.Md.AlbumId
album := dbh.GetAlbumById(albumId)
log.Println("album: ", albumId)
if album.User == su.Id {
status = "Success"
code = 200
var image models.Image
for _, f := range f.Md.Images {
imageId, err := strconv.ParseInt(f.Id, 10, 10)
if err != nil {
log.Println("invalid image id: ", f.Id)
}
log.Println("Moving: ", f.Name, imageId)
image = dbh.GetImageById(imageId)
if image.User != su.Id {
status = "Permission Denied"
code = 401
continue
}
fromId := image.AlbumId
image.AlbumId = albumId
_ = dbh.UpdateImage(image)
log.Printf("Moved %s from: %d to: %d status: %s code: %d\n", f.Name, fromId, albumId, status, code)
}
}
p.Data = status
encoder.Render("json", code, "", p, r)
}
func ImageTrash(args martini.Params, su models.User, dbh *db.Dbh, r render.Render, p *models.Page) {
album := args["album"]
name := args["name"]
// Default status
var status string = "Permission Denied"
var code int = 401
log.Printf("Trashing image: %s from %s", name, album)
image := dbh.FirstImageByName(name)
if su.Id == image.User {
image.Trashed = true
dbh.UpdateImage(image)
status = "Success"
code = 200
}
p.SetUser(su)
p.Data = TrashRecover{Name: name, Album: album, Action: "Trash", Status: status}
encoder.Render("json", code, "", p, r)
}
func UploadImage(w http.ResponseWriter, upload models.ImageUpload, req *http.Request, r render.Render, dbh *db.Dbh) {
ur := &models.UploadResult{}
rEmail := upload.Email
rAlbum := upload.Album
rPrivateKey := upload.PrivateKey
fiName := upload.File.Filename
upload_time := time.Now().Unix()
ur.SetCode(200)
ur.SetResult("success")
ur.SetName(fiName)
tmp_file := utils.ImageCfg.Root() + ur.GetName()
if Exists(tmp_file) {
log.Println("Error: File exists. (" + tmp_file + ")")
ur.SetCode(2)
ur.SetResult("File exists")
r.JSON(500, ur)
return
}
out, err := os.Create(tmp_file)
if err != nil {
log.Println("Error: Unable to open file.")
ur.SetCode(500)
ur.SetResult("Failed to open the file for writing.")
r.JSON(500, ur)
return
}
defer out.Close()
fi, err := upload.File.Open()
if err != nil {
log.Println("fi err: ", err)
ur.SetCode(500)
ur.SetResult(err.Error())
r.JSON(500, ur)
return
}
defer fi.Close()
_, err = io.Copy(out, fi)
if err != nil {
log.Println("Error: Failed to copy file.")
ur.SetCode(500)
ur.SetResult("Failed to copy file to new location.")
r.JSON(500, ur)
return
}
log.Printf("tmp_file: %s\n", tmp_file)
// Add image uploader to database
dbh.AddUploader(models.Uploader{Email: rEmail, Timestamp: upload_time})
// Setup hashid to create unique file name
var hid models.HashID
hid.Init(utils.AppCfg.SecretKey(), 10)
// Get user id
user := dbh.GetUserByEmail(rEmail)
log.Println("user: "******"uploader user: "******"album: ", album)
nAlbum := dbh.GetAlbum(rAlbum)
// Add image
image := dbh.AddImage(models.Image{
Name: fiName,
Album: rAlbum,
User: user.Id,
AlbumId: nAlbum.Id,
Timestamp: upload_time})
// Add image id to hashid - seg 3
hid.AddId(int(image.Id))
// Add upload time to hashid - seg 4
hid.AddId(int(upload_time))
// Get file extension and create new file name
extension := filepath.Ext(fiName)
nname := hid.Encrypt() + extension
log.Printf("New name: %s\n", nname)
image.HashId = nname
dbh.UpdateImage(image)
// Rename file to new name
hash_name := utils.ImageCfg.Root() + nname
os.Rename(tmp_file, hash_name)
ur.SetName(utils.AppCfg.Url() + "/image/" + nname)
// Create Thumb
tname := utils.ImageCfg.Thumbs() + nname
if !Exists(string(tname)) {
okc := make(chan bool, 1)
utils.CreateThumb(okc, hash_name, tname, 150, 150)
<-okc
}
log.Printf("%# v\n", pretty.Formatter(album))
log.Printf("%# v\n", image)
log.Printf("%# v\n", pretty.Formatter(ur))
r.JSON(200, ur)
}