owl-blogs/repository.go

package owl

import (
	_ "embed"
	"fmt"
	"io/ioutil"
	"os"
	"path"

	"gopkg.in/yaml.v2"
)

//go:embed embed/initial/base.html
var base_template string

var VERSION = "0.0.1"

type Repository struct {
	name             string
	single_user_mode bool
	active_user      string
	allow_raw_html   bool
	HttpClient       HttpClient
	Parser           HtmlParser
}

type RepoConfig struct {
	Domain string `yaml:"domain"`
}

func CreateRepository(name string) (Repository, error) {
	newRepo := Repository{name: name, Parser: OwlHtmlParser{}, HttpClient: OwlHttpClient{}}
	// check if repository already exists
	if dirExists(newRepo.Dir()) {
		return Repository{}, fmt.Errorf("Repository already exists")
	}

	os.Mkdir(newRepo.Dir(), 0755)
	os.Mkdir(newRepo.UsersDir(), 0755)
	os.Mkdir(newRepo.StaticDir(), 0755)

	// copy all files from static_files embed.FS to StaticDir
	staticFiles, _ := embed_files.ReadDir("embed/initial/static")
	for _, file := range staticFiles {
		if file.IsDir() {
			continue
		}
		src_data, _ := embed_files.ReadFile("embed/initial/static/" + file.Name())
		os.WriteFile(newRepo.StaticDir()+"/"+file.Name(), src_data, 0644)
	}

	// copy repo/ to newRepo.Dir()
	init_files, _ := embed_files.ReadDir("embed/initial/repo")
	for _, file := range init_files {
		if file.IsDir() {
			continue
		}
		src_data, _ := embed_files.ReadFile("embed/initial/repo/" + file.Name())
		os.WriteFile(newRepo.Dir()+"/"+file.Name(), src_data, 0644)
	}
	return newRepo, nil
}

func OpenRepository(name string) (Repository, error) {

	repo := Repository{name: name, Parser: OwlHtmlParser{}, HttpClient: OwlHttpClient{}}
	if !dirExists(repo.Dir()) {
		return Repository{}, fmt.Errorf("Repository does not exist: " + repo.Dir())
	}

	return repo, nil
}

func OpenSingleUserRepo(name string, user_name string) (Repository, error) {
	repo, err := OpenRepository(name)
	if err != nil {
		return Repository{}, err
	}
	user, err := repo.GetUser(user_name)
	if err != nil {
		return Repository{}, err
	}
	repo.SetSingleUser(user)
	return repo, nil
}

func (repo Repository) AllowRawHtml() bool {
	return repo.allow_raw_html
}

func (repo *Repository) SetAllowRawHtml(allow bool) {
	repo.allow_raw_html = allow
}

func (repo *Repository) SetSingleUser(user User) {
	repo.single_user_mode = true
	repo.active_user = user.name
}

func (repo Repository) SingleUserName() string {
	return repo.active_user
}

func (repo Repository) Dir() string {
	return repo.name
}

func (repo Repository) StaticDir() string {
	return path.Join(repo.Dir(), "static")
}

func (repo Repository) UsersDir() string {
	return path.Join(repo.Dir(), "users")
}

func (repo Repository) UserUrlPath(user User) string {
	if repo.single_user_mode {
		return "/"
	}
	return "/user/" + user.name + "/"
}

func (repo Repository) FullUrl() string {
	config, _ := repo.Config()
	return config.Domain
}

func (repo Repository) Template() (string, error) {
	// load base.html
	path := path.Join(repo.Dir(), "base.html")
	base_html, err := ioutil.ReadFile(path)
	if err != nil {
		return "", err
	}
	return string(base_html), nil
}

func (repo Repository) Users() ([]User, error) {
	if repo.single_user_mode {
		return []User{{repo: &repo, name: repo.active_user}}, nil
	}

	userNames := listDir(repo.UsersDir())
	users := make([]User, len(userNames))
	for i, name := range userNames {
		users[i] = User{repo: &repo, name: name}
	}
	return users, nil
}

func (repo *Repository) CreateUser(name string) (User, error) {
	new_user := User{repo: repo, name: name}
	// check if user already exists
	if dirExists(new_user.Dir()) {
		return User{}, fmt.Errorf("User already exists")
	}

	// creates repo/name folder if it doesn't exist
	user_dir := new_user.Dir()
	os.Mkdir(user_dir, 0755)
	os.Mkdir(path.Join(user_dir, "meta"), 0755)
	// create public folder
	os.Mkdir(path.Join(user_dir, "public"), 0755)

	// create Meta files
	os.WriteFile(path.Join(user_dir, "meta", "VERSION"), []byte(VERSION), 0644)
	os.WriteFile(path.Join(user_dir, "meta", "base.html"), []byte(base_template), 0644)

	meta, _ := yaml.Marshal(UserConfig{
		Title:       name,
		SubTitle:    "",
		HeaderColor: "#bdd6be",
	})
	os.WriteFile(new_user.ConfigFile(), meta, 0644)

	return new_user, nil
}

func (repo Repository) GetUser(name string) (User, error) {
	user := User{repo: &repo, name: name}
	if !dirExists(user.Dir()) {
		return User{}, fmt.Errorf("User does not exist")
	}
	return user, nil
}

func (repo Repository) PostAliases() (map[string]*Post, error) {
	users, err := repo.Users()
	if err != nil {
		return nil, err
	}
	aliases := make(map[string]*Post)
	for _, user := range users {
		user_aliases, err := user.PostAliases()
		if err != nil {
			return nil, err
		}
		for alias, post := range user_aliases {
			aliases[alias] = post
		}
	}
	return aliases, nil
}

func (repo Repository) Config() (RepoConfig, error) {
	config_path := path.Join(repo.Dir(), "config.yml")
	config_data, err := ioutil.ReadFile(config_path)
	if err != nil {
		return RepoConfig{}, err
	}
	var meta RepoConfig
	err = yaml.Unmarshal(config_data, &meta)
	if err != nil {
		return RepoConfig{}, err
	}
	return meta, nil
}