-
Notifications
You must be signed in to change notification settings - Fork 0
/
templater.go
58 lines (51 loc) · 1.34 KB
/
templater.go
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
package gintemplater
import (
"log"
"os"
"github.com/gin-gonic/gin"
)
//Templater instance to perform hot reload on Gin Web Framework
type Templater struct {
logger *log.Logger
engine *gin.Engine
builder *templaterBuilder
watcher *templaterWatcher
hotReload bool
outputDir string
}
func (t *Templater) Run() {
t.builder.initBuild()
t.reload()
if t.hotReload {
t.watcher.Run()
t.logger.Println("Templater started")
}
}
func (t *Templater) Stop() {
if t.hotReload {
t.watcher.Stop()
t.logger.Println("Templater stopped")
}
}
func (t *Templater) reload() {
t.engine.LoadHTMLGlob(t.outputDir + "/*")
t.logger.Println("Templater reloaded")
}
//NewTemplater will return a new and ready to use Templater instance
func NewTemplater(engine *gin.Engine, conf *Config) *Templater {
instance := new(Templater)
instance.engine = engine
instance.outputDir = conf.OutputDir
instance.hotReload = conf.AutoReload
instance.logger = log.New(os.Stdout, "[GIN-TEMPLATER] ", log.Ltime)
instance.builder = newTemplaterBuilder(conf.InputDir, conf.OutputDir, instance.logger)
instance.watcher = newTemplaterWatcher(conf.InputDir, instance.logger)
if conf.AutoReload {
rebuildEv := func(data string) {
instance.reload()
}
instance.watcher.addListener(instance.builder.generate)
instance.watcher.addListener(rebuildEv)
}
return instance
}