Gitea
3 years ago
15 changed files with 302 additions and 70 deletions
@ -1,22 +0,0 @@ |
|||||||
/* |
|
||||||
gochat - A dead simple real time webchat. |
|
||||||
Copyright (C) 2022 Kasyanov Nikolay Alexeyevich (Unbewohnte) |
|
||||||
This file is a part of gochat |
|
||||||
This program is free software: you can redistribute it and/or modify |
|
||||||
it under the terms of the GNU General Public License as published by |
|
||||||
the Free Software Foundation, either version 3 of the License, or |
|
||||||
(at your option) any later version. |
|
||||||
This program is distributed in the hope that it will be useful, |
|
||||||
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
||||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
||||||
GNU General Public License for more details. |
|
||||||
You should have received a copy of the GNU General Public License |
|
||||||
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
||||||
*/ |
|
||||||
|
|
||||||
|
|
||||||
function to_hex_string(byteArr) { |
|
||||||
return Array.from(byteArr, function(byte) { |
|
||||||
return ('0' + (byte & 0xFF).toString(16).slice(-2)); |
|
||||||
}).join('') |
|
||||||
} |
|
||||||
@ -0,0 +1,7 @@ |
|||||||
|
package api |
||||||
|
|
||||||
|
const AttachmentFormPostKey string = "attachment" |
||||||
|
|
||||||
|
type PartialAttachmentURL struct { |
||||||
|
URL string `json:"url"` |
||||||
|
} |
||||||
@ -0,0 +1,115 @@ |
|||||||
|
package server |
||||||
|
|
||||||
|
import ( |
||||||
|
"encoding/json" |
||||||
|
"fmt" |
||||||
|
"io" |
||||||
|
"net/http" |
||||||
|
"os" |
||||||
|
"path/filepath" |
||||||
|
"time" |
||||||
|
|
||||||
|
"unbewohnte.xyz/Unbewohnte/gochat/api" |
||||||
|
"unbewohnte.xyz/Unbewohnte/gochat/log" |
||||||
|
) |
||||||
|
|
||||||
|
// Handle incoming attachments
|
||||||
|
func (s *Server) handleAttachments(w http.ResponseWriter, req *http.Request) { |
||||||
|
userAuthHeader := api.GetUserAuthHeaderData(req) |
||||||
|
if !s.db.DoesUserExist(userAuthHeader.Name) { |
||||||
|
http.Error(w, "Not authorized", http.StatusUnauthorized) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
userDB, _ := s.db.GetUser(userAuthHeader.Name) |
||||||
|
if userDB.SecretHash != userAuthHeader.SecretHash { |
||||||
|
http.Error(w, "Not authorized", http.StatusUnauthorized) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// accept incoming file
|
||||||
|
file, header, err := req.FormFile(api.AttachmentFormPostKey) |
||||||
|
if err != nil { |
||||||
|
log.Error("could not get attached file: %s", err) |
||||||
|
http.Error(w, "Error getting attached file", http.StatusInternalServerError) |
||||||
|
return |
||||||
|
} |
||||||
|
defer file.Close() |
||||||
|
|
||||||
|
if uint64(header.Size) > api.MaxAttachmentSize { |
||||||
|
http.Error(w, "Too big file", http.StatusBadRequest) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
localFilename := fmt.Sprintf("%s_%d_%s", userDB.Name, header.Size, header.Filename) |
||||||
|
localFile, err := os.Create(filepath.Join(s.workingDir, attachmentsDirName, localFilename)) |
||||||
|
if err != nil { |
||||||
|
log.Error("could not create local attachment file: %s", err) |
||||||
|
http.Error(w, "Could not create file", http.StatusInternalServerError) |
||||||
|
return |
||||||
|
} |
||||||
|
defer localFile.Close() |
||||||
|
|
||||||
|
_, err = io.Copy(localFile, file) |
||||||
|
if err != nil { |
||||||
|
log.Error("could not copy attachment file contents: %s", err) |
||||||
|
http.Error(w, "Could not copy file contents", http.StatusInternalServerError) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
// send partial URL pointing to the file
|
||||||
|
url := api.PartialAttachmentURL{ |
||||||
|
URL: fmt.Sprintf("%s/%s", attachmentsDirName, localFilename), |
||||||
|
} |
||||||
|
urlJsonBytes, err := json.Marshal(&url) |
||||||
|
if err != nil { |
||||||
|
log.Error("could not marshal partial attachment URL for \"%s\": %s", localFilename, err) |
||||||
|
http.Error(w, "Error constructing a request", http.StatusInternalServerError) |
||||||
|
return |
||||||
|
} |
||||||
|
|
||||||
|
w.Write(urlJsonBytes) |
||||||
|
} |
||||||
|
|
||||||
|
// Remove the oldest attachments when the memory limit is exceeded
|
||||||
|
func manageAttachmentsStorage(attachmentsDirPath string, sizeLimit uint64, checkDelay time.Duration) { |
||||||
|
for { |
||||||
|
dirEntries, err := os.ReadDir(attachmentsDirPath) |
||||||
|
if err != nil { |
||||||
|
log.Error("error reading attachments directory: %s", err) |
||||||
|
} |
||||||
|
|
||||||
|
var dirSize uint64 = 0 |
||||||
|
var oldestAttachmentModTime time.Time = time.Now() |
||||||
|
var oldestAttachmentPath string = "" |
||||||
|
var oldestAttachmentSize uint64 = 0 |
||||||
|
if dirEntries != nil { |
||||||
|
for _, entry := range dirEntries { |
||||||
|
entryInfo, err := entry.Info() |
||||||
|
if err != nil { |
||||||
|
continue |
||||||
|
} |
||||||
|
|
||||||
|
entrySize := entryInfo.Size() |
||||||
|
dirSize += uint64(entrySize) |
||||||
|
|
||||||
|
entryModTime := entryInfo.ModTime() |
||||||
|
if entryModTime.Before(oldestAttachmentModTime) { |
||||||
|
oldestAttachmentModTime = entryModTime |
||||||
|
oldestAttachmentPath = filepath.Join(attachmentsDirPath, entry.Name()) |
||||||
|
oldestAttachmentSize = uint64(entrySize) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
if dirSize > sizeLimit { |
||||||
|
// A cleanup !
|
||||||
|
os.Remove(oldestAttachmentPath) |
||||||
|
log.Info( |
||||||
|
"removed %s during attachments storage management. Cleared %d bytes", |
||||||
|
oldestAttachmentPath, oldestAttachmentSize) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
time.Sleep(checkDelay) |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue