131 lines
3.8 KiB
Go
131 lines
3.8 KiB
Go
package handlers
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"threadr/models"
|
|
|
|
"github.com/gorilla/sessions"
|
|
)
|
|
|
|
func ProfileEditHandler(app *App) http.HandlerFunc {
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
session := r.Context().Value("session").(*sessions.Session)
|
|
userID, ok := session.Values["user_id"].(int)
|
|
if !ok {
|
|
http.Redirect(w, r, app.Config.ThreadrDir+"/login/", http.StatusFound)
|
|
return
|
|
}
|
|
|
|
if r.Method == http.MethodPost {
|
|
// Handle file upload
|
|
file, handler, err := r.FormFile("pfp")
|
|
if err == nil {
|
|
defer file.Close()
|
|
|
|
// Create a hash of the file
|
|
h := sha256.New()
|
|
if _, err := io.Copy(h, file); err != nil {
|
|
log.Printf("Error hashing file: %v", err)
|
|
http.Error(w, "Failed to process file", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
fileHash := fmt.Sprintf("%x", h.Sum(nil))
|
|
|
|
// Create file record in the database
|
|
fileRecord := models.File{
|
|
OriginalName: handler.Filename,
|
|
Hash: fileHash,
|
|
HashAlgorithm: "sha256",
|
|
}
|
|
fileID, err := models.CreateFile(app.DB, fileRecord)
|
|
if err != nil {
|
|
log.Printf("Error creating file record: %v", err)
|
|
http.Error(w, "Failed to save file information", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
// Save the file to disk
|
|
fileExt := filepath.Ext(handler.Filename)
|
|
newFileName := fmt.Sprintf("%d%s", fileID, fileExt)
|
|
filePath := filepath.Join(app.Config.FileStorageDir, newFileName)
|
|
|
|
// Reset file pointer
|
|
file.Seek(0, 0)
|
|
|
|
dst, err := os.Create(filePath)
|
|
if err != nil {
|
|
log.Printf("Error creating file on disk: %v", err)
|
|
http.Error(w, "Failed to save file", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
defer dst.Close()
|
|
|
|
if _, err := io.Copy(dst, file); err != nil {
|
|
log.Printf("Error saving file to disk: %v", err)
|
|
http.Error(w, "Failed to save file", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
// Update user's pfp_file_id
|
|
err = models.UpdateUserPfp(app.DB, userID, fileID)
|
|
if err != nil {
|
|
log.Printf("Error updating user pfp: %v", err)
|
|
http.Error(w, "Failed to update profile", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
|
|
// Update other profile fields
|
|
displayName := r.FormValue("display_name")
|
|
bio := r.FormValue("bio")
|
|
err = models.UpdateUserProfile(app.DB, userID, displayName, bio)
|
|
if err != nil {
|
|
log.Printf("Error updating profile: %v", err)
|
|
http.Error(w, "Failed to update profile", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
http.Redirect(w, r, app.Config.ThreadrDir+"/profile/", http.StatusFound)
|
|
return
|
|
}
|
|
|
|
user, err := models.GetUserByID(app.DB, userID)
|
|
if err != nil {
|
|
log.Printf("Error fetching user: %v", err)
|
|
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
if user == nil {
|
|
http.Error(w, "User not found", http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
data := struct {
|
|
PageData
|
|
User models.User
|
|
}{
|
|
PageData: PageData{
|
|
Title: "ThreadR - Edit Profile",
|
|
Navbar: "profile",
|
|
LoggedIn: true,
|
|
ShowCookieBanner: false,
|
|
BasePath: app.Config.ThreadrDir,
|
|
StaticPath: app.Config.ThreadrDir + "/static",
|
|
CurrentURL: r.URL.Path,
|
|
},
|
|
User: *user,
|
|
}
|
|
if err := app.Tmpl.ExecuteTemplate(w, "profile_edit", data); err != nil {
|
|
log.Printf("Error executing template in ProfileEditHandler: %v", err)
|
|
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
}
|
|
}
|