mirror of
https://github.com/facundoolano/jorge.git
synced 2024-12-25 21:58:28 +01:00
117 lines
2.9 KiB
Go
117 lines
2.9 KiB
Go
package commands
|
|
|
|
import (
|
|
"embed"
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/alecthomas/kong"
|
|
"github.com/facundoolano/jorge/site"
|
|
)
|
|
|
|
//go:embed all:initfiles
|
|
var initfiles embed.FS
|
|
|
|
var INIT_CONFIG string = `name: "%s"
|
|
author: "%s"
|
|
url: "%s"
|
|
`
|
|
var INIT_README string = `
|
|
# %s
|
|
|
|
A jorge blog by %s.
|
|
`
|
|
|
|
type Init struct {
|
|
ProjectDir string `arg:"" name:"path" help:"Directory where to initialize the website project."`
|
|
}
|
|
|
|
// Initialize a new jorge project in the given directory,
|
|
// prompting for basic site config and creating default files.
|
|
func (cmd *Init) Run(ctx *kong.Context) error {
|
|
if err := ensureEmptyProjectDir(cmd.ProjectDir); err != nil {
|
|
return err
|
|
}
|
|
|
|
siteName := Prompt("site name")
|
|
siteUrl := Prompt("site url")
|
|
siteAuthor := Prompt("author")
|
|
fmt.Println()
|
|
|
|
// creating config and readme files manually, since I want to use the supplied config values in their
|
|
// contents. (I don't want to render liquid templates in the WalkDir below since some of the initfiles
|
|
// are actual templates that should be left as is).
|
|
configPath := filepath.Join(cmd.ProjectDir, "config.yml")
|
|
configFile := fmt.Sprintf(INIT_CONFIG, siteName, siteAuthor, siteUrl)
|
|
os.WriteFile(configPath, []byte(configFile), site.FILE_RW_MODE)
|
|
fmt.Println("added", configPath)
|
|
|
|
readmePath := filepath.Join(cmd.ProjectDir, "README.md")
|
|
readmeFile := fmt.Sprintf(INIT_README, siteName, siteAuthor)
|
|
os.WriteFile(readmePath, []byte(readmeFile), site.FILE_RW_MODE)
|
|
fmt.Println("added", readmePath)
|
|
|
|
// walk over initfiles fs
|
|
// copy create directories and copy files at target
|
|
|
|
initfilesRoot := "initfiles"
|
|
return fs.WalkDir(initfiles, initfilesRoot, func(path string, entry fs.DirEntry, err error) error {
|
|
if path == initfilesRoot {
|
|
return nil
|
|
}
|
|
subpath, _ := filepath.Rel(initfilesRoot, path)
|
|
targetPath := filepath.Join(cmd.ProjectDir, subpath)
|
|
|
|
// if it's a directory create it at the same location
|
|
if entry.IsDir() {
|
|
return os.MkdirAll(targetPath, DIR_RWE_MODE)
|
|
}
|
|
|
|
// TODO duplicated in site, extract to somewhere else
|
|
// if its a file, copy it over
|
|
targetFile, err := os.Create(targetPath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer targetFile.Close()
|
|
|
|
source, err := initfiles.Open(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer source.Close()
|
|
|
|
_, err = io.Copy(targetFile, source)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Println("added", targetPath)
|
|
return targetFile.Sync()
|
|
})
|
|
}
|
|
|
|
func ensureEmptyProjectDir(projectDir string) error {
|
|
if err := os.Mkdir(projectDir, DIR_RWE_MODE); err != nil {
|
|
// if it fails with dir already exist, check if it's empty
|
|
// https://stackoverflow.com/a/30708914/993769
|
|
if os.IsExist(err) {
|
|
// check if empty
|
|
dir, err := os.Open(projectDir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer dir.Close()
|
|
|
|
// if directory is non empty, fail
|
|
_, err = dir.Readdirnames(1)
|
|
if err == nil {
|
|
return fmt.Errorf("non empty directory %s", projectDir)
|
|
}
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|