summaryrefslogtreecommitdiffstats
path: root/internal/cms/import.go
blob: 3b0d697acd883cbbfcfd49fae65c8f3a7b8a975b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
package cms

import (
	"errors"
	"fmt"
	"io/fs"
	"os"
	"path/filepath"
	"sort"
	"strings"
)

func AddPath(root string, opts AddOptions) (AddReport, error) {
	sourcePath, err := expandPath(opts.Path)
	if err != nil {
		return AddReport{}, err
	}
	info, err := os.Stat(sourcePath)
	if err != nil {
		return AddReport{}, err
	}
	if !info.IsDir() {
		doc, err := AddFile(root, opts)
		if err != nil {
			return AddReport{}, err
		}
		return AddReport{Documents: []Document{doc}}, nil
	}
	if opts.Title != "" || opts.Slug != "" {
		return AddReport{}, errors.New("--title and --slug can only be used when importing one file")
	}

	files, err := importableFiles(sourcePath)
	if err != nil {
		return AddReport{}, err
	}
	if len(files) == 0 {
		return AddReport{}, fmt.Errorf("%s has no .md or .gmi files", sourcePath)
	}

	report := AddReport{}
	for _, file := range files {
		fileOpts := opts
		fileOpts.Path = file
		doc, err := AddFile(root, fileOpts)
		if err != nil {
			return report, fmt.Errorf("import %s: %w", file, err)
		}
		report.Documents = append(report.Documents, doc)
	}
	return report, nil
}

func importableFiles(root string) ([]string, error) {
	var files []string
	err := filepath.WalkDir(root, func(pathName string, entry fs.DirEntry, err error) error {
		if err != nil {
			return err
		}
		if entry.IsDir() {
			if strings.HasPrefix(entry.Name(), ".") && pathName != root {
				return filepath.SkipDir
			}
			return nil
		}
		if isImportableContent(pathName) {
			files = append(files, pathName)
		}
		return nil
	})
	sort.Strings(files)
	return files, err
}

func isImportableContent(pathName string) bool {
	switch strings.ToLower(filepath.Ext(pathName)) {
	case ".md", ".markdown", ".gmi", ".gemini":
		return true
	default:
		return false
	}
}