#!/usr/bin/env python

import glob
import fnmatch
import itertools
import os
import re

TODO_PATTERN = re.compile(r'\s*// TODO:')
DOC_PATTERN = re.compile(r'\s*//')
EXPECT_PATTERN = re.compile(r'// expect')

num_files = 0
num_docs = 0
num_code = 0
num_empty = 0
num_todos = 0
num_semicolons = 0
num_test_files = 0
num_test_todos = 0
num_expects = 0
num_test_empty = 0
num_test = 0
num_benchmark_files = 0
num_benchmark_todos = 0
num_benchmark_empty = 0
num_benchmark = 0

files = itertools.chain(glob.iglob("src/vm/*.[ch]"),
                        glob.iglob("src/include/*.[ch]"))
for source_path in files:
  num_files += 1
  with open(source_path, "r") as input:
    for line in input:
      num_semicolons += line.count(';')
      match = TODO_PATTERN.match(line)
      if match:
        num_todos += 1
        continue

      match = DOC_PATTERN.match(line)
      if match:
        num_docs += 1
        continue

      stripped = line.strip()
      # Don't count { or } lines since Wren's coding style puts them on their
      # own lines but they don't add anything meaningful to the length of the
      # program.
      if (stripped == "" or stripped == "{" or stripped == "}"):
        num_empty += 1
        continue

      num_code += 1

for dir_path, dir_names, file_names in os.walk("test"):
  for file_name in fnmatch.filter(file_names, "*.wren"):
    num_test_files += 1
    with open(os.path.join(dir_path, file_name), "r") as input:
      for line in input:
        if (line.strip() == ""):
          num_test_empty += 1
        else:
          num_test += 1

        match = TODO_PATTERN.match(line)
        if match:
          num_test_todos += 1
          continue

        match = EXPECT_PATTERN.search(line)
        if match:
          num_expects += 1
          continue

print("source:")
print("  files           " + str(num_files))
print("  semicolons      " + str(num_semicolons))
print("  TODOs           " + str(num_todos))
print("  comment lines   " + str(num_docs))
print("  code lines      " + str(num_code))
print("  empty lines     " + str(num_empty))
print("\n")
print("test:")
print("  files           " + str(num_test_files))
print("  TODOs           " + str(num_test_todos))
print("  expectations    " + str(num_expects))
print("  non-empty lines " + str(num_test))
print("  empty lines     " + str(num_test_empty))
