FreePad/lib/objects/objects_post.go

package objects

import (
	"encoding/json"
	"errors"
	"fmt"
	"os"
	"path"
	"path/filepath"
	"sync"
	"time"

	"github.com/JustKato/FreePad/lib/helper"
)

// Initialize the views cache
var ViewsCache map[string]uint32 = make(map[string]uint32)

// Mutex lock for the ViewsCache
var viewersLock sync.Mutex

type Post struct {
	Name         string `json:"name"`
	LastModified string `json:"last_modified"`
	Content      string `json:"content"`
	Views        uint32 `json:"views"`
}

// Get the path to the views JSON
func getViewsFilePath() (string, error) {
	// Get the path to the storage then append the const name for the storage file
	filePath := path.Join(getStorageDirectory(), "views_storage.json")

	// Check if the file exists
	if _, err := os.Stat(filePath); errors.Is(err, os.ErrNotExist) {
		// Create the file
		err := os.WriteFile(filePath, []byte(""), 0640)
		if err != nil {
			return ``, err
		}
	}

	// Return the file path
	return filePath, nil
}

// Load the views cache from file
func LoadViewsCache() error {
	// Get the views file path
	viewsFilePath, err := getViewsFilePath()
	if err != nil {
		// This is now a problem for the upstairs
		return err
	}

	// Read the contents of the file as a map
	f, err := os.ReadFile(viewsFilePath)
	if err != nil {
		// This is now a problem for the upstairs
		return err
	}

	// Check if the contents are valid
	if len(f) <= 0 && len(string(f)) <= 0 {
		// The file is completely empty!
		return nil
	}

	var parsedData map[string]uint32

	// Parse the data
	err = json.Unmarshal(f, &parsedData)
	if err != nil {
		// This is now a problem for the function caller :D
		return err
	}

	storageDir := getStorageDirectory()
	// Loop through all of the mapped files
	for fileName := range parsedData {
		// Grab the path to the file
		// TODO: Create a generic function that checks if a post exists by name to use here adn in the GetPost method.
		filePath := path.Join(storageDir, fileName)
		// Check if the file exists
		if _, err := os.Stat(filePath); !os.IsNotExist(err) {
			// Looks like the file does not exist anymore, remove it from the map
			delete(parsedData, fileName)
		}
	}

	// Update the current cache
	ViewsCache = parsedData

	return nil
}

func AddViewToPost(postName string) uint32 {
	// Lock the viewers mapping
	viewersLock.Lock()

	// Check if the map has any value set to this elem
	if _, ok := ViewsCache[postName]; !ok {
		// Set the map value
		ViewsCache[postName] = 0
	}

	// Add to the counter
	ViewsCache[postName]++

	// Unlock
	viewersLock.Unlock()

	// Return the value
	return ViewsCache[postName]
}

func SavePostViewsCache() error {

	data, err := json.Marshal(ViewsCache)
	if err != nil {
		return err
	}

	viewsFilePath, err := getViewsFilePath()
	if err != nil {
		return err
	}

	f, err := os.OpenFile(viewsFilePath, os.O_WRONLY|os.O_CREATE, 0640)
	if err != nil {
		return err
	}
	// Actually close the file
	defer f.Close()

	// Delete all past content
	err = f.Truncate(0)
	if err != nil {
		return err
	}

	// Reset pointer
	_, err = f.Seek(0, 0)
	if err != nil {
		return err
	}

	// Write the json into storage
	_, err = f.Write(data)

	return err
}

// Get the path to the storage directory
func getStorageDirectory() string {

	baseStoragePath, exists := os.LookupEnv("PAD_STORAGE_PATH")
	if !exists {
		baseStoragePath = "/tmp/"
	}

	// Check if the base storage path exists
	if _, err := os.Stat(baseStoragePath); os.IsNotExist(err) {
		// Looks like the base storage path was NOT set, create the dir
		err = os.Mkdir(baseStoragePath, 0640)
		// Check for errors
		if err != nil {
			// No way this sends an error unless it goes horribly wrong.
			panic(err)
		}
	}

	// Return the base storage path
	return baseStoragePath
}

// Get a post from the file system
func GetPost(fileName string) Post {
	// Get the base storage directory and make sure it exists
	storageDir := getStorageDirectory()

	// Generate the file path
	filePath := fmt.Sprintf("%s%s", storageDir, fileName)

	// Get the post views and add 1 to them
	postViews := AddViewToPost(fileName)

	p := Post{
		Name:         fileName,
		Content:      "",
		Views:        postViews,
		LastModified: "Never Before",
	}

	// Check if the file exits
	if _, err := os.Stat(filePath); !os.IsNotExist(err) {
		// File does exist, read it and set the content
		data, err := os.ReadFile(filePath)
		if err != nil {
			fmt.Println("Error:", err)
		}

		// Get the content of the file and put it in the response
		p.Content = string(data)

		// Get last modified date
		fileData, err := os.Stat(filePath)
		if err == nil {
			p.LastModified = fileData.ModTime().Format("02/01/2006 03:04:05 PM")
		} else {
			fmt.Println(err)
		}
	}

	return p
}

// Write a post to the file system
func WritePost(p Post) error {

	maximumPadSize := helper.GetMaximumPadSize()
	if len(p.Content) > maximumPadSize {
		return errors.New("The pad is too big, please limit to the maximum of " + fmt.Sprint(maximumPadSize) + " characters")
	}

	// Get the base storage directory and make sure it exists
	storageDir := getStorageDirectory()

	// Generate the file path
	filePath := fmt.Sprintf("%s%s", storageDir, p.Name)

	f, err := os.OpenFile(filePath, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0755)
	if err != nil {
		return err
	}

	// Write the contnets
	_, err = f.WriteString(p.Content)
	if err != nil {
		return err
	}

	return f.Close()
}

// Cleanup all of the older posts based on the environment settings
func CleanupPosts(age int) {
	// Initialize the files buffer
	var files []string

	// Get the base path
	root := getStorageDirectory()

	// Scan the directory
	err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {

		// Check that the file is not a dir
		if !info.IsDir() {
			files = append(files, path)
		}

		return nil
	})

	// Check if we had any errors
	if err != nil {
		fmt.Println("[Task::Cleanup]: Error", err)
	}

	// The timestamp where the file should be deleted
	tooOldTime := time.Now()

	// Go through all files and process them
	for _, filePath := range files {

		// Get last modified date
		fileData, err := os.Stat(filePath)
		if err == nil {
			fileAge := fileData.ModTime()

			// Check if the file is too old
			if fileAge.Add(time.Duration(age)*time.Minute).Unix() < tooOldTime.Unix() {
				fmt.Println("[Task::Cleanup]: Removing File", filePath)

				// Remove the file
				err = os.Remove(filePath)
				if err != nil {
					fmt.Println("[Task::Cleanup]: Failed to remove file", filePath)
				}
			}

		} else {
			fmt.Println("[Task::Cleanup]: Error", err)
		}

	}
}