1
1
mirror of https://github.com/github/semantic.git synced 2024-12-21 13:51:44 +03:00
semantic/prototype/doubt-difftool/main.swift
2015-10-28 14:23:58 -04:00

106 lines
3.3 KiB
Swift

import Cocoa
import Doubt
import Prelude
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>>
struct Info: Equatable {
let range: Range<Int>
let category: Category
enum Category: String {
case Arguments = "arguments"
case Assignment = "assignment"
case Comment = "comment"
case ExpressionStatement = "expression_statement"
case FormalParameters = "formal_parameters"
case Function = "function"
case FunctionCall = "function_call"
case Identifier = "identifier"
case IfStatement = "if_statement"
case MemberAccess = "member_access"
case NewExpression = "new_expression"
case NullLiteral = "null"
case Object = "object"
case Pair = "pair"
case Program = "program"
case RelationalOperator = "rel_op"
case ReturnStatement = "return_statement"
case StatementBlock = "statement_block"
case StringLiteral = "string"
case SubscriptAccess = "subscript_access"
}
}
func == (left: Info, right: Info) -> Bool {
return left.range == right.range && left.category == right.category
}
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)
struct E: ErrorType {}
return try? Cofree
.ana { node, category in
let count = ts_node_named_child_count(node)
guard count > 0 else { return Syntax.Leaf(category.rawValue) }
return try .Indexed((0..<count).map { index in
let child = ts_node_named_child(node, index)
guard let name = String.fromCString(ts_node_name(child, document)) else { throw E() }
guard let category = Info.Category(rawValue: name) else { throw E() }
return (child, category)
})
} (root, Info.Category.Program)
.map { node, category in
let start = ts_node_pos(node).chars
return start..<(start + ts_node_size(node).chars)
}
}
}
let arguments = BoundsCheckedArray(array: Process.arguments)
if let aString = arguments[1].flatMap(readFile), bString = arguments[2].flatMap(readFile), c = arguments[3], ui = arguments[4] {
if let a = termWithInput(aString), b = termWithInput(bString) {
let diff = Interpreter<Term>(equal: Term.equals(annotation: const(true), leaf: ==), comparable: const(true), cost: Free.sum(Patch.difference)).run(a, b)
let range: Range<Int> -> Doubt.JSON = {
let start = $0.startIndex
let end = $0.endIndex
return [
.Number(Double(start)),
.Number(Double(end - start)),
]
}
let JSON: Doubt.JSON = [
"before": .String(aString),
"after": .String(bString),
"diff": diff.JSON(pure: { $0.JSON { $0.JSON(annotation: range, leaf: Doubt.JSON.String) } }, leaf: Doubt.JSON.String, annotation: {
[
"before": range($0),
"after": range($1),
]
}),
]
let data = JSON.serialize()
try data.writeToFile(c, options: .DataWritingAtomic)
let components = NSURLComponents()
components.scheme = "file"
components.path = ui
components.query = c
if let URL = components.URL {
NSWorkspace.sharedWorkspace().openURL(URL)
}
}
}