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(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(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(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(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(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, make an issue at https://git.privacyquest.net", 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) } }