golang-fave/modules/module_shop_act_upload_image.go

package modules

import (
	"bytes"
	"image"
	"io/ioutil"
	"os"
	"path/filepath"
	"time"

	_ "image/jpeg"
	_ "image/png"

	"golang-fave/engine/wrapper"
	"golang-fave/utils"
)

func (this *Modules) RegisterAction_ShopUploadImage() *Action {
	return this.newAction(AInfo{
		WantDB:    true,
		Mount:     "shop-upload-image",
		WantAdmin: true,
	}, func(wrap *wrapper.Wrapper) {
		pf_id := wrap.R.FormValue("id")

		if !utils.IsNumeric(pf_id) {
			wrap.MsgError(`Inner system error`)
			return
		}

		// Read file from request
		file, handler, err := wrap.R.FormFile("file")
		if err != nil {
			wrap.MsgError(err.Error())
			return
		}
		defer file.Close()

		// Check file name
		if handler.Filename == "" {
			wrap.MsgError(`Inner system error`)
			return
		}

		// Read file to bytes
		fileBytes, err := ioutil.ReadAll(file)
		if err != nil {
			wrap.MsgError(err.Error())
			return
		}

		// Check if file is really an image
		if _, _, err := image.Decode(bytes.NewReader(fileBytes)); err != nil {
			wrap.MsgError(err.Error())
			return
		}

		// Create dirs
		if err := os.MkdirAll(wrap.DHtdocs+string(os.PathSeparator)+"products"+string(os.PathSeparator)+"images"+string(os.PathSeparator)+pf_id, os.ModePerm); err != nil {
			wrap.MsgError(err.Error())
			return
		}

		target_file_name := utils.Int64ToStr(time.Now().Unix()) + filepath.Ext(handler.Filename)
		target_file_full := wrap.DHtdocs + string(os.PathSeparator) + "products" + string(os.PathSeparator) + "images" + string(os.PathSeparator) + pf_id + string(os.PathSeparator) + target_file_name

		if err := wrap.DB.Transaction(func(tx *wrapper.Tx) error {
			// Block rows
			if _, err := tx.Exec("SELECT id FROM shop_products WHERE id = ? FOR UPDATE;", utils.StrToInt(pf_id)); err != nil {
				return err
			}

			// Insert row
			if _, err := tx.Exec(
				`INSERT INTO shop_product_images SET
					product_id = ?,
					filename = ?
				;`,
				utils.StrToInt(pf_id),
				target_file_name,
			); err != nil {
				return err
			}

			// Write file to disk
			if err := ioutil.WriteFile(target_file_full, fileBytes, 0664); err != nil {
				return err
			}
			return nil
		}); err != nil {
			wrap.MsgError(err.Error())
			return
		}

		wrap.Write(`$('#list-images').append('<div class="attached-img"><a href="/products/images/` + pf_id + `/` + target_file_name + `" target="_blank">` + target_file_name + `</a>, <a href="javascript:fave.ShopProductsDeleteImage(this, ` + pf_id + `, \'` + target_file_name + `\');">Delete</a></div>');`)
	})
}