29 lines
750 B
Text
29 lines
750 B
Text
|
/-
|
||
|
|
Copyright (c) 2024 Joseph Tooby-Smith. All rights reserved.
|
||
|
|
Released under Apache 2.0 license.
|
||
|
|
Authors: Joseph Tooby-Smith
|
||
|
|
-/
|
||
|
|
import Batteries.Lean.HashSet
|
||
|
|
import Lean
|
||
|
|
/-!
|
||
|
|
|
||
|
|
## Getting an array of all file paths in HepLean.
|
||
|
|
|
||
|
|
-/
|
||
|
|
|
||
|
|
open Lean Elab System
|
||
|
|
|
||
|
|
/-! TODO: Make this definition more functional in style. -/
|
||
|
|
|
||
|
|
/-- Gets an array of all file paths in `HepLean`. -/
|
||
|
|
partial def allFilePaths : IO (Array FilePath) := do
|
||
|
|
go (#[] : Array FilePath) `HepLean ("./HepLean" : FilePath)
|
||
|
|
where go prev root path := do
|
||
|
|
let mut r := prev
|
||
|
|
for entry in ← path.readDir do
|
||
|
|
if ← entry.path.isDir then
|
||
|
|
r ← go r (root.mkStr entry.fileName) entry.path
|
||
|
|
else
|
||
|
|
r := r.push (root.toString.replace "." "/" ++ "/" ++ entry.fileName)
|
||
|
|
pure r
|