support multiple schemes

kropp · Aug 4, 2015 · 2abf8cc · 2abf8cc
1 parent 0a226de
commit 2abf8cc
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/generator/src/main/kotlin/generator.kt b/generator/src/main/kotlin/generator.kt
@@ -76,6 +76,9 @@ class GeneratorSink : TripleSink {
     }
 
     override fun endStream() {
+    }
+
+    public fun postProcess() {
         for (type in types.values()) {
             if (type.isField && type.isInterface) {
                 type.isField = false
@@ -458,9 +461,13 @@ class GeneratorSink : TripleSink {
 
 fun main(args: Array<String>) {
     val generator = GeneratorSink()
-    val sp = StreamProcessor(RdfaParser.connect(generator))
-    val stream = FileInputStream("generator/resources/schema.rdfa")
-    sp.process(stream, "http://schema.org/")
+    val processor = StreamProcessor(RdfaParser.connect(generator))
+
+    File("generator/resources").listFiles { it.extension == "rdfa" }?.forEach {
+        println("Processing ${it.name}")
+        processor.process(FileInputStream(it), "http://schema.org/")
+    }
+    generator.postProcess()
 
     generator.writeJava(File("src/main/java"), "org.schema")
 }