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)
}