From 67a3654451bb99b691d3619fe8f103f1dad1f355 Mon Sep 17 00:00:00 2001
From: Francesco <francesco.bariatti@irisa.fr>
Date: Sat, 16 Apr 2022 14:04:28 +0200
Subject: [PATCH] Prepare: added command-line parameters parsing

---
 prepare/build.gradle                          |  3 +-
 prepare/src/main/guessNNprepare/Main.scala    | 28 +++++++++++++++++++
 .../{ => guessNNprepare}/NamedEntity.scala    |  2 ++
 .../mains/ExtractEntities.scala}              | 21 ++++++++++++--
 .../guessNNprepare/mains/MainCommand.scala    | 13 +++++++++
 5 files changed, 64 insertions(+), 3 deletions(-)
 create mode 100644 prepare/src/main/guessNNprepare/Main.scala
 rename prepare/src/main/{ => guessNNprepare}/NamedEntity.scala (95%)
 rename prepare/src/main/{Main.scala => guessNNprepare/mains/ExtractEntities.scala} (71%)
 create mode 100644 prepare/src/main/guessNNprepare/mains/MainCommand.scala

diff --git a/prepare/build.gradle b/prepare/build.gradle
index cafc7c9..d1e6caa 100644
--- a/prepare/build.gradle
+++ b/prepare/build.gradle
@@ -14,6 +14,7 @@ dependencies {
 	implementation 'org.apache.jena:apache-jena-libs:4.4.0'
 	implementation 'org.apache.logging.log4j:log4j-core:2.17.2'
 	implementation 'org.slf4j:slf4j-simple:1.7.36'
+	implementation 'net.sourceforge.argparse4j:argparse4j:0.9.0'
 }
 
 sourceSets {
@@ -35,7 +36,7 @@ compileJava {
 }
 
 application {
-	mainClass = "Main"
+	mainClass = "guessNNprepare.Main"
 }
 
 
diff --git a/prepare/src/main/guessNNprepare/Main.scala b/prepare/src/main/guessNNprepare/Main.scala
new file mode 100644
index 0000000..95ac12c
--- /dev/null
+++ b/prepare/src/main/guessNNprepare/Main.scala
@@ -0,0 +1,28 @@
+package guessNNprepare
+
+import guessNNprepare.mains.ExtractEntities
+import net.sourceforge.argparse4j.ArgumentParsers
+import net.sourceforge.argparse4j.inf.{ArgumentParser, ArgumentParserException, Namespace}
+
+object Main {
+	def main(args: Array[String]): Unit = {
+		val parser: ArgumentParser = ArgumentParsers.newFor("GuessNN-prepare").singleMetavar(true).build()
+		val subParsers = parser.addSubparsers().dest("command").title("Commands").metavar("COMMAND").help("Command to execute")
+		ExtractEntities.addCliArgs(subParsers.addParser("entities").help(ExtractEntities.description))
+
+		try {
+			val ARGS: Namespace = parser.parseArgs(args)
+			val command = ARGS.getString("command")
+			command match {
+				case "entities" => ExtractEntities.execute(ARGS)
+				case _ =>
+					System.err.printf(s"Unknown command ${command}");
+					System.exit(1)
+			}
+		} catch {
+			case e: ArgumentParserException =>
+				parser.handleError(e)
+				System.exit(1)
+		}
+	}
+}
diff --git a/prepare/src/main/NamedEntity.scala b/prepare/src/main/guessNNprepare/NamedEntity.scala
similarity index 95%
rename from prepare/src/main/NamedEntity.scala
rename to prepare/src/main/guessNNprepare/NamedEntity.scala
index a03e2b0..23155aa 100644
--- a/prepare/src/main/NamedEntity.scala
+++ b/prepare/src/main/guessNNprepare/NamedEntity.scala
@@ -1,3 +1,5 @@
+package guessNNprepare
+
 import org.apache.jena.rdf.model.Resource
 
 class NamedEntity(val rdfResource: Resource, val Name: String, val readableType: String) {
diff --git a/prepare/src/main/Main.scala b/prepare/src/main/guessNNprepare/mains/ExtractEntities.scala
similarity index 71%
rename from prepare/src/main/Main.scala
rename to prepare/src/main/guessNNprepare/mains/ExtractEntities.scala
index c6a6a24..26e2633 100644
--- a/prepare/src/main/Main.scala
+++ b/prepare/src/main/guessNNprepare/mains/ExtractEntities.scala
@@ -1,11 +1,27 @@
+package guessNNprepare.mains
+
+import guessNNprepare.NamedEntity
+import net.sourceforge.argparse4j.impl.Arguments
+import net.sourceforge.argparse4j.inf.{ArgumentParser, Namespace}
 import org.apache.jena.query.QueryExecutionFactory
 import org.apache.jena.rdf.model.{InfModel, Model, ModelFactory}
 import org.apache.jena.reasoner.ReasonerRegistry
 
 import java.io.InputStream
 
-object Main extends App {
-	{
+object ExtractEntities extends MainCommand {
+
+	override def description: String = "Save the list of guessable entities to a json file"
+
+	override def addCliArgs(parser: ArgumentParser): Unit = {
+		parser.description(description)
+
+		parser.addArgument("json_file").`type`(Arguments.fileType().verifyCanWriteParent()).help("Where the guessable entities will be saved")
+	}
+
+	override def execute(ARGS: Namespace): Unit = {
+		val jsonFilePath = ARGS.getString("json_file")
+
 		println("Loading knowledge graph...")
 		val kg: InfModel = ModelFactory.createInfModel(ReasonerRegistry.getTransitiveReasoner, ModelFactory.createDefaultModel)
 		val mondialDataset: InputStream = Option(ClassLoader.getSystemResourceAsStream("mondial_2022_04_04.n3"))
@@ -23,6 +39,7 @@ object Main extends App {
 			getMondialNamedEntities(kg, "Sea"))
 			.distinct
 		printf("Found %d guessable named entities\n", guessableEntities.size)
+		println(s"They would be written to ${jsonFilePath} (not implemented yet)")
 	}
 
 	def getMondialNamedEntities(kg: Model, prefixedType: String): List[NamedEntity] = {
diff --git a/prepare/src/main/guessNNprepare/mains/MainCommand.scala b/prepare/src/main/guessNNprepare/mains/MainCommand.scala
new file mode 100644
index 0000000..175799b
--- /dev/null
+++ b/prepare/src/main/guessNNprepare/mains/MainCommand.scala
@@ -0,0 +1,13 @@
+package guessNNprepare.mains
+
+import net.sourceforge.argparse4j.inf.{ArgumentParser, Namespace}
+
+trait MainCommand {
+
+	def description: String
+
+	def addCliArgs(parser: ArgumentParser): Unit
+
+	def execute(ARGS: Namespace): Unit
+
+}