page_image_detail.go

package main

import (
	"fmt"
	"git.kuschku.de/justjanne/imghost/shared"
	_ "github.com/lib/pq"
	"net/http"
	"os"
	"path"
)

type ImageDetailData struct {
	User    shared.UserInfo
	Image   shared.Image
	IsMine  bool
	BaseUrl string
}

func pageImageDetail(env PageEnvironment) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		user := shared.ParseUser(r)
		_, imageId := path.Split(r.URL.Path)

		result, err := env.Database.Query(`
			SELECT
				id,
				owner,
				title,
				description,
				created_at,
				original_name,
				type
			FROM images
			WHERE id = $1
			`, imageId)
		if err != nil {
			formatError(w, ErrorData{http.StatusInternalServerError, user, r.URL, err}, "html")
			return
		}
		defer result.Close()

		var info shared.Image

		if result.Next() {
			var owner string
			if err := result.Scan(&info.Id, &owner, &info.Title, &info.Description, &info.CreatedAt, &info.OriginalName, &info.MimeType); err != nil {
				formatError(w, ErrorData{http.StatusInternalServerError, user, r.URL, err}, "html")
				return
			}

			switch r.PostFormValue("action") {
			case "update":
				if _, err := env.Database.Exec(
					"UPDATE images SET title = $1, description = $2 WHERE id = $3 AND owner = $4",
					r.PostFormValue("title"),
					r.PostFormValue("description"),
					info.Id,
					user.Id,
				); err != nil {
					formatError(w, ErrorData{http.StatusInternalServerError, user, r.URL, err}, "html")
					return
				}
				if r.PostFormValue("from_js") == "true" {
					if err := returnJson(w, true); err != nil {
						formatError(w, ErrorData{http.StatusInternalServerError, user, r.URL, err}, "html")
						return
					}
				} else {
					http.Redirect(w, r, r.URL.Path, http.StatusFound)
				}
				return
			case "delete":
				if _, err := env.Database.Exec("DELETE FROM images WHERE id = $1 AND owner = $2", info.Id, user.Id); err != nil {
					formatError(w, ErrorData{http.StatusInternalServerError, user, r.URL, err}, "html")
					return
				}
				for _, definition := range env.Config.Sizes {
					if err := os.Remove(path.Join(env.Config.TargetFolder, fmt.Sprintf("%s%s", info.Id, definition.Suffix))); err != nil {
						formatError(w, ErrorData{http.StatusInternalServerError, user, r.URL, err}, "html")
						return
					}
				}
				http.Redirect(w, r, "/me/images", http.StatusFound)
				return
			}

			if err = formatTemplate(w, "image_detail.html", ImageDetailData{
				user,
				info,
				owner == user.Id,
				env.Config.BaseUrl,
			}); err != nil {
				formatError(w, ErrorData{http.StatusInternalServerError, user, r.URL, err}, "html")
				return
			}
			return
		}

		if err := returnError(w, http.StatusNotFound, "Image Not Found"); err != nil {
			formatError(w, ErrorData{http.StatusInternalServerError, user, r.URL, err}, "html")
			return
		}
	})
}