112 lines
1.9 KiB
Go
112 lines
1.9 KiB
Go
package main
|
|
|
|
// TODO: read a file line by line
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
type heading struct {
|
|
text string
|
|
level int
|
|
line int
|
|
}
|
|
|
|
func is_heading(line string) bool {
|
|
if strings.HasPrefix(line, "#") {
|
|
return true
|
|
} else {
|
|
return false
|
|
}
|
|
}
|
|
|
|
func get_heading_level(heading_text string) int {
|
|
level := 0
|
|
for i := 0; heading_text[i] == '#'; i++ {
|
|
level += 1
|
|
}
|
|
return level
|
|
}
|
|
|
|
func print_toc(headings []heading) {
|
|
tab_count := 0
|
|
for index, _ := range headings {
|
|
cur_level := get_heading_level(headings[index].text)
|
|
|
|
headings[index].level = cur_level
|
|
|
|
if index > 0 {
|
|
prev_level := get_heading_level(headings[index-1].text)
|
|
|
|
if cur_level > prev_level {
|
|
tab_count += (cur_level - prev_level)
|
|
} else if cur_level < prev_level {
|
|
tab_count -= (prev_level - cur_level)
|
|
}
|
|
for i := 0; i < tab_count; i++ {
|
|
fmt.Printf("\t")
|
|
}
|
|
}
|
|
fmt.Printf("%v\n", headings[index])
|
|
}
|
|
}
|
|
|
|
func count_levels(headings []heading) map[string]int {
|
|
level_count := map[string]int{
|
|
"header1": 0,
|
|
"header2": 0,
|
|
"header3": 0,
|
|
"header4": 0,
|
|
"header5": 0,
|
|
"header6": 0,
|
|
}
|
|
|
|
for index, _ := range headings {
|
|
switch headings[index].level {
|
|
case 1:
|
|
level_count["header1"]++
|
|
case 2:
|
|
level_count["header2"]++
|
|
case 3:
|
|
level_count["header3"]++
|
|
case 4:
|
|
level_count["header4"]++
|
|
case 5:
|
|
level_count["header5"]++
|
|
case 6:
|
|
level_count["header6"]++
|
|
}
|
|
}
|
|
|
|
return level_count
|
|
}
|
|
|
|
func main() {
|
|
file_content_raw, err := os.ReadFile(os.Args[1])
|
|
|
|
var headings []heading = nil
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
var file_content string = string(file_content_raw)
|
|
|
|
file_lines := strings.Split(file_content, "\n")
|
|
|
|
for index, value := range file_lines {
|
|
if is_heading(value) {
|
|
headings = append(headings, heading{value, 0, (index + 1)})
|
|
}
|
|
}
|
|
|
|
print_toc(headings)
|
|
|
|
header_levels := count_levels(headings)
|
|
|
|
fmt.Printf("%v\n", header_levels)
|
|
}
|