ready hm 21

otus-21/src/otus_21/homework/core.clj

@@ -1,9 +1,74 @@
-(ns otus-21.homework.core)
-
-(defn sum-of-sizes [input]
-  "По журналу сеанса работы в терминале воссоздаёт файловую систему
-и подсчитывает сумму размеров директорий, занимающих на диске до
-100000 байт (сумма размеров не учитывает случай, когда найденные
-директории вложены друг в друга: размеры директорий всё так же
-суммируются)."
-  0)
+(ns otus-21.homework.core
+  (:require [clojure.zip :as z]
+            [clojure.walk :as w]
+            [clojure.string :as s]))
+
+
+(defn map-zipper [m]
+  (z/zipper
+    (fn [x] (or (map? x) (map? (nth x 1))))
+    (fn [x] (seq (if (map? x) x (nth x 1))))
+    (fn [x children]
+      (if (map? x)
+        (into {} children)
+        (assoc x 1 (into {} children))))
+    m))
+
+(defn insert-dir [loc name]
+  (z/append-child loc {name {}}))
+
+(defn insert-file [loc name size]
+  (z/append-child loc [name (parse-long size)]))
+
+(defn dir-loc [name loc]
+  (let [[key value] (z/node loc)]
+    (if (and (= name key) (map? value)) loc)
+    ))
+(defn nav-to-dir [loc name]
+  (some #(dir-loc name %) (->> loc z/down (iterate z/right))))
+(defn parse-command
+  [loc input]
+  (condp re-find input
+    #"\$ cd /" (map-zipper {"/" {}})
+    #"\$ ls" loc
+    #"dir (.+)" :>> (fn [[_ name]] (insert-dir loc name))
+    #"(\d+) (.+)" :>> (fn [[_ size name]] (insert-file loc name size))
+    #"cd \.\." (z/up loc)
+    #"cd (.+)" :>> (fn [[_ name]] (nav-to-dir loc name))))
+
+
+(defn parse [input]
+  (->> input
+       s/split-lines
+       (reduce parse-command nil)
+       z/root))
+
+(defn dir-in-walk? [x]
+  ;  [a {e {i 584}, f 29116, g 2557, h.lst 62596}]
+  (if (and (vector? x) (= 2 (count x)) (map? (second x))) true)
+  )
+(defn sum-of-size-dirs [input max-dir-size]
+  (let [result (atom 0)]
+    (w/postwalk (fn [x]
+                  (if (dir-in-walk? x)
+                    (let
+                      [name (nth x 0)
+                       content (nth x 1)
+                       size (apply + (vals content))]
+                      (if (<= size max-dir-size) (swap! result #(+ size %)))
+                      ;(println "dir " name "size" size ":" content)
+                      [name size]
+                      )
+                    x
+                    )
+                  )
+                (parse input))
+    @result))
+
+(defn sum-of-sizes [input]
+  "По журналу сеанса работы в терминале воссоздаёт файловую систему
+и подсчитывает сумму размеров директорий, занимающих на диске до
+100000 байт (сумма размеров не учитывает случай, когда найденные
+директории вложены друг в друга: размеры директорий всё так же
+суммируются)."
+  (sum-of-size-dirs input 100000))