day07_1.nim

import os

if paramCount() != 1:
    echo "Usage: ./dayXX <input file>"
    quit(1)
let filename = paramStr(1)
if not fileExists(filename):
    echo "File not found: ", filename
    quit(1)

## RUN: TEST
# RUN: FULL

import std/strutils
import std/sequtils

type
    File = tuple
        name: string
        size: int64

    Directory = object
        name: string
        size: int64
        parent: ptr Directory
        directories: seq[Directory]
        files: seq[File]

proc newDirectory(name: string, parent: ptr Directory): Directory =
    result = Directory(
        name: name,
        size: 0,
        parent: parent,
        directories: @[],
        files: @[],
    )

proc addToSize(dir: var Directory, size: int64) =
    dir.size += size
    if dir.parent != nil:
        dir.parent[].addToSize(size)

proc addFile(dir: var Directory, name: string, size: int64) = 
    dir.files.add((name: name, size: size))
    dir.addToSize(size)

proc `in`(name: string, dir: Directory): bool =
    result = false
    for subdir in dir.directories:
        if subdir.name == name:
            result = true
            break
    # for filename in dir.files:
    #     if filename.name == name:
    #         result = true
    #         break

proc `notin`(name: string, dir: Directory): bool =
    result = not (name in dir)

let root = newDirectory("/", nil)
var cd: ptr Directory = unsafeAddr(root)

var i = 0
for line in lines(filename):
    # echo line
    let split_line = line.split(" ")
    if i == 0:
        assert(split_line[0] == "$")
        assert(split_line[1] == "cd")
        assert(split_line[2] == "/")
        i += 1
        continue

    if split_line[0] == "$":
        assert(split_line.len > 1)
        if split_line[1] == "cd":
            assert(split_line.len == 3)
            let name = split_line[2]
            if name == "..":
                cd = cd[].parent
            elif name notin cd[]:
                let new_dir = newDirectory(name, cd)
                cd[].directories.add(new_dir)
                cd = unsafeAddr(new_dir)
            else:
                var found = false
                for child in cd.directories:
                    if child.name == name:
                        cd = unsafeAddr(child)
                        found = true
                        break
                assert(found)
        elif split_line[1] == "ls":
            assert(split_line.len == 2)
        else:
            raise newException(ValueError, "Unknown command: " & split_line[1])
    else:
        assert(split_line.len == 2)
        if split_line[0] == "dir":
            let name = split_line[1]
            if name notin cd[]:
                cd[].directories.add(newDirectory(name, cd))
        else:
            let size = split_line[0].parseInt
            let name = split_line[1]
            cd[].addFile(name, size)

    i += 1


proc print(dir: Directory, indent: int = 0) =
    echo ' '.repeat(indent) & "- " & dir.name & " (dir, size=" & $dir.size & ")"
    for child in dir.directories:
        print(child, indent + 2)
    for file in dir.files:
        echo ' '.repeat(indent + 2) & "- " & file.name & " (file, size=" & $file.size & ")"

# root.print()

const SIZE_LIMIT = 100000

proc findSmall(dir: Directory): seq[Directory] =
    result = @[]
    if dir.size < SIZE_LIMIT:
        result.add(dir)
    for child in dir.directories:
        result = result.concat(findSmall(child))

echo root.findSmall().map(proc (dir: Directory): int64 = dir.size).foldl(a + b)