mirror of
https://github.com/github/semantic.git
synced 2024-12-21 05:41:54 +03:00
39 lines
1.1 KiB
Swift
39 lines
1.1 KiB
Swift
import Cocoa
|
|
import Doubt
|
|
|
|
func readFile(path: String) -> String? {
|
|
guard let data = try? NSString(contentsOfFile: path, encoding: NSUTF8StringEncoding) else { return nil }
|
|
return data as String?
|
|
}
|
|
|
|
typealias Term = Cofree<String, Range<Int>>
|
|
|
|
func termWithInput(string: String) -> Term? {
|
|
let document = ts_document_make()
|
|
defer { ts_document_free(document) }
|
|
return string.withCString {
|
|
ts_document_set_language(document, ts_language_javascript())
|
|
ts_document_set_input_string(document, $0)
|
|
ts_document_parse(document)
|
|
let root = ts_document_root_node(document)
|
|
|
|
return Cofree
|
|
.ana { node in
|
|
let count = ts_node_named_child_count(node)
|
|
guard count > 0 else {
|
|
return String.fromCString(ts_node_name(node, document)).map(Syntax.Leaf)!
|
|
}
|
|
return .Indexed((0..<count).map { ts_node_named_child(node, $0) })
|
|
} (root)
|
|
.map {
|
|
let start = ts_node_pos($0).chars
|
|
return start..<(start + ts_node_size($0).chars)
|
|
}
|
|
}
|
|
}
|
|
|
|
let arguments = BoundsCheckedArray(array: Process.arguments)
|
|
if let a = arguments[1].flatMap(readFile).flatMap(termWithInput) {
|
|
print(a)
|
|
}
|