2022 day 7: Bash parts 1 and 2 (messy, tons of debug)

2022/day07/aoc.bash

@@ -0,0 +1,139 @@
+#!/usr/bin/env bash
+#
+# aoc.bash: Advent of Code 2022, day 7
+#
+# Copyright (C) 2022 Bruno Raoult ("br")
+# Licensed under the GNU General Public License v3.0 or later.
+# Some rights reserved. See COPYING.
+#
+# You should have received a copy of the GNU General Public License along with this
+# program. If not, see <https://www.gnu.org/licenses/gpl-3.0-standalone.html>.
+#
+# SPDX-License-Identifier: GPL-3.0-or-later <https://spdx.org/licenses/GPL-3.0-or-later.html>
+
+. common.bash
+
+declare -A files=()                               # list of files in dir
+declare -A sizes=()                               # file size
+declare -A dirs=()                                # directories
+declare -a lines                                  # shell lines
+declare -i curline=0                              # current line in shell
+declare curdir
+
+print_sizes() {
+    local idx
+    echo +++++++++++++++++++
+    for idx in "${!sizes[@]}"; do
+        printf "size(%s)=%d\n" "$idx" "${sizes[$idx]}"
+    done
+    echo +++++++++++++++++++
+}
+
+do_cd() {
+    local dir="$1"
+    echo ">>> $curline ${lines[$curline]}"
+    case "$dir" in
+        "..")
+            echo ".. before: $curdir"
+            curdir=${curdir%/*}
+            [[ -z $curdir ]] && curdir="/"
+            echo ".. after: $curdir"
+            ;;
+        "/")
+            curdir="/"
+            ;;
+        *)
+            [[ $curdir != "/" ]] && curdir+="/"
+            curdir+="$dir"
+    esac
+    dirs[$curdir]="$curdir"
+    echo "> CD $dir newdir=$curdir"
+}
+
+do_ls() {
+    local info file subdir remain
+    echo ">>> $curline ${lines[$curline]}"
+    ((curline++))
+    while [[ $curline -lt ${#lines[@]} && ${lines[$curline]:0:1} != "\$" ]]; do
+        read -r info file <<< "${lines[$curline]}"
+        files[$curdir]+="$file"
+        if [[ $info != dir ]]; then          # file
+            remain="$curdir/$file"
+            remain=${remain//+(\/)/\/}
+            #sizes[$fullname]=$info
+            # recurse up curdir and adjust sizes
+            #subdir=${curdir}
+            echo "info=$info [$remain]"
+            while [[ -n $remain ]]; do
+                echo "remain=$remain subdir=$subdir"
+                ((sizes[$remain] += info))
+                remain=${remain%/*}
+                subdir=${remain##*/}
+                echo "  -> remain=$remain subdir=$subdir"
+            done
+            (( sizes["/"] += info ))
+        fi
+        echo "ls: ${lines[$curline]}"
+        ((curline++))
+    done
+    ((curline--))
+    print_sizes
+}
+
+parse() {
+    readarray -t lines
+    curline=0
+    declare -a line
+
+    while ((curline < ${#lines[@]})); do
+        read -ra line <<< "${lines[$curline]}"
+        if [[ "${line[0]}" != "\$" ]]; then
+            printf "ERROR line %d = %s\n" "$curline" "${lines[$curline]}"
+            exit 1
+        fi
+        case "${line[1]}" in
+            "cd")
+                do_cd "${line[2]}"
+                ;;
+            "ls")
+                do_ls
+                ;;
+        esac
+        ((curline++))
+        printf "WARNING curline=%d\n lines=%d\n" "$curline" "${#lines[@]}"
+    done
+}
+
+solve() {
+    declare -ig res
+    local part="$1" dir mindir="/"
+    local -i needed
+
+    if ((part == 1)); then
+        for dir in "${dirs[@]}"; do
+            printf "size(%s)=%d\n" "$dir" "${sizes[$dir]}"
+            if ((sizes[$dir] <= 100000 )); then
+                ((res+=sizes[$dir]))
+            fi
+        done
+    else
+        (( needed = sizes["/"] - (70000000-30000000) ))
+        printf "remain=%d\n" "$needed"
+        ((res = sizes["/"]))
+        for dir in "${!dirs[@]}"; do
+            #printf "size(%s)=%d\n" "$dir" "${sizes[$dir]}"
+            if (( sizes[$dir] >= needed )); then
+                printf "dir %s (%d) will free enough res=%d\n" "$dir" "${sizes[$dir]}" "$res"
+                if (( sizes[$dir] <= res )); then
+                    mindir=$dir
+                    ((res = sizes[$mindir]))
+                    printf "new mindir=%s (%d)\n" "$mindir" "$res"
+                fi
+            fi
+        done
+
+    fi
+}
+
+main "$@"
+exit 0