bjoernf/mdtoc
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

used struct to represent the headers

Browse Source
This commit is contained in:
bjt-user 2024-04-30 10:50:58 +02:00
parent d17e61f08f
commit abd4c60339
1 changed files with 17 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
main.go
View File

@ -13,6 +13,12 @@ import (
"strings"
)
type heading struct {
text string
level int
line int
}
func is_heading(line string) bool {
if strings.HasPrefix(line, "#") {
return true
@ -21,20 +27,22 @@ func is_heading(line string) bool {
}
}
func get_heading_level(heading string) int {
func get_heading_level(heading_text string) int {
level := 0
for i := 0; heading[i] == '#'; i++ {
for i := 0; heading_text[i] == '#'; i++ {
level += 1
}
return level
}
func print_toc(headings []string) {
func print_toc(headings []heading) {
tab_count := 0
for index, value := range headings {
for index, _ := range headings {
if index > 0 {
prev_level := get_heading_level(headings[index-1])
cur_level := get_heading_level(headings[index])
prev_level := get_heading_level(headings[index-1].text)
cur_level := get_heading_level(headings[index].text)
headings[index].level = cur_level
if cur_level > prev_level {
tab_count += (cur_level - prev_level)
@ -45,14 +53,14 @@ func print_toc(headings []string) {
fmt.Printf("\t")
}
}
fmt.Printf("%s\n", value)
fmt.Printf("%v\n", headings[index])
}
}
func main() {
file_content_raw, err := os.ReadFile(os.Args[1])
var headings []string = nil
var headings []heading = nil
if err != nil {
log.Fatal(err)
@ -64,7 +72,7 @@ func main() {
for _, value := range file_lines {
if is_heading(value) {
headings = append(headings, value)
headings = append(headings, heading{value, 0, 0})
}
}