Makefile

@@ -1,17 +1,11 @@
 all:
-	go build .
+	go run main.go tree.go test_files/test.md
-
-run:
-	go run . test_files/test.md
 
 test:
-	go run . test_files/test.md
+	go run main.go tree.go test_files/test.md
-	go run . test_files/README.md
+	go run main.go tree.go test_files/README.md
-	go run . test_files/weird_headers.md
+	go run main.go tree.go test_files/weird_headers.md
-	go run . test_files/audio.md
+	go run main.go tree.go test_files/audio.md
-
-install:
-	cp -v mdtoc /usr/local/bin/.
 
 clean:
-	go clean
+	rm -vf main

main.go

@@ -61,19 +61,8 @@ func parse_file(file_name string) []heading {
 	return headings
 }
 
-func usage() {
-	fmt.Printf("Usage: mdtoc [FILE]\n")
-	os.Exit(1)
-}
-
 func main() {
-	var file_name string = ""
+	file_name := os.Args[1]
-
-	if len(os.Args) == 2 {
-		file_name = os.Args[1]
-	} else {
-		usage()
-	}
 
 	var headings []heading = nil