File Uploading feature by tfasano1
This commit is contained in:
parent
9b231a73d6
commit
d0da1a9114
@ -25,6 +25,11 @@ type Configuration struct {
|
||||
Template struct {
|
||||
BaseName string `json:"BaseTemplateName"`
|
||||
}
|
||||
|
||||
Upload struct {
|
||||
BaseName string `json:"UploadDirectoryName"`
|
||||
MaxSize int64 `json:"MaxUploadSize"`
|
||||
}
|
||||
}
|
||||
|
||||
// LoadConfig loads and returns a configuration struct
|
||||
|
@ -61,6 +61,13 @@ func (getController *GetController) ShowLogin(w http.ResponseWriter, r *http.Req
|
||||
templating.RenderTemplate(getController.App, w, "templates/pages/login.html", data)
|
||||
}
|
||||
|
||||
func (getController *GetController) ShowFile(w http.ResponseWriter, r *http.Request) {
|
||||
// GET /uploads?name=file.jpg
|
||||
// will serve file.jpg
|
||||
name := r.URL.Query().Get("name")
|
||||
http.ServeFile(w, r, getController.App.Config.Upload.BaseName+name)
|
||||
}
|
||||
|
||||
func (getController *GetController) Logout(w http.ResponseWriter, r *http.Request) {
|
||||
models.LogoutUser(getController.App, w, r)
|
||||
http.Redirect(w, r, "/", http.StatusFound)
|
||||
|
@ -4,8 +4,10 @@ import (
|
||||
"GoWeb/app"
|
||||
"GoWeb/models"
|
||||
"GoWeb/security"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"time"
|
||||
)
|
||||
|
||||
@ -69,3 +71,46 @@ func (postController *PostController) Register(w http.ResponseWriter, r *http.Re
|
||||
|
||||
http.Redirect(w, r, "/login", http.StatusFound)
|
||||
}
|
||||
|
||||
func (postController *PostController) FileUpload(w http.ResponseWriter, r *http.Request) {
|
||||
|
||||
max := postController.App.Config.Upload.MaxSize
|
||||
r.ParseMultipartForm(max)
|
||||
|
||||
// FormFile returns the first file for the given key `file`
|
||||
// it also returns the FileHeader so we can get the Filename,
|
||||
// the Header and the size of the file
|
||||
file, handler, err := r.FormFile("file")
|
||||
if err != nil {
|
||||
log.Println("Error Retrieving the File")
|
||||
log.Println(err)
|
||||
return
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
if handler.Size > max {
|
||||
log.Println("User tried uploading a file which is too large.")
|
||||
http.Redirect(w, r, "/", http.StatusRequestHeaderFieldsTooLarge)
|
||||
return
|
||||
}
|
||||
|
||||
// Create a temporary file within upload directory
|
||||
tempFile, err := os.Create(postController.App.Config.Upload.BaseName + handler.Filename)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
http.Redirect(w, r, "/", http.StatusNotAcceptable)
|
||||
}
|
||||
defer tempFile.Close()
|
||||
|
||||
// read all of the contents of our uploaded file into a
|
||||
// byte array
|
||||
fileBytes, err := ioutil.ReadAll(file)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
// write this byte array to our temporary file
|
||||
tempFile.Write(fileBytes)
|
||||
// return that we have successfully uploaded our file!
|
||||
|
||||
http.Redirect(w, r, "/", http.StatusFound)
|
||||
}
|
||||
|
@ -13,5 +13,9 @@
|
||||
},
|
||||
"Template": {
|
||||
"BaseTemplateName": "templates/base.html"
|
||||
},
|
||||
"Upload": {
|
||||
"UploadDirectoryName": "goweb-uploads/",
|
||||
"MaxUploadSize": 10485760
|
||||
}
|
||||
}
|
||||
}
|
||||
|
9
main.go
9
main.go
@ -8,6 +8,7 @@ import (
|
||||
"GoWeb/routes"
|
||||
"context"
|
||||
"embed"
|
||||
"errors"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
@ -43,6 +44,14 @@ func main() {
|
||||
file, err := os.OpenFile("logs/"+time.Now().Format("2006-01-02")+".log", os.O_CREATE|os.O_WRONLY|os.O_APPEND, 0666)
|
||||
log.SetOutput(file)
|
||||
|
||||
// Create upload directory if it doesn't exist
|
||||
uploadPath := appLoaded.Config.Upload.BaseName
|
||||
if _, err := os.Stat(uploadPath); errors.Is(err, os.ErrNotExist) {
|
||||
if err := os.MkdirAll(uploadPath, os.ModePerm); err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
// Connect to database and run migrations
|
||||
appLoaded.Db = database.ConnectDB(&appLoaded)
|
||||
if appLoaded.Config.Db.AutoMigrate {
|
||||
|
@ -30,4 +30,7 @@ func GetRoutes(app *app.App) {
|
||||
http.HandleFunc("/login", getController.ShowLogin)
|
||||
http.HandleFunc("/register", getController.ShowRegister)
|
||||
http.HandleFunc("/logout", getController.Logout)
|
||||
|
||||
// Files
|
||||
http.HandleFunc("/uploads", getController.ShowFile)
|
||||
}
|
||||
|
@ -16,4 +16,5 @@ func PostRoutes(app *app.App) {
|
||||
// User authentication
|
||||
http.HandleFunc("/register-handle", postController.Register)
|
||||
http.HandleFunc("/login-handle", postController.Login)
|
||||
http.HandleFunc("/upload-handle", postController.FileUpload)
|
||||
}
|
||||
|
@ -1,5 +1,25 @@
|
||||
{{ define "pageTitle" }}Home{{ end }}
|
||||
|
||||
{{ define "file-upload" }}
|
||||
<form
|
||||
enctype="multipart/form-data"
|
||||
action="/upload-handle"
|
||||
method="post"
|
||||
>
|
||||
<input type="file" accept="*/*" name="file" />
|
||||
<input type="submit" value="upload" />
|
||||
</form>
|
||||
{{ end }}
|
||||
|
||||
{{ define "content" }}
|
||||
{{ .Test }}
|
||||
{{ end }}
|
||||
|
||||
<!-- Uncomment below to demo file upload system -->
|
||||
|
||||
<!-- {{ template "file-upload" . }} -->
|
||||
<!-- <p>Upload an image called test.jpg to test the file upload system</p> -->
|
||||
<!-- <img src="/uploads?name=test.jpg" alt=""> -->
|
||||
{{ end }}
|
||||
|
||||
{{ define "content" }}
|
||||
{{ end }}
|
||||
|
Loading…
Reference in New Issue
Block a user