admin/go-socket
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

idk

Browse Source
This commit is contained in:
gitGnome
2026-04-16 14:23:14 +02:00
parent 6d0eaa92e9
commit f0fdaedd8c
4 changed files with 28 additions and 154 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/http/file.go
+18 -57
View File
@@ -1,21 +1,14 @@
package http
import (
"crypto/sha256"
"go-socket/packages/convertions"
"go-socket/packages/globals"
"go-socket/packages/minio"
"go-socket/packages/postgresql"
"io"
"net/http"
"github.com/google/uuid"
)
func handleExistingUpload() {
}
func HandleFileUpload(response http.ResponseWriter, request *http.Request) {
if !postValidCheckWithResponseOnFail(&response, request, true) {
return
@@ -28,6 +21,13 @@ func HandleFileUpload(response http.ResponseWriter, request *http.Request) {
return
}
request.Body = http.MaxBytesReader(response, request.Body, int64(globals.MaxPostWithFileBytes))
if err = request.ParseMultipartForm(int64(globals.MaxPostBytes)); err != nil {
http.Error(response, "invalid multipart form", http.StatusBadRequest)
return
}
connectionId, err := convertions.ConvertStringUuid(request.FormValue("connectionid"))
if err != nil {
http.Error(response, "invalid connectionid", http.StatusBadRequest)
@@ -39,64 +39,25 @@ func HandleFileUpload(response http.ResponseWriter, request *http.Request) {
return
}
request.Body = http.MaxBytesReader(response, request.Body, int64(globals.MaxPostWithFileBytes))
multipartReader, err := request.MultipartReader()
file, header, err := request.FormFile("file")
if err != nil {
http.Error(response, "expected multipart form", http.StatusBadRequest)
http.Error(response, "missing file", http.StatusBadRequest)
return
}
defer file.Close()
uploaded := false
for {
part, err := multipartReader.NextPart()
if err == io.EOF {
break
}
if err != nil {
http.Error(response, "bad request", http.StatusBadRequest)
return
}
contentType := header.Header.Get("Content-Type")
if part.FileName() == "" {
part.Close()
continue
}
key :=
if len(part.FileName()) != sha256.Size*2 {
part.Close()
http.Error(response, "filename must be hex-encoded sha256 of whole file", http.StatusBadRequest)
return
}
contentType := part.Header.Get("Content-Type")
key := minio.GetKey(part.FileName(), contentType)
if minio.DoesExist(ctx, key) {
part.Close()
uploaded = true
handleExistingUpload()
continue
}
id := uuid.New()
err = minio.Upload(ctx, key, part, -1, contentType, id)
postgresql.FileMetadataSave()
part.Close()
if err != nil {
http.Error(response, "upload failed", http.StatusInternalServerError)
return
}
uploaded = true
}
if !uploaded {
http.Error(response, "no file in request", http.StatusBadRequest)
if err = minio.Upload(ctx, fileId.String(), file, header.Size, contentType, map[string]string{
"orginalName": header.Filename,
"uploaderId": user.Id.String(),
}); err != nil {
http.Error(response, "upload failed", http.StatusInternalServerError)
return
}
response.WriteHeader(http.StatusAccepted)
response.Write([]byte(fileId.String()))
}
packages/minio/minio.go
+10 -11
View File
@@ -8,7 +8,6 @@ import (
"go-socket/packages/globals"
"github.com/google/uuid"
"github.com/minio/minio-go/v7"
"github.com/minio/minio-go/v7/pkg/credentials"
)
@@ -65,16 +64,16 @@ func Init() {
}
func Upload(ctx context.Context, key string, body io.Reader, size int64, contentType string, id uuid.UUID) error {
_, err := minClient.PutObject(ctx, globals.FileStorageBucketName, key, body, size,
minio.PutObjectOptions{
ContentType: contentType,
PartSize: globals.FileProcessingPartSize,
NumThreads: globals.FileProcessingThreads,
UserMetadata: map[string]string{
"id": id.String(),
},
})
func Upload(ctx context.Context, key string, body io.Reader, size int64, contentType string, metadata map[string]string) error {
opt := minio.PutObjectOptions{
ContentType: contentType,
PartSize: globals.FileProcessingPartSize,
NumThreads: globals.FileProcessingThreads,
UserMetadata: metadata,
}
opt.SetMatchETagExcept("*")
_, err := minClient.PutObject(ctx, globals.FileStorageBucketName, key, body, size, opt)
return err
}
packages/postgresql/postgresql.go
-77
View File
@@ -66,29 +66,6 @@ func Init(ctx context.Context) {
if err != nil {
panic(err)
}
_, err = dbConn.Exec(ctx, `
CREATE TABLE IF NOT EXISTS files_metadata (
id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
uploader_id UUID NOT NULL REFERENCES users(id),
name TEXT,
hash TEXT NOT NULL,
created_at TIMESTAMP NOT NULL DEFAULT NOW()
)
`)
if err != nil {
panic(err)
}
_, err = dbConn.Exec(ctx, `
CREATE TABLE IF NOT EXISTS file_connections (
file_id UUID NOT NULL REFERENCES files_metadata(id) ON DELETE CASCADE,
connection_id UUID NOT NULL REFERENCES user_connections(id) ON DELETE CASCADE
)
`)
if err != nil {
panic(err)
}
}
func UserSave(ctx context.Context, user *types.User) error {
@@ -243,57 +220,3 @@ func ConnectionGetMessagesBefore(ctx context.Context, before time.Time, connecti
}
return messages, rows.Err()
}
func FileMetadataSave(ctx context.Context, metadata *types.FileMetadata) error {
_, err := dbConn.Exec(ctx, `
INSERT INTO files_metadata (id, uploader_id, name, hash, created_at) VALUES ($1, $2, $3, $4, $5)
`, metadata.Id, metadata.UploaderId, metadata.Name, metadata.Hash, metadata.CreatedAt)
return err
}
func FileMetadataGet(ctx context.Context, metadata *types.FileMetadata) error {
return dbConn.QueryRow(ctx, `
SELECT uploader_id, name, hash, created_at FROM files_metadata WHERE id = $1
`, metadata.Id).Scan(&metadata.UploaderId, &metadata.Name, &metadata.Hash, &metadata.CreatedAt)
}
func FileMetadataDelete(ctx context.Context, metadataId *uuid.UUID) error {
_, err := dbConn.Exec(ctx, `
DELETE FROM files_metadata WHERE id = $1
`, metadataId)
return err
}
func FileConnectionsAdd(ctx context.Context, metadataId *uuid.UUID, connectionId *uuid.UUID) error {
_, err := dbConn.Exec(ctx, `
INSERT INTO file_connections (file_id, connection_id) VALUES ($1, $2)
`, metadataId, connectionId)
return err
}
func FileConnectionsGet(ctx context.Context, metadata *types.FileMetadata) error {
rows, err := dbConn.Query(ctx, `
SELECT connection_id FROM file_connections WHERE file_id = $1
`, metadata.Id)
if err != nil {
return err
}
defer rows.Close()
for rows.Next() {
connectionId := uuid.UUID{}
if err = rows.Scan(&connectionId); err != nil {
return err
}
metadata.Connections = append(metadata.Connections, connectionId)
}
return rows.Err()
}
func FileMetadataHashExists(ctx context.Context, hash *types.Sha256Hash) (bool, error) {
var exists bool
err := dbConn.QueryRow(ctx, `
SELECT EXISTS (SELECT FROM files_metadata WHERE hash = $1)
`, hash).Scan(&exists)
return exists, err
}
packages/types/types.go
-9
View File
@@ -104,12 +104,3 @@ type WsAuthMessage struct {
Success bool `json:"success"`
Error string `json:"error"`
}
type FileMetadata struct {
CreatedAt time.Time `json:"createdAt"`
Connections []uuid.UUID `json:"connections"`
Name string `json:"name"`
Hash Sha256Hash `json:"hash"`
Id uuid.UUID `json:"id"`
UploaderId uuid.UUID `json:"uploaderId"`
}