legit

Fork of https://git.icyphox.sh/legit

  1. 1
  2. 2
  3. 3
  4. 4
  5. 5
  6. 6
  7. 7
  8. 8
  9. 9
  10. 10
  11. 11
  12. 12
  13. 13
  14. 14
  15. 15
  16. 16
  17. 17
  18. 18
  19. 19
  20. 20
  21. 21
  22. 22
  23. 23
  24. 24
  25. 25
  26. 26
  27. 27
  28. 28
  29. 29
  30. 30
  31. 31
  32. 32
  33. 33
  34. 34
  35. 35
  36. 36
  37. 37
  38. 38
  39. 39
  40. 40
  41. 41
  42. 42
  43. 43
  44. 44
  45. 45
  46. 46
  47. 47
  48. 48
  49. 49
  50. 50
  51. 51
  52. 52
  53. 53
  54. 54
  55. 55
  56. 56
  57. 57
  58. 58
  59. 59
  60. 60
  61. 61
  62. 62
  63. 63
  64. 64
  65. 65
  66. 66
  67. 67
  68. 68
  69. 69
  70. 70
  71. 71
  72. 72
  73. 73
  74. 74
  75. 75
  76. 76
  77. 77
  78. 78
  79. 79
  80. 80
  81. 81
  82. 82
  83. 83
  84. 84
  85. 85
  86. 86
  87. 87
  88. 88
  89. 89
  90. 90
  91. 91
  92. 92
  93. 93
  94. 94
  95. 95
  96. 96
  97. 97
  98. 98
  99. 99
  100. 100
  101. 101
  102. 102
  103. 103
  104. 104
  105. 105
  106. 106
  107. 107
  108. 108
  109. 109
  110. 110
// Copyright 2026 Shota FUJI <pockawoooh@gmail.com>
// SPDX-License-Identifier: MIT

package core

import (
	"errors"
	"fmt"
	"html/template"
	"io/fs"
	"os"
	"path/filepath"
	"slices"
	"strings"

	"github.com/microcosm-cc/bluemonday"

	"github.com/pocka/legit/config"
	"github.com/pocka/legit/embed"
	"github.com/pocka/legit/renderer/html"
)

var (
	ErrRepositoryIsIgnored = errors.New("repository is ignored")
)

type Core struct {
	// Config used to instantiate this core.
	Config *config.Config

	StaticDir fs.FS

	ScanDir *os.Root

	Markdown  html.MarkdownRenderer
	Plaintext html.PlaintextRenderer

	// t stores compiled templates for caching.
	tmpl         *template.Template
	templatesDir fs.FS
}

func New(config *config.Config) (*Core, error) {
	var err error

	ugcPolicy := bluemonday.UGCPolicy()

	core := Core{
		Config:    config,
		Markdown:  html.NewMarkdownRenderer(ugcPolicy),
		Plaintext: html.NewPlaintextRenderer(ugcPolicy),
	}

	if config.Dirs.Static != "" {
		root, err := os.OpenRoot(config.Dirs.Static)
		if err != nil {
			return nil, fmt.Errorf("unable to open static dir: %w", err)
		}

		core.StaticDir = root.FS()
	} else {
		core.StaticDir = embed.StaticDir()
	}

	if config.Dirs.Templates != "" {
		core.templatesDir = os.DirFS(config.Dirs.Templates)
	} else {
		core.templatesDir = embed.TemplatesDir()
	}

	if !config.CompileTemplatesOnRequest {
		core.tmpl = template.Must(template.ParseFS(core.templatesDir, "*"))
	}

	core.ScanDir, err = os.OpenRoot(config.Repo.ScanPath)
	if err != nil {
		return nil, fmt.Errorf("unable to open scan dir: %w", err)
	}

	return &core, nil
}

// Template returns compiled HTML templates.
func (core *Core) Template() *template.Template {
	if core.tmpl != nil {
		return core.tmpl
	}

	return template.Must(template.ParseFS(core.templatesDir, "*"))
}

// RepositoryPath finds a repository by name and returns its filepath.
func (core *Core) RepositoryPath(name string) (string, error) {
	stat, err := core.ScanDir.Stat(name)
	if err != nil {
		return "", err
	}

	dirname := stat.Name()
	if slices.Contains(core.Config.Repo.Ignore, dirname) {
		return "", ErrRepositoryIsIgnored
	}

	return filepath.Join(core.ScanDir.Name(), dirname), nil
}

// RepositoryName returns formatted display name of a repository.
func (_ *Core) RepositoryName(path string) string {
	return strings.TrimSuffix(filepath.Base(path), ".git")
}