mdtoc/main.go

package main

// TODO: read a file line by line

import (
	"fmt"
	"log"
	"os"
	"path"
	"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
		if i == (len(heading_text) - 1) {
			break
		}
	}
	return level
}

func parse_file(file_name string) []heading {
	var headings []heading

	file_content_raw, err := os.ReadFile(file_name)

	if err != nil {
		log.Fatal(err)
	}

	file_lines := strings.Split(string(file_content_raw), "\n")

	is_codeblock := false

	for index, value := range file_lines {
		if strings.HasPrefix(value, "```") {
			is_codeblock = !is_codeblock
		}

		if !is_codeblock && is_heading(value) {
			headings = append(
				headings, heading{value, Get_heading_level(value), (index + 1)})
		}
	}
	return headings
}

func main() {
	file_name := os.Args[1]

	var headings []heading = nil

	headings = parse_file(file_name)

	fmt.Println(path.Base(file_name))
	tree(-1, "", headings)
}