Skip to content

Commit 44f80a8

Browse files
committed
fix iroiro
Signed-off-by: gotti <18141824+gotti@users.noreply.github.com>
1 parent fafe57a commit 44f80a8

File tree

3 files changed

+18
-21
lines changed

3 files changed

+18
-21
lines changed

cmd/serverside/main.go

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,11 @@ func main() {
2121
if err != nil {
2222
panic(err)
2323
}
24-
data, err := g.Generate()
24+
pages, err := framework.LoadLocalPages()
25+
if err != nil {
26+
panic(err)
27+
}
28+
data, err := g.Generate(&pages)
2529
if err != nil {
2630
panic(err)
2731
}
@@ -39,12 +43,5 @@ func main() {
3943
}
4044
w.Write([]byte(d))
4145
})
42-
http.HandleFunc("/upload", func(w http.ResponseWriter, r *http.Request) {
43-
if r.Method != "POST" {
44-
w.WriteHeader(http.StatusMethodNotAllowed)
45-
w.Write([]byte("Method not allowed"))
46-
return
47-
}
48-
})
4946
http.ListenAndServe(":8080", nil)
5047
}

cmd/static-generate/main.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,12 @@ func main() {
2727
panic(err)
2828
}
2929
// data is a map of path to content
30-
data, err := g.Generate()
30+
31+
pages, err := framework.LoadLocalPages()
32+
if err != nil {
33+
panic(err)
34+
}
35+
data, err := g.Generate(&pages)
3136
if err != nil {
3237
panic(err)
3338
}

pkg/framework/framework.go

Lines changed: 7 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -44,27 +44,23 @@ func NewGenerator(iadd []IndexingAddon, radd []ReplaceAddon, tadd []TemplateAddo
4444
}
4545

4646
// Generate generates the pages
47-
func (g *Generator) Generate() (map[string]string, error) {
47+
func (g *Generator) Generate(pages *Pages) (map[string]string, error) {
4848
ret := make(map[string]string)
49-
pages, err := Load()
50-
if err != nil {
51-
return nil, fmt.Errorf("error loading pages: %w", err)
52-
}
5349
for _, i := range g.indexingAddon {
54-
npages, err := i.GenerateIndexes(&pages)
50+
npages, err := i.GenerateIndexes(pages)
5551
if err != nil {
5652
return nil, fmt.Errorf("error generating indexes: %w", err)
5753
}
58-
pages = *npages
54+
pages = npages
5955
}
6056
for _, r := range g.replaceAddon {
61-
npages, err := r.ReplacePages(&pages)
57+
npages, err := r.ReplacePages(pages)
6258
if err != nil {
6359
return nil, fmt.Errorf("error replacing pages: %w", err)
6460
}
65-
pages = *npages
61+
pages = npages
6662
}
67-
for _, p := range pages {
63+
for _, p := range *pages {
6864
html := p.Contents.ToHTML()
6965
for _, t := range g.templateAddon {
7066
npage, err := t.GeneratePage(p, template.HTML(html))
@@ -99,7 +95,7 @@ type Page struct {
9995
Path string
10096
}
10197

102-
func Load() (Pages, error) {
98+
func LoadLocalPages() (Pages, error) {
10399
parser := mdparser.NewLineParser(nil)
104100
pages := NewPages()
105101
filepath.WalkDir("pages", func(path string, d os.DirEntry, err error) error {
@@ -139,7 +135,6 @@ func Load() (Pages, error) {
139135
//get filename
140136
filename := filepath.Base(path)
141137

142-
143138
_, ok := pages[path]
144139
if ok {
145140
fmt.Printf("duplicated filename!!!, filename=%v\n", filename)

0 commit comments

Comments
 (0)