goploy/controller/PackageController.go

package controller

import (
	"database/sql"
	"io/ioutil"
	"strconv"

	"github.com/zhenorzz/goploy/core"
	"github.com/zhenorzz/goploy/model"
)

// Package struct
type Package Controller

// GetList list
func (Package) GetList(gp *core.Goploy) *core.Response {
	type RespData struct {
		Packages model.Packages `json:"list"`
	}
	pagination, err := model.PaginationFrom(gp.URLQuery)
	if err != nil {
		return &core.Response{Code: core.Error, Message: err.Error()}
	}
	packageList, err := model.Package{}.GetList(pagination)
	if err != nil {
		return &core.Response{Code: core.Error, Message: err.Error()}
	}
	return &core.Response{Data: RespData{Packages: packageList}}
}

// GetTotal total
func (Package) GetTotal(gp *core.Goploy) *core.Response {
	type RespData struct {
		Total int64 `json:"total"`
	}
	total, err := model.Package{}.GetTotal()

	if err != nil {
		return &core.Response{Code: core.Error, Message: err.Error()}
	}
	return &core.Response{Data: RespData{Total: total}}
}

// GetOption list
func (Package) GetOption(gp *core.Goploy) *core.Response {
	type RespData struct {
		Packages model.Packages `json:"list"`
	}

	packageList, err := model.Package{}.GetAll()
	if err != nil {
		return &core.Response{Code: core.Error, Message: err.Error()}
	}
	return &core.Response{Data: RespData{Packages: packageList}}
}

// Upload file
func (Package) Upload(gp *core.Goploy) *core.Response {
	type RespData struct {
		Filename string `json:"filename"`
	}
	file, handler, err := gp.Request.FormFile("file")
	if err != nil {
		return &core.Response{Code: core.Error, Message: err.Error()}
	}
	defer file.Close()
	_, err = model.Package{Name: handler.Filename}.GetDataByName()
	if err != sql.ErrNoRows {
		return &core.Response{Code: core.Deny, Message: "The same file already exists"}
	}
	// read all of the contents of our uploaded file into a
	// byte array
	fileBytes, err := ioutil.ReadAll(file)
	if err != nil {
		return &core.Response{Code: core.Error, Message: err.Error()}
	}
	filePath := core.PackagePath + handler.Filename
	if err := ioutil.WriteFile(filePath, fileBytes, 0755); err != nil {
		return &core.Response{Code: core.Error, Message: err.Error()}
	}

	packageIDStr := gp.URLQuery.Get("packageId")

	if packageIDStr == "" {
		_, err = model.Package{
			Name: handler.Filename,
			Size: handler.Size,
		}.AddRow()
	} else {
		packageID, err := strconv.ParseInt(packageIDStr, 10, 64)
		if err != nil {
			return &core.Response{Code: core.Error, Message: err.Error()}
		}

		err = model.Package{
			ID:   packageID,
			Name: handler.Filename,
			Size: handler.Size,
		}.EditRow()
	}

	if err != nil {
		return &core.Response{Code: core.Error, Message: err.Error()}
	}

	return &core.Response{Data: RespData{Filename: handler.Filename}}
}