⚡: add emoji key and fix matching (#1)

dstrap.cr

@@ -1,38 +1,40 @@
 require "colorize"
 
+targets : Array(String) = [".", ".."]
+emojis : Hash(String, String) = {
+    "LICENSE" => "⚖️",
+    ".*\.(md|rst|txt)" => "📝",
+    ".*\.(py|js|rb|go|rs|c|cs|cpp)$" => "🖥️",
+    "(config.json)|(.*\.(yml|yaml|toml|ini))$" => "⚙️",
+    "(package.json)|(package-lock.json)|(.*\.lock)$" => "📦"
+}
 
 # Display current directory
-print("pwd".colorize(:red).mode(:bold), ':')
-print(" #{File.basename(Dir.current)}/\n".colorize.mode(:bold))
-
-
-# List current directory children
-print("\n./".colorize.mode(:bold))
-Dir.children(".").each do |d|
-    info = File.info("./#{d}")
-    aft = File.directory?(d) ? "/" : ""
-    pre = File.directory?(d) ? "  📂  " : "  📃  "
-
-    print(
-        "\n#{pre}#{d.colorize.mode(:bold)}#{aft}".ljust(30),
-        info.permissions.to_s.split(' ')[0].colorize(:light_cyan),  
-        "  ", "#{info.size} B".colorize(:light_yellow).mode(:underline)
-    )
-end
+print("pwd".colorize(:red).mode(:bold))
+print(": #{File.basename(Dir.current)}/\n".colorize.mode(:bold))
+
+
+# List contents of each target
+targets.each do |dirname|
+    print("\n#{dirname}/".colorize.mode(:bold))
+
+    Dir.children("#{dirname}/").each do |f|
+        file : File = File.new("#{dirname}/#{f}")
+        info : File::Info = file.info
 
+        suf : String = File.directory?(file.path) ? "/" : " "
+        pre : String = File.directory?(file.path) ? "📂" : "📃"
 
-# List parent directory children
-print("\n../".colorize.mode(:bold))
-Dir.children("..").each do |d|
-    info = File.info("../#{d}")
-    aft = File.directory?(d) ? "/" : ""
-    pre = File.directory?(d) ? "  📂  " : "  📃  "
+        emojis.each do |key, val| 
+            pre = /#{key}/ =~ f ? val : pre
+        end
 
-    print(
-        "\n#{pre}#{d.colorize.mode(:bold)}#{aft}".ljust(30),
-        info.permissions.to_s.split(' ')[0].colorize(:light_cyan), 
-        "  ", "#{info.size} B".colorize(:light_yellow).mode(:underline)
-    )
+        print(
+            "\n  #{pre}  #{f}#{suf}".ljust(26, ' ').colorize.mode(:bold),
+            info.permissions.to_s.split(' ')[0].colorize(:light_cyan),
+            " ", "#{info.size} B".colorize(:light_yellow).mode(:underline)
+        )
+    end
 end