ITL/Uploader
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add upload folder

Browse Source
This commit is contained in:
AustrianToast 2024-06-03 19:44:53 +02:00
parent 7c782522f5
commit f63932ddaf
No known key found for this signature in database
GPG Key ID: 5CD422268E489EB4
1 changed files with 11 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
main.go
View File

@ -16,6 +16,8 @@ import (
var dbpool *pgxpool.Pool
var uploadFolder string
func main() {
var err error
dbpool, err = pgxpool.New(context.Background(), "postgresql://postgres:postgres@172.19.0.2:5432/postgres")
@ -48,6 +50,12 @@ func main() {
}
}
currentDir, err := os.Getwd()
if err != nil {
log.Fatalf("main: %v\n", err)
}
uploadFolder = fmt.Sprintf("%s/videos", currentDir)
router := gin.Default()
router.SetTrustedProxies(nil)
@ -70,12 +78,6 @@ func initUpload(c *gin.Context) {
return
}
currentDir, err := os.Getwd()
if err != nil {
c.JSON(http.StatusInternalServerError, "")
log.Fatalf("initUpload: %v\n", err)
}
rows, _ := dbpool.Query(context.Background(), "select filepath from videos")
for rows.Next() {
var filepath string
@ -85,7 +87,7 @@ func initUpload(c *gin.Context) {
log.Fatalf("initUpload: %v\n", err)
}
fmt.Println(filepath)
if filepath == fmt.Sprintf("%s/%s", currentDir, fileName) {
if filepath == fmt.Sprintf("%s/%s", uploadFolder, fileName) {
c.JSON(http.StatusForbidden, "File already exists")
return
}
@ -106,7 +108,7 @@ func ReceiveChunk(c *gin.Context) {
return
}
f, err := os.OpenFile(c.GetHeader("file-name"), os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
f, err := os.OpenFile(fmt.Sprintf("%s/%s", uploadFolder, c.GetHeader("file-name")), os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
c.JSON(http.StatusInternalServerError, "")
log.Fatal(err)
@ -124,19 +126,13 @@ func ReceiveChunk(c *gin.Context) {
}
func finishUpload(c *gin.Context) {
currentDir, err := os.Getwd()
if err != nil {
c.JSON(http.StatusInternalServerError, "")
log.Fatalf("finishUpload: %v\n", err)
}
fileName, err := io.ReadAll(c.Request.Body)
if err != nil {
c.JSON(http.StatusBadRequest, "Couldn't read html request body")
return
}
_, err = dbpool.Exec(context.Background(), "insert into videos(filepath) values($1)", fmt.Sprintf("%s/%s", currentDir, fileName))
_, err = dbpool.Exec(context.Background(), "insert into videos(filepath) values($1)", fmt.Sprintf("%s/%s", uploadFolder, fileName))
if err != nil {
c.JSON(http.StatusInternalServerError, "")
log.Fatalf("finishUpload: %v\n", err)