owl-blogs/user.go

163 lines
3.5 KiB
Go
Raw Normal View History

2022-08-03 14:55:48 +00:00
package owl
2022-07-20 17:35:31 +00:00
import (
"fmt"
2022-07-21 17:44:07 +00:00
"io/ioutil"
2022-07-20 17:35:31 +00:00
"os"
"path"
"time"
2022-07-27 19:53:56 +00:00
"gopkg.in/yaml.v2"
2022-07-20 17:35:31 +00:00
)
type User struct {
2022-08-05 20:04:03 +00:00
repo *Repository
2022-07-20 17:35:31 +00:00
name string
}
2022-07-27 19:53:56 +00:00
type UserConfig struct {
Title string `yaml:"title"`
SubTitle string `yaml:"subtitle"`
HeaderColor string `yaml:"header_color"`
}
2022-07-20 17:35:31 +00:00
func (user User) Dir() string {
2022-07-24 14:19:21 +00:00
return path.Join(user.repo.UsersDir(), user.name)
2022-07-20 17:35:31 +00:00
}
2022-08-03 17:41:13 +00:00
func (user User) UrlPath() string {
return user.repo.UserUrlPath(user)
2022-07-23 15:19:47 +00:00
}
func (user User) PostDir() string {
return path.Join(user.Dir(), "public")
}
2022-07-27 19:53:56 +00:00
func (user User) MetaDir() string {
return path.Join(user.Dir(), "meta")
}
func (user User) ConfigFile() string {
return path.Join(user.MetaDir(), "config.yml")
}
2022-07-20 17:35:31 +00:00
func (user User) Name() string {
return user.name
}
2022-07-21 17:44:07 +00:00
func (user User) Posts() ([]string, error) {
2022-08-03 16:03:10 +00:00
postFiles := listDir(path.Join(user.Dir(), "public"))
2022-07-23 15:19:47 +00:00
posts := make([]string, 0)
for _, id := range postFiles {
2022-08-03 16:03:10 +00:00
// if is a directory and has index.md, add to posts
if dirExists(path.Join(user.Dir(), "public", id)) {
if fileExists(path.Join(user.Dir(), "public", id, "index.md")) {
posts = append(posts, id)
}
2022-07-23 15:19:47 +00:00
}
}
return posts, nil
2022-07-21 17:44:07 +00:00
}
func (user User) GetPost(id string) (Post, error) {
// check if posts index.md exists
if !fileExists(path.Join(user.Dir(), "public", id, "index.md")) {
return Post{}, fmt.Errorf("post %s does not exist", id)
}
2022-08-05 20:04:03 +00:00
post := Post{user: &user, id: id}
2022-07-21 17:44:07 +00:00
_, metaData := post.MarkdownData()
title := metaData["title"]
post.title = fmt.Sprint(title)
return post, nil
2022-07-20 19:12:18 +00:00
}
2022-07-21 17:02:37 +00:00
func (user User) CreateNewPost(title string) (Post, error) {
2022-07-20 17:35:31 +00:00
timestamp := time.Now().UTC().Unix()
folder_name := fmt.Sprintf("%d-%s", timestamp, title)
post_dir := path.Join(user.Dir(), "public", folder_name)
// if post already exists, add -n to the end of the name
i := 0
for {
if dirExists(post_dir) {
i++
folder_name = fmt.Sprintf("%d-%s-%d", timestamp, title, i)
post_dir = path.Join(user.Dir(), "public", folder_name)
} else {
break
}
}
2022-08-05 20:04:03 +00:00
post := Post{user: &user, id: folder_name, title: title}
2022-07-21 17:44:07 +00:00
initial_content := ""
initial_content += "---\n"
initial_content += "title: " + title + "\n"
initial_content += "---\n"
initial_content += "\n"
initial_content += "Write your post here.\n"
2022-07-20 17:35:31 +00:00
// create post file
os.Mkdir(post_dir, 0755)
2022-07-21 17:02:37 +00:00
os.WriteFile(post.ContentFile(), []byte(initial_content), 0644)
2022-08-01 17:50:29 +00:00
// create media dir
os.Mkdir(post.MediaDir(), 0755)
2022-07-21 17:02:37 +00:00
return post, nil
2022-07-20 17:35:31 +00:00
}
2022-07-21 17:44:07 +00:00
func (user User) Template() (string, error) {
// load base.html
path := path.Join(user.Dir(), "meta", "base.html")
base_html, err := ioutil.ReadFile(path)
if err != nil {
return "", err
}
return string(base_html), nil
}
2022-07-27 19:53:56 +00:00
func (user User) Config() (UserConfig, error) {
config_path := user.ConfigFile()
config_data, err := ioutil.ReadFile(config_path)
if err != nil {
return UserConfig{}, err
}
var meta UserConfig
err = yaml.Unmarshal(config_data, &meta)
if err != nil {
return UserConfig{}, err
}
return meta, nil
}
func (user User) SetConfig(new_config UserConfig) error {
config_path := user.ConfigFile()
config_data, err := yaml.Marshal(new_config)
if err != nil {
return err
}
err = ioutil.WriteFile(config_path, config_data, 0644)
if err != nil {
return err
}
return nil
}
2022-08-06 17:38:13 +00:00
func (user User) PostAliases() (map[string]*Post, error) {
post_aliases := make(map[string]*Post)
posts, err := user.Posts()
if err != nil {
return post_aliases, err
}
for _, id := range posts {
post, err := user.GetPost(id)
if err != nil {
return post_aliases, err
}
for _, alias := range post.Aliases() {
post_aliases[alias] = &post
}
}
return post_aliases, nil
}