From 80ac5c569114e9f99a37119d97a6050b438f573a Mon Sep 17 00:00:00 2001
From: Akshay <nerdy@peppe.rs>
Date: Sun, 14 Jul 2024 19:31:42 +0100
Subject: init

---
 .gitignore    |   1 +
 Cargo.lock    | 573 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 Cargo.toml    |  25 +++
 readme.txt    |  10 +
 src/app.rs    | 259 ++++++++++++++++++++++++++
 src/config.rs |  26 +++
 src/main.rs   | 102 +++++++++++
 7 files changed, 996 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 Cargo.lock
 create mode 100644 Cargo.toml
 create mode 100644 readme.txt
 create mode 100644 src/app.rs
 create mode 100644 src/config.rs
 create mode 100644 src/main.rs

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..ea8c4bf
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+/target
diff --git a/Cargo.lock b/Cargo.lock
new file mode 100644
index 0000000..a3ecfa2
--- /dev/null
+++ b/Cargo.lock
@@ -0,0 +1,573 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 3
+
+[[package]]
+name = "aho-corasick"
+version = "1.1.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8e60d3430d3a69478ad0993f19238d2df97c507009a52b3c10addcd7f6bcb916"
+dependencies = [
+ "memchr",
+]
+
+[[package]]
+name = "bitflags"
+version = "1.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
+
+[[package]]
+name = "cc"
+version = "1.0.106"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "066fce287b1d4eafef758e89e09d724a24808a9196fe9756b8ca90e86d0719a2"
+
+[[package]]
+name = "cfg-if"
+version = "1.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
+
+[[package]]
+name = "console"
+version = "0.15.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0e1f83fc076bd6dd27517eacdf25fef6c4dfe5f1d7448bafaaf3a26f13b5e4eb"
+dependencies = [
+ "encode_unicode",
+ "lazy_static",
+ "libc",
+ "unicode-width",
+ "windows-sys 0.52.0",
+]
+
+[[package]]
+name = "encode_unicode"
+version = "0.3.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a357d28ed41a50f9c765dbfe56cbc04a64e53e5fc58ba79fbc34c10ef3df831f"
+
+[[package]]
+name = "filetime"
+version = "0.2.23"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1ee447700ac8aa0b2f2bd7bc4462ad686ba06baa6727ac149a2d6277f0d240fd"
+dependencies = [
+ "cfg-if",
+ "libc",
+ "redox_syscall",
+ "windows-sys 0.52.0",
+]
+
+[[package]]
+name = "inotify"
+version = "0.9.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f8069d3ec154eb856955c1c0fbffefbf5f3c40a104ec912d4797314c1801abff"
+dependencies = [
+ "bitflags",
+ "inotify-sys",
+ "libc",
+]
+
+[[package]]
+name = "inotify-sys"
+version = "0.1.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e05c02b5e89bff3b946cedeca278abc628fe811e604f027c45a8aa3cf793d0eb"
+dependencies = [
+ "libc",
+]
+
+[[package]]
+name = "kqueue"
+version = "1.0.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7447f1ca1b7b563588a205fe93dea8df60fd981423a768bc1c0ded35ed147d0c"
+dependencies = [
+ "kqueue-sys",
+ "libc",
+]
+
+[[package]]
+name = "kqueue-sys"
+version = "1.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ed9625ffda8729b85e45cf04090035ac368927b8cebc34898e7c120f52e4838b"
+dependencies = [
+ "bitflags",
+ "libc",
+]
+
+[[package]]
+name = "lazy_static"
+version = "1.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bbd2bcb4c963f2ddae06a2efc7e9f3591312473c50c6685e1f298068316e66fe"
+
+[[package]]
+name = "libc"
+version = "0.2.155"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "97b3888a4aecf77e811145cadf6eef5901f4782c53886191b2f693f24761847c"
+
+[[package]]
+name = "log"
+version = "0.4.22"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a7a70ba024b9dc04c27ea2f0c0548feb474ec5c54bba33a7f72f873a39d07b24"
+
+[[package]]
+name = "memchr"
+version = "2.7.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "78ca9ab1a0babb1e7d5695e3530886289c18cf2f87ec19a575a0abdce112e3a3"
+
+[[package]]
+name = "mio"
+version = "0.8.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a4a650543ca06a924e8b371db273b2756685faae30f8487da1b56505a8f78b0c"
+dependencies = [
+ "libc",
+ "log",
+ "wasi",
+ "windows-sys 0.48.0",
+]
+
+[[package]]
+name = "notify"
+version = "5.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "729f63e1ca555a43fe3efa4f3efdf4801c479da85b432242a7b726f353c88486"
+dependencies = [
+ "bitflags",
+ "filetime",
+ "inotify",
+ "kqueue",
+ "libc",
+ "mio",
+ "walkdir",
+ "windows-sys 0.45.0",
+]
+
+[[package]]
+name = "redox_syscall"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4722d768eff46b75989dd134e5c353f0d6296e5aaa3132e776cbdb56be7731aa"
+dependencies = [
+ "bitflags",
+]
+
+[[package]]
+name = "regex"
+version = "1.10.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b91213439dad192326a0d7c6ee3955910425f441d7038e0d6933b0aec5c4517f"
+dependencies = [
+ "aho-corasick",
+ "memchr",
+ "regex-automata",
+ "regex-syntax",
+]
+
+[[package]]
+name = "regex-automata"
+version = "0.4.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "38caf58cc5ef2fed281f89292ef23f6365465ed9a41b7a7754eb4e26496c92df"
+dependencies = [
+ "aho-corasick",
+ "memchr",
+ "regex-syntax",
+]
+
+[[package]]
+name = "regex-syntax"
+version = "0.8.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7a66a03ae7c801facd77a29370b4faec201768915ac14a721ba36f20bc9c209b"
+
+[[package]]
+name = "same-file"
+version = "1.0.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "93fc1dc3aaa9bfed95e02e6eadabb4baf7e3078b0bd1b4d7b6b0b68378900502"
+dependencies = [
+ "winapi-util",
+]
+
+[[package]]
+name = "tree-sitter"
+version = "0.20.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e747b1f9b7b931ed39a548c1fae149101497de3c1fc8d9e18c62c1a66c683d3d"
+dependencies = [
+ "cc",
+ "regex",
+]
+
+[[package]]
+name = "tree-sitter"
+version = "0.21.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "705bf7c0958d0171dd7d3a6542f2f4f21d87ed5f1dc8db52919d3a6bed9a359a"
+dependencies = [
+ "cc",
+ "regex",
+]
+
+[[package]]
+name = "tree-sitter-elm"
+version = "5.7.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1ee736eec2208cd9ea7f2bf1af505c27df03edaed8c7bac38c29f2b147c6035d"
+dependencies = [
+ "cc",
+ "tree-sitter 0.20.10",
+]
+
+[[package]]
+name = "tree-sitter-go"
+version = "0.19.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "71967701c8214be4aa77e0260e98361e6fd71ceec1d9d03abb37a22c9f60d0ff"
+dependencies = [
+ "cc",
+ "tree-sitter 0.20.10",
+]
+
+[[package]]
+name = "tree-sitter-javascript"
+version = "0.21.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8710a71bc6779e33811a8067bdda3ed08bed1733296ff915e44faf60f8c533d7"
+dependencies = [
+ "cc",
+ "tree-sitter 0.21.0",
+]
+
+[[package]]
+name = "tree-sitter-json"
+version = "0.21.0"
+source = "git+https://github.com/tree-sitter/tree-sitter-json#94f5c527b2965465956c2000ed6134dd24daf2a7"
+dependencies = [
+ "cc",
+ "tree-sitter 0.21.0",
+]
+
+[[package]]
+name = "tree-sitter-md"
+version = "0.2.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d9c3cfd068f2527250bbd8ff407431164e12b17863e7eafb76e311dd3f96965a"
+dependencies = [
+ "cc",
+ "tree-sitter 0.21.0",
+]
+
+[[package]]
+name = "tree-sitter-mdx"
+version = "0.0.1"
+source = "git+https://github.com/jlopezcur/tree-sitter-mdx#df43681bff333228fa60b69c09a1e7a6f9ed1610"
+dependencies = [
+ "cc",
+ "tree-sitter 0.20.10",
+]
+
+[[package]]
+name = "tree-sitter-python"
+version = "0.21.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b4066c6cf678f962f8c2c4561f205945c84834cce73d981e71392624fdc390a9"
+dependencies = [
+ "cc",
+ "tree-sitter 0.21.0",
+]
+
+[[package]]
+name = "tree-sitter-ruby"
+version = "0.21.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c0031f687c0772f2dad7b77104c43428611099a1804c81244ada21560f41f0b1"
+dependencies = [
+ "cc",
+ "tree-sitter 0.21.0",
+]
+
+[[package]]
+name = "tree-sitter-rust"
+version = "0.21.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "277690f420bf90741dea984f3da038ace46c4fe6047cba57a66822226cde1c93"
+dependencies = [
+ "cc",
+ "tree-sitter 0.21.0",
+]
+
+[[package]]
+name = "tree-sitter-typescript"
+version = "0.21.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ecb35d98a688378e56c18c9c159824fd16f730ccbea19aacf4f206e5d5438ed9"
+dependencies = [
+ "cc",
+ "tree-sitter 0.21.0",
+]
+
+[[package]]
+name = "tree-viz"
+version = "0.1.0"
+dependencies = [
+ "console",
+ "notify",
+ "tree-sitter 0.21.0",
+ "tree-sitter-elm",
+ "tree-sitter-go",
+ "tree-sitter-javascript",
+ "tree-sitter-json",
+ "tree-sitter-md",
+ "tree-sitter-mdx",
+ "tree-sitter-python",
+ "tree-sitter-ruby",
+ "tree-sitter-rust",
+ "tree-sitter-typescript",
+]
+
+[[package]]
+name = "unicode-width"
+version = "0.1.13"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0336d538f7abc86d282a4189614dfaa90810dfc2c6f6427eaf88e16311dd225d"
+
+[[package]]
+name = "walkdir"
+version = "2.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "29790946404f91d9c5d06f9874efddea1dc06c5efe94541a7d6863108e3a5e4b"
+dependencies = [
+ "same-file",
+ "winapi-util",
+]
+
+[[package]]
+name = "wasi"
+version = "0.11.0+wasi-snapshot-preview1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9c8d87e72b64a3b4db28d11ce29237c246188f4f51057d65a7eab63b7987e423"
+
+[[package]]
+name = "winapi-util"
+version = "0.1.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4d4cc384e1e73b93bafa6fb4f1df8c41695c8a91cf9c4c64358067d15a7b6c6b"
+dependencies = [
+ "windows-sys 0.52.0",
+]
+
+[[package]]
+name = "windows-sys"
+version = "0.45.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "75283be5efb2831d37ea142365f009c02ec203cd29a3ebecbc093d52315b66d0"
+dependencies = [
+ "windows-targets 0.42.2",
+]
+
+[[package]]
+name = "windows-sys"
+version = "0.48.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "677d2418bec65e3338edb076e806bc1ec15693c5d0104683f2efe857f61056a9"
+dependencies = [
+ "windows-targets 0.48.5",
+]
+
+[[package]]
+name = "windows-sys"
+version = "0.52.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "282be5f36a8ce781fad8c8ae18fa3f9beff57ec1b52cb3de0789201425d9a33d"
+dependencies = [
+ "windows-targets 0.52.6",
+]
+
+[[package]]
+name = "windows-targets"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8e5180c00cd44c9b1c88adb3693291f1cd93605ded80c250a75d472756b4d071"
+dependencies = [
+ "windows_aarch64_gnullvm 0.42.2",
+ "windows_aarch64_msvc 0.42.2",
+ "windows_i686_gnu 0.42.2",
+ "windows_i686_msvc 0.42.2",
+ "windows_x86_64_gnu 0.42.2",
+ "windows_x86_64_gnullvm 0.42.2",
+ "windows_x86_64_msvc 0.42.2",
+]
+
+[[package]]
+name = "windows-targets"
+version = "0.48.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9a2fa6e2155d7247be68c096456083145c183cbbbc2764150dda45a87197940c"
+dependencies = [
+ "windows_aarch64_gnullvm 0.48.5",
+ "windows_aarch64_msvc 0.48.5",
+ "windows_i686_gnu 0.48.5",
+ "windows_i686_msvc 0.48.5",
+ "windows_x86_64_gnu 0.48.5",
+ "windows_x86_64_gnullvm 0.48.5",
+ "windows_x86_64_msvc 0.48.5",
+]
+
+[[package]]
+name = "windows-targets"
+version = "0.52.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9b724f72796e036ab90c1021d4780d4d3d648aca59e491e6b98e725b84e99973"
+dependencies = [
+ "windows_aarch64_gnullvm 0.52.6",
+ "windows_aarch64_msvc 0.52.6",
+ "windows_i686_gnu 0.52.6",
+ "windows_i686_gnullvm",
+ "windows_i686_msvc 0.52.6",
+ "windows_x86_64_gnu 0.52.6",
+ "windows_x86_64_gnullvm 0.52.6",
+ "windows_x86_64_msvc 0.52.6",
+]
+
+[[package]]
+name = "windows_aarch64_gnullvm"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "597a5118570b68bc08d8d59125332c54f1ba9d9adeedeef5b99b02ba2b0698f8"
+
+[[package]]
+name = "windows_aarch64_gnullvm"
+version = "0.48.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2b38e32f0abccf9987a4e3079dfb67dcd799fb61361e53e2882c3cbaf0d905d8"
+
+[[package]]
+name = "windows_aarch64_gnullvm"
+version = "0.52.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "32a4622180e7a0ec044bb555404c800bc9fd9ec262ec147edd5989ccd0c02cd3"
+
+[[package]]
+name = "windows_aarch64_msvc"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e08e8864a60f06ef0d0ff4ba04124db8b0fb3be5776a5cd47641e942e58c4d43"
+
+[[package]]
+name = "windows_aarch64_msvc"
+version = "0.48.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dc35310971f3b2dbbf3f0690a219f40e2d9afcf64f9ab7cc1be722937c26b4bc"
+
+[[package]]
+name = "windows_aarch64_msvc"
+version = "0.52.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "09ec2a7bb152e2252b53fa7803150007879548bc709c039df7627cabbd05d469"
+
+[[package]]
+name = "windows_i686_gnu"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c61d927d8da41da96a81f029489353e68739737d3beca43145c8afec9a31a84f"
+
+[[package]]
+name = "windows_i686_gnu"
+version = "0.48.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a75915e7def60c94dcef72200b9a8e58e5091744960da64ec734a6c6e9b3743e"
+
+[[package]]
+name = "windows_i686_gnu"
+version = "0.52.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8e9b5ad5ab802e97eb8e295ac6720e509ee4c243f69d781394014ebfe8bbfa0b"
+
+[[package]]
+name = "windows_i686_gnullvm"
+version = "0.52.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0eee52d38c090b3caa76c563b86c3a4bd71ef1a819287c19d586d7334ae8ed66"
+
+[[package]]
+name = "windows_i686_msvc"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "44d840b6ec649f480a41c8d80f9c65108b92d89345dd94027bfe06ac444d1060"
+
+[[package]]
+name = "windows_i686_msvc"
+version = "0.48.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8f55c233f70c4b27f66c523580f78f1004e8b5a8b659e05a4eb49d4166cca406"
+
+[[package]]
+name = "windows_i686_msvc"
+version = "0.52.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "240948bc05c5e7c6dabba28bf89d89ffce3e303022809e73deaefe4f6ec56c66"
+
+[[package]]
+name = "windows_x86_64_gnu"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8de912b8b8feb55c064867cf047dda097f92d51efad5b491dfb98f6bbb70cb36"
+
+[[package]]
+name = "windows_x86_64_gnu"
+version = "0.48.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "53d40abd2583d23e4718fddf1ebec84dbff8381c07cae67ff7768bbf19c6718e"
+
+[[package]]
+name = "windows_x86_64_gnu"
+version = "0.52.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "147a5c80aabfbf0c7d901cb5895d1de30ef2907eb21fbbab29ca94c5b08b1a78"
+
+[[package]]
+name = "windows_x86_64_gnullvm"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "26d41b46a36d453748aedef1486d5c7a85db22e56aff34643984ea85514e94a3"
+
+[[package]]
+name = "windows_x86_64_gnullvm"
+version = "0.48.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0b7b52767868a23d5bab768e390dc5f5c55825b6d30b86c844ff2dc7414044cc"
+
+[[package]]
+name = "windows_x86_64_gnullvm"
+version = "0.52.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "24d5b23dc417412679681396f2b49f3de8c1473deb516bd34410872eff51ed0d"
+
+[[package]]
+name = "windows_x86_64_msvc"
+version = "0.42.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9aec5da331524158c6d1a4ac0ab1541149c0b9505fde06423b02f5ef0106b9f0"
+
+[[package]]
+name = "windows_x86_64_msvc"
+version = "0.48.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ed94fce61571a4006852b7389a063ab983c02eb1bb37b47f8272ce92d06d9538"
+
+[[package]]
+name = "windows_x86_64_msvc"
+version = "0.52.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec"
diff --git a/Cargo.toml b/Cargo.toml
new file mode 100644
index 0000000..73ec928
--- /dev/null
+++ b/Cargo.toml
@@ -0,0 +1,25 @@
+[package]
+name = "tree-viz"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+tree-sitter-elm = "5.6.3"
+tree-sitter-ruby = "0.21"
+tree-sitter-rust = "0.21"
+tree-sitter = "0.21"
+console = "^0.15"
+tree-sitter-go = "0.19.1"
+tree-sitter-typescript = "0.21"
+tree-sitter-javascript = "0.21"
+tree-sitter-mdx = { git = "https://github.com/jlopezcur/tree-sitter-mdx" }
+tree-sitter-json = { git = "https://github.com/tree-sitter/tree-sitter-json" }
+tree-sitter-python = "0.21"
+tree-sitter-md = "0.2.3"
+# tree-sitter-cobol = { path = "../../tree-sitter-cobol/" }
+
+[dependencies.notify]
+version = "5.0.0"
+default-features = false
diff --git a/readme.txt b/readme.txt
new file mode 100644
index 0000000..008008b
--- /dev/null
+++ b/readme.txt
@@ -0,0 +1,10 @@
+tree-viz -- <language> <file> [scope_query]
+
+language is one of:
+
+- typescript
+- javascript
+- python
+- rust
+- markdown
+- ruby
diff --git a/src/app.rs b/src/app.rs
new file mode 100644
index 0000000..285034b
--- /dev/null
+++ b/src/app.rs
@@ -0,0 +1,259 @@
+use crate::config::Config;
+
+use std::{
+    collections::HashMap,
+    fmt::Write,
+    path::{Path, PathBuf},
+};
+
+use console::{style, Style, Term};
+use tree_sitter::{Node, Parser, Query, QueryCursor, Range, Tree};
+
+pub struct App {
+    config: Config,
+    language: tree_sitter::Language,
+    path: PathBuf,
+    query: Option<Query>,
+    query_path: Option<PathBuf>,
+    src: Vec<u8>,
+    tree: Tree,
+}
+
+impl App {
+    pub fn new<'a, P: AsRef<Path>>(
+        src: &'a [u8],
+        path: P,
+        query_path: Option<P>,
+        language: tree_sitter::Language,
+    ) -> Self {
+        let path = path.as_ref().to_owned();
+
+        let mut parser = Parser::new();
+        parser.set_language(&language).unwrap();
+
+        let tree = parser.parse(&src, None).unwrap();
+        let query_path = query_path.map(|q| q.as_ref().to_owned());
+        let query = query_path.as_ref().map(|p| {
+            let query_src = std::fs::read_to_string(&p).expect("unable to read query");
+            Query::new(&language, &query_src).expect("query parse error")
+        });
+
+        Self {
+            config: Default::default(),
+            path,
+            query,
+            query_path,
+            src: src.to_owned(),
+            tree,
+            language,
+        }
+    }
+
+    pub fn draw(&self) {
+        let term = Term::stdout();
+        term.clear_screen().unwrap();
+        let mut done = false;
+        let mut depth = 0;
+        let mut in_capture: Option<Range> = None;
+        let mut cursor = self.tree.walk();
+
+        let capture_names = self
+            .query
+            .as_ref()
+            .map(|q| q.capture_names())
+            .unwrap_or_default();
+        let capture_map = self
+            .query
+            .as_ref()
+            .map(|query| {
+                QueryCursor::new()
+                    .matches(&query, self.tree.root_node(), self.src.as_slice())
+                    .flat_map(|match_| match_.captures)
+                    .fold(
+                        HashMap::new(),
+                        |mut map: HashMap<Node, Vec<u32>>, capture| {
+                            map.entry(capture.node)
+                                .and_modify(|idxs| idxs.push(capture.index))
+                                .or_insert_with(|| vec![capture.index]);
+                            map
+                        },
+                    )
+            })
+            .unwrap_or_default();
+
+        while !done {
+            let node = cursor.node();
+            let mut tree_string = String::new();
+            in_capture = match in_capture {
+                Some(range)
+                    if !contains(&range, &node.range()) && capture_map.contains_key(&node) =>
+                {
+                    Some(node.range())
+                }
+                Some(range) if !contains(&range, &node.range()) => None,
+                None if capture_map.contains_key(&node) => Some(node.range()),
+                i => i,
+            };
+
+            write!(
+                tree_string,
+                "{}",
+                (if in_capture.is_some() {
+                    Style::new().on_yellow().on_bright()
+                } else {
+                    Style::new()
+                })
+                .bright()
+                .black()
+                .apply_to(
+                    format!("{}{}", "|", " ".repeat(self.config.indent_level))
+                        .repeat(depth as usize)
+                )
+            )
+            .unwrap();
+
+            if self.config.show_field_name {
+                if let Some(f) = cursor.field_name() {
+                    write!(
+                        tree_string,
+                        "{} ",
+                        if in_capture.is_some() {
+                            Style::new().on_yellow().on_bright()
+                        } else {
+                            Style::new()
+                        }
+                        .yellow()
+                        .apply_to(f)
+                    )
+                    .unwrap()
+                }
+            }
+
+            write!(
+                tree_string,
+                "{} ",
+                if node.is_error() {
+                    Style::new().red()
+                } else if in_capture.is_some() {
+                    Style::new().on_yellow().on_bright()
+                } else {
+                    Style::new()
+                }
+                .apply_to(node.kind()),
+            )
+            .unwrap();
+
+            if let Some(idxs) = capture_map.get(&node) {
+                for index in idxs {
+                    write!(
+                        tree_string,
+                        "@{} ",
+                        style(capture_names[*index as usize]).magenta()
+                    )
+                    .unwrap();
+                }
+            }
+
+            if self.config.show_ranges {
+                let range = node.range();
+                write!(
+                    tree_string,
+                    " {}",
+                    style(format!("{:?}..{:?}", range.start_byte, range.end_byte,))
+                        .bright()
+                        .black()
+                )
+                .unwrap();
+            }
+
+            if self.config.show_src {
+                write!(
+                    tree_string,
+                    " {:.?}",
+                    style(node.utf8_text(&self.src).unwrap()).cyan()
+                )
+                .unwrap();
+            }
+
+            term.write_line(&tree_string).unwrap();
+            term.clear_to_end_of_screen().unwrap();
+
+            if cursor.goto_first_child() {
+                depth += 1;
+                continue;
+            }
+            if cursor.goto_next_sibling() {
+                continue;
+            }
+
+            loop {
+                if !cursor.goto_parent() {
+                    done = true;
+                    break;
+                } else {
+                    depth -= 1;
+                }
+
+                if cursor.goto_next_sibling() {
+                    break;
+                }
+            }
+        }
+
+        // see https://github.com/console-rs/console/issues/36#issuecomment-624731432
+        // for the reasoning behing this hackjob
+
+        term.write_line("\n(>) increase indent").unwrap();
+        term.clear_to_end_of_screen().unwrap();
+
+        term.write_line("(<) decrease indent ").unwrap();
+        term.clear_to_end_of_screen().unwrap();
+
+        term.write_line("(n) toggle ranges").unwrap();
+        term.clear_to_end_of_screen().unwrap();
+
+        term.write_line("(s) toggle source text").unwrap();
+        term.clear_to_end_of_screen().unwrap();
+
+        term.write_line("(r) reload from disk").unwrap();
+        term.clear_to_end_of_screen().unwrap();
+
+        term.write_line("(C-c) quit").unwrap();
+        term.clear_to_end_of_screen().unwrap();
+    }
+
+    pub fn increase_indent(&mut self) {
+        self.config.indent_level = self.config.indent_level.saturating_add(1);
+    }
+
+    pub fn decrease_indent(&mut self) {
+        self.config.indent_level = self.config.indent_level.saturating_sub(1);
+    }
+
+    pub fn toggle_ranges(&mut self) {
+        self.config.show_ranges = !self.config.show_ranges;
+    }
+
+    pub fn toggle_source(&mut self) {
+        self.config.show_src = !self.config.show_src;
+    }
+
+    pub fn reload(&mut self) {
+        let src = std::fs::read_to_string(&self.path).unwrap();
+        let new = Self::new(
+            src.as_bytes(),
+            &self.path,
+            self.query_path.as_ref(),
+            self.language.clone(),
+        );
+        *self = Self {
+            config: self.config,
+            ..new
+        };
+    }
+}
+
+// does a encompass b
+fn contains(a: &Range, b: &Range) -> bool {
+    a.start_byte <= b.start_byte && a.end_byte >= b.end_byte
+}
diff --git a/src/config.rs b/src/config.rs
new file mode 100644
index 0000000..18ed5cc
--- /dev/null
+++ b/src/config.rs
@@ -0,0 +1,26 @@
+use std::default::Default;
+
+#[derive(Clone, Copy)]
+pub struct Config {
+    pub indent_level: usize,
+    pub show_ranges: bool,
+    pub show_src: bool,
+    pub show_field_name: bool,
+}
+
+impl Default for Config {
+    fn default() -> Self {
+        Config::new()
+    }
+}
+
+impl Config {
+    fn new() -> Self {
+        Self {
+            indent_level: 2,
+            show_ranges: true,
+            show_src: true,
+            show_field_name: true,
+        }
+    }
+}
diff --git a/src/main.rs b/src/main.rs
new file mode 100644
index 0000000..ce7c84c
--- /dev/null
+++ b/src/main.rs
@@ -0,0 +1,102 @@
+mod app;
+mod config;
+
+use std::{
+    env, fs,
+    path::Path,
+    sync::{mpsc, Arc, RwLock},
+    thread,
+    time::Duration,
+};
+
+use app::App;
+use console::{Key, Term};
+use notify::{Event as WatchEvent, EventKind as WatchEventKind, RecursiveMode, Watcher};
+
+fn main() {
+    let mut args = env::args();
+    let _ = args.next();
+
+    let language = match args.next().as_ref().map(|s| s.as_str()) {
+        Some("rust") => tree_sitter_rust::language(),
+        Some("tsx") | Some("typescript") => tree_sitter_typescript::language_tsx(),
+        Some("javascript") => tree_sitter_javascript::language(),
+        Some("python") => tree_sitter_python::language(),
+        Some("ruby") => tree_sitter_ruby::language(),
+        Some("markdown") => tree_sitter_md::language(),
+        Some(s) => panic!("invalid language passed: {s}"),
+        None => panic!("no language passed"),
+    };
+    let path = args.next().expect("no arg passed");
+    let query_path = args.next();
+    let src = fs::read_to_string(&path).expect("unable to read file");
+
+    let app = Arc::new(RwLock::new(App::new(
+        src.as_bytes(),
+        &path,
+        query_path.as_ref(),
+        language,
+    )));
+
+    let watch_fn = |watcher_app: Arc<RwLock<App>>| {
+        move |ev| {
+            if let Ok(WatchEvent {
+                kind: WatchEventKind::Modify(..),
+                ..
+            }) = ev
+            {
+                if let Ok(mut locked) = watcher_app.try_write() {
+                    locked.reload();
+                    locked.draw();
+                };
+            }
+        }
+    };
+
+    let mut watcher1 = notify::recommended_watcher(watch_fn(Arc::clone(&app))).unwrap();
+    watcher1
+        .watch(Path::new(&path), RecursiveMode::NonRecursive)
+        .unwrap();
+
+    let mut watcher2 = notify::recommended_watcher(watch_fn(Arc::clone(&app))).unwrap();
+    if let Some(query_path) = query_path {
+        watcher2
+            .watch(Path::new(&query_path), RecursiveMode::NonRecursive)
+            .unwrap();
+    }
+
+    let (tx, rx) = mpsc::channel();
+    let tx0 = tx.clone();
+    thread::spawn(move || {
+        let term = Term::stdout();
+        loop {
+            if let Ok(Key::Char(ev)) = term.read_key() {
+                tx0.send(ev).unwrap();
+            }
+        }
+    });
+
+    if let Ok(locked) = app.try_read() {
+        locked.draw();
+    }
+
+    loop {
+        match rx.try_recv() {
+            Ok(ev) => {
+                if let Ok(mut locked) = app.try_write() {
+                    match ev {
+                        '>' => locked.increase_indent(),
+                        '<' => locked.decrease_indent(),
+                        'n' => locked.toggle_ranges(),
+                        's' => locked.toggle_source(),
+                        'r' => locked.reload(),
+                        _ => (),
+                    }
+                    locked.draw();
+                }
+            }
+            _ => (),
+        }
+        thread::sleep(Duration::from_millis(10));
+    }
+}
-- 
cgit v1.2.3