Add Go support (#200)

Author: JuanCortezEclyp

Cargo.lock

@@ -5,6 +5,7 @@ members = [
     "treeedb-c",
     "treeedb-cpp",
     "treeedb-csharp",
+    "treeedb-go",
     "treeedb-java",
     "treeedb-javascript",
     "treeedb-rust",
@@ -17,6 +18,7 @@ members = [
     "treeedbgen-souffle-c",
     "treeedbgen-souffle-cpp",
     "treeedbgen-souffle-csharp",
+    "treeedbgen-souffle-go",
     "treeedbgen-souffle-java",
     "treeedbgen-souffle-javascript",
     "treeedbgen-souffle-php",

Cargo.toml

@@ -0,0 +1,16 @@
+[package]
+name = "treeedb-go"
+version = "0.1.0"
+edition = "2021"
+description = "Generate Datalog facts from Go source code"
+keywords = ["datalog", "go", "parsing", "souffle"]
+authors = ["Langston Barrett <langston.barrett@gmail.com>"]
+license = "MIT"
+readme = "../README.md"
+homepage = "https://github.com/langston-barrett/treeedb"
+repository = "https://github.com/langston-barrett/treeedb"
+
+[dependencies]
+anyhow = "1.0"
+treeedb = { version = "0.1.0", path = "../treeedb", features = ["cli"] }
+tree-sitter-go = { version = "0.20" }

treeedb-go/Cargo.toml

@@ -0,0 +1,5 @@
+use anyhow::Result;
+
+fn main() -> Result<()> {
+    treeedb::cli::main(tree_sitter_go::language())
+}

treeedb-go/src/main.rs

@@ -0,0 +1,21 @@
+[package]
+name = "treeedbgen-souffle-go"
+version = "0.1.0"
+edition = "2021"
+description = "Generate Soufflé types and relations from the Go tree-sitter grammar"
+keywords = ["datalog", "go", "parsing", "souffle"]
+authors = ["Langston Barrett <langston.barrett@gmail.com>"]
+license = "MIT"
+readme = "../README.md"
+homepage = "https://github.com/langston-barrett/treeedb"
+repository = "https://github.com/langston-barrett/treeedb"
+
+[dependencies]
+anyhow = "1"
+treeedbgen-souffle = { version = "0.1.0", path = "../treeedbgen-souffle", features = ["cli"] }
+tree-sitter-go = { version = "0.20" }
+
+[dev-dependencies]
+assert_cmd = "2.0"
+predicates = "3.1"
+tempfile = "3"

treeedbgen-souffle-go/Cargo.toml

@@ -0,0 +1,5 @@
+use anyhow::Result;
+
+fn main() -> Result<()> {
+    treeedbgen_souffle::cli::main(tree_sitter_go::NODE_TYPES)
+}

treeedbgen-souffle-go/src/main.rs

@@ -0,0 +1,7 @@
+package main
+
+import "fmt"
+
+func main() {
+    fmt.Println("Hello, World!")
+}

treeedbgen-souffle-go/tests/go/hello-world.go

@@ -0,0 +1,33 @@
+use std::process::Command;
+
+use assert_cmd::prelude::*;
+use predicates::prelude::*;
+use tempfile::NamedTempFile;
+
+#[test]
+fn test_gen() -> Result<(), Box<dyn std::error::Error>> {
+    let mut cmd = Command::cargo_bin("treeedbgen-souffle-go")?;
+    let tmp = NamedTempFile::new()?;
+    cmd.arg("-o").arg(tmp.path());
+    cmd.arg("--prefix=c").arg("--printsize");
+    cmd.assert()
+        .success()
+        .stdout(predicate::str::is_empty())
+        .stderr(predicate::str::is_empty());
+
+    let mut cmd = Command::cargo_bin("treeedb-go")?;
+    cmd.arg("tests/go/hello-world.go");
+    cmd.assert()
+        .success()
+        .stdout(predicate::str::is_empty())
+        .stderr(predicate::str::is_empty());
+
+    let mut souffle = Command::new("souffle");
+    souffle.arg(tmp.path());
+    souffle
+        .assert()
+        .success()
+        .stdout(predicate::str::contains("go_node\t17"))
+        .stderr(predicate::str::is_empty());
+    Ok(())
+}