Switch Repo to notatio/notatio
This commit is contained in:
commit
9da51a1e6b
26 changed files with 4117 additions and 0 deletions
326
list.go
Normal file
326
list.go
Normal file
|
|
@ -0,0 +1,326 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
func ListFiles(w http.ResponseWriter, r *http.Request) {
|
||||
userSession, err := validateSession(w, r)
|
||||
if err != nil {
|
||||
// Handle the error as needed
|
||||
handleInternalServerError(w, err)
|
||||
return
|
||||
}
|
||||
|
||||
userFolder := filepath.Join(fileUploadPath, userSession.username)
|
||||
|
||||
successParam := r.URL.Query().Get("success")
|
||||
successMessage := ""
|
||||
if successParam == "1" {
|
||||
successMessage = "Upload was successful!"
|
||||
}
|
||||
if successParam == "2" {
|
||||
successMessage = "File creation was successful!"
|
||||
}
|
||||
|
||||
// Open the user's directory
|
||||
userDir, err := os.Open(userFolder)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
// Directory does not exist, create it
|
||||
log.Println(userSession.username + "'s directory does not exist. Creating it!")
|
||||
createUserFolder(userSession.username)
|
||||
// Open the newly created user directory
|
||||
userDir, err = os.Open(userFolder)
|
||||
if err != nil {
|
||||
log.Printf("Error opening user directory: %v", err)
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
} else {
|
||||
// Other error occurred while opening the directory
|
||||
log.Printf("Error opening user directory: %v", err)
|
||||
// Handle the error as needed
|
||||
handleInternalServerError(w, err)
|
||||
return
|
||||
}
|
||||
}
|
||||
defer userDir.Close()
|
||||
|
||||
// Read the names of files in the directory
|
||||
fileNames, err := userDir.Readdirnames(-1)
|
||||
if err != nil {
|
||||
log.Printf("Error reading file names in directory: %v", err)
|
||||
// Handle the error as needed
|
||||
handleInternalServerError(w, err)
|
||||
return
|
||||
}
|
||||
|
||||
type File struct {
|
||||
Filename string
|
||||
CreationTime int64
|
||||
LastEdited int64
|
||||
}
|
||||
|
||||
var files []File
|
||||
for _, fileName := range fileNames {
|
||||
if strings.HasSuffix(fileName, ".html") || strings.HasSuffix(fileName, ".md") {
|
||||
file := File{
|
||||
Filename: fileName,
|
||||
}
|
||||
|
||||
userUUID, err := getUserUUID(userSession.username)
|
||||
if err != nil {
|
||||
log.Printf("Error getting user UUID: %v", err)
|
||||
// Handle the error as needed
|
||||
handleInternalServerError(w, err)
|
||||
return
|
||||
}
|
||||
|
||||
err = db.QueryRow("SELECT EXTRACT(epoch FROM creation_time)::bigint, EXTRACT(epoch FROM last_edited)::bigint FROM files WHERE user_id = $1 AND filename = $2",
|
||||
userUUID, fileName).Scan(&file.CreationTime, &file.LastEdited)
|
||||
|
||||
if err == sql.ErrNoRows {
|
||||
// No rows found, handle the case as needed
|
||||
log.Printf("No rows found for file: %s", fileName)
|
||||
currentTime := time.Now().Unix()
|
||||
|
||||
err = insertFileIntoDatabase(userSession.username, fileName, currentTime, currentTime, currentTime)
|
||||
if err != nil {
|
||||
log.Printf("Error inserting file into the database: %v", err)
|
||||
// Handle the error as needed
|
||||
handleInternalServerError(w, err)
|
||||
return
|
||||
}
|
||||
http.Redirect(w, r, "/home", http.StatusSeeOther)
|
||||
// Redirect or do something else
|
||||
return
|
||||
} else if err != nil {
|
||||
log.Printf("Error retrieving file timestamps from the database: %v", err)
|
||||
log.Printf("Asking %s to create a new file", userSession.username)
|
||||
http.Redirect(w, r, "/welcome", http.StatusSeeOther)
|
||||
return
|
||||
}
|
||||
|
||||
files = append(files, file)
|
||||
}
|
||||
}
|
||||
|
||||
name, err := getName(userSession.username)
|
||||
if err != nil {
|
||||
log.Printf("Error retrieving preferred name from the database: %v", err)
|
||||
}
|
||||
|
||||
data := struct {
|
||||
Username string
|
||||
Files []File
|
||||
SuccessMessage string
|
||||
}{
|
||||
Username: name,
|
||||
Files: files,
|
||||
SuccessMessage: successMessage,
|
||||
}
|
||||
|
||||
renderTemplate(w, "list.html", data)
|
||||
}
|
||||
|
||||
func UploadFile(w http.ResponseWriter, r *http.Request) {
|
||||
userSession, err := validateSession(w, r)
|
||||
if err != nil {
|
||||
// Handle the error as needed
|
||||
return
|
||||
}
|
||||
|
||||
if r.Method == http.MethodPost {
|
||||
reader, err := r.MultipartReader()
|
||||
if err != nil {
|
||||
log.Printf("Error creating multipart reader: %v", err)
|
||||
http.Error(w, "Bad Request", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
for {
|
||||
part, err := reader.NextPart()
|
||||
if err == io.EOF {
|
||||
break
|
||||
}
|
||||
if err != nil {
|
||||
log.Printf("Error reading part: %v", err)
|
||||
http.Error(w, "Bad Request", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
// Check the file extension to allow only .md or .html files
|
||||
fileExt := filepath.Ext(part.FileName())
|
||||
if fileExt != ".md" && fileExt != ".html" {
|
||||
log.Printf("Invalid file format: %s", part.FileName())
|
||||
http.Error(w, "Invalid File Format. Only .md and .html files are allowed.", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
userFolder := filepath.Join(fileUploadPath, userSession.username)
|
||||
|
||||
// Generate a unique filename by appending sequential numbers if needed
|
||||
filename := part.FileName()
|
||||
count := 1
|
||||
for {
|
||||
filePath := filepath.Join(userFolder, filename)
|
||||
if _, err := os.Stat(filePath); os.IsNotExist(err) {
|
||||
break
|
||||
}
|
||||
filename = fmt.Sprintf("%s_%d%s", strings.TrimSuffix(part.FileName(), filepath.Ext(part.FileName())), count, filepath.Ext(part.FileName()))
|
||||
count++
|
||||
}
|
||||
|
||||
// Save the file to disk in the user's folder
|
||||
filePath := filepath.Join(userFolder, filename)
|
||||
destFile, err := os.Create(filePath)
|
||||
if err != nil {
|
||||
log.Printf("Error creating destination file: %v", err)
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer destFile.Close()
|
||||
|
||||
_, err = io.Copy(destFile, part)
|
||||
if err != nil {
|
||||
log.Printf("Error saving file to disk: %v", err)
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Insert the file record into the database with creation time
|
||||
currentTime := time.Now().Unix()
|
||||
err = insertFileIntoDatabase(userSession.username, filename, currentTime, currentTime, currentTime)
|
||||
if err != nil {
|
||||
log.Printf("Error inserting file into database: %v", err)
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
http.Redirect(w, r, "/home?success=1", http.StatusSeeOther)
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func createNewFile(w http.ResponseWriter, r *http.Request) {
|
||||
userSession, err := validateSession(w, r)
|
||||
if err != nil {
|
||||
// Handle the error as needed
|
||||
return
|
||||
}
|
||||
|
||||
// Parse the form data to obtain the filename
|
||||
if err := r.ParseForm(); err != nil {
|
||||
http.Error(w, "Bad Request", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
filename := r.FormValue("newFileName")
|
||||
// Ensure the file has the ".md" extension
|
||||
if !strings.HasSuffix(filename, ".md") {
|
||||
filename += ".md"
|
||||
}
|
||||
|
||||
// Construct the file path
|
||||
userFolder := filepath.Join(fileUploadPath, userSession.username)
|
||||
filePath := filepath.Join(userFolder, filename)
|
||||
|
||||
// Check if the file already exists
|
||||
if _, err := os.Stat(filePath); !os.IsNotExist(err) {
|
||||
// File with the same name already exists
|
||||
http.Error(w, "File already exists", http.StatusConflict)
|
||||
return
|
||||
}
|
||||
|
||||
// Create and open the file for writing
|
||||
file, err := os.Create(filePath)
|
||||
if err != nil {
|
||||
log.Printf("Error creating file: %v", err)
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
// Insert the file record into the database with creation time
|
||||
currentTime := time.Now().Unix()
|
||||
|
||||
err = insertFileIntoDatabase(userSession.username, filename, currentTime, currentTime, currentTime)
|
||||
if err != nil {
|
||||
log.Printf("Error inserting file into the database: %v", err)
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Redirect to editing the newly created file
|
||||
http.Redirect(w, r, "/edit?filename="+filename, http.StatusSeeOther)
|
||||
}
|
||||
|
||||
func DeleteFiles(w http.ResponseWriter, r *http.Request) {
|
||||
userSession, err := validateSession(w, r)
|
||||
if err != nil {
|
||||
// Handle the error as needed
|
||||
return
|
||||
}
|
||||
|
||||
if r.Method == http.MethodPost {
|
||||
// Parse the JSON request body to get the list of files to delete
|
||||
var request struct {
|
||||
Files []string `json:"files"`
|
||||
}
|
||||
// Print the list of files to delete
|
||||
for _, filename := range request.Files {
|
||||
fmt.Println("File to delete:", filename)
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&request); err != nil {
|
||||
http.Error(w, "Bad Request", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
// Delete each selected file from the database and the filesystem
|
||||
for _, filename := range request.Files {
|
||||
userFolder := filepath.Join(fileUploadPath, userSession.username)
|
||||
filePath := filepath.Join(userFolder, filename)
|
||||
|
||||
// Delete the file from the database
|
||||
if err := deleteFileFromDatabase(userSession.username, filename); err != nil {
|
||||
// Handle the error as needed
|
||||
log.Printf("Error deleting file record: %v", err)
|
||||
}
|
||||
|
||||
// Delete the file from the filesystem
|
||||
if err := os.Remove(filePath); err != nil {
|
||||
// Handle the error as needed
|
||||
log.Printf("Error deleting file from filesystem: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Send a success response
|
||||
w.WriteHeader(http.StatusOK)
|
||||
}
|
||||
}
|
||||
|
||||
func makeFolder(folderPath string) error {
|
||||
err := os.MkdirAll(folderPath, os.ModePerm)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func createUserFolder(username string) {
|
||||
userFolder := filepath.Join(fileUploadPath, username)
|
||||
err := makeFolder(userFolder)
|
||||
if err != nil {
|
||||
log.Printf("Error creating user folder: %v", err)
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue