feat: include database configuration

Signed-off-by: Otavio Santana <[email protected]>

Author: otaviojava

src/main/java/org/soujava/demos/arangodb/graph/ArangoDBGraphConfiguration.java

@@ -0,0 +1,97 @@
+package org.soujava.demos.arangodb.graph;
+
+import com.arangodb.tinkerpop.gremlin.utils.ArangoDBConfigurationBuilder;
+import org.apache.commons.configuration.BaseConfiguration;
+import org.eclipse.jnosql.communication.Settings;
+import org.apache.tinkerpop.gremlin.structure.Graph;
+import org.apache.tinkerpop.gremlin.structure.util.GraphFactory;
+import org.eclipse.jnosql.databases.tinkerpop.communication.GraphConfiguration;
+
+import java.util.Objects;
+
+import static org.soujava.demos.arangodb.graph.ArangoDBGraphConfigurations.EDGE;
+import static org.soujava.demos.arangodb.graph.ArangoDBGraphConfigurations.EDGE_RELATIONSHIP;
+import static org.soujava.demos.arangodb.graph.ArangoDBGraphConfigurations.GRAPH;
+import static org.soujava.demos.arangodb.graph.ArangoDBGraphConfigurations.HOST;
+import static org.soujava.demos.arangodb.graph.ArangoDBGraphConfigurations.PASSWORD;
+import static org.soujava.demos.arangodb.graph.ArangoDBGraphConfigurations.USER;
+import static org.soujava.demos.arangodb.graph.ArangoDBGraphConfigurations.VERTEX;
+
+public class ArangoDBGraphConfiguration implements GraphConfiguration {
+
+
+    @Override
+    public Graph apply(Settings settings) {
+        Objects.requireNonNull(settings, "settings is required");
+        ArangoDBConfigurationBuilder builder = new ArangoDBConfigurationBuilder();
+
+        settings.prefix(HOST)
+                .stream()
+                .map(Object::toString)
+                .forEach(builder::arangoHosts);
+
+        settings.prefix(VERTEX)
+                .stream()
+                .map(Object::toString)
+                .forEach(builder::withVertexCollection);
+
+        settings.prefix(EDGE)
+                .stream()
+                .map(Object::toString)
+                .forEach(builder::withEdgeCollection);
+
+
+        settings.get(USER)
+                .map(Object::toString)
+                .ifPresent(builder::arangoUser);
+
+        settings.get(PASSWORD)
+                .map(Object::toString)
+                .ifPresent(builder::arangoPassword);
+
+        settings.get(GRAPH)
+                .map(Object::toString)
+                .ifPresent(builder::graph);
+
+        settings.prefix(EDGE_RELATIONSHIP)
+                .stream()
+                .map(EdgeConfiguration::parse)
+                .forEach(e -> e.add(builder));
+        BaseConfiguration configuration = builder.build();
+
+        var conf2 = new org.apache.commons.configuration2.BaseConfiguration();
+        configuration.getKeys().forEachRemaining(k -> {
+            conf2.addProperty(k, configuration.getProperty(k));
+        });
+        return GraphFactory.open(conf2);
+    }
+
+
+    private static class EdgeConfiguration {
+
+        private final String edge;
+
+        private final String source;
+
+        private final String target;
+
+        private EdgeConfiguration(String source, String edge, String target) {
+            this.edge = edge;
+            this.source = source;
+            this.target = target;
+        }
+
+        static EdgeConfiguration parse(Object value) {
+            final String[] values = value.toString().split("\\|");
+            if (values.length != 3) {
+                throw new IllegalArgumentException("The element is valid it must have" +
+                        " three element split by pipe: " + value);
+            }
+            return new EdgeConfiguration(values[0], values[1], values[2]);
+        }
+
+        private void add(ArangoDBConfigurationBuilder builder) {
+            builder.configureEdge(edge, source, target);
+        }
+    }
+}

src/main/java/org/soujava/demos/arangodb/graph/ArangoDBGraphConfigurations.java

@@ -0,0 +1,48 @@
+package org.soujava.demos.arangodb.graph;
+
+import java.util.function.Supplier;
+
+public enum ArangoDBGraphConfigurations implements Supplier<String> {
+
+    /**
+     * The edge collection. It uses as a prefix. E.g.:jnosql.arangodb.graph.edge.1=edge
+     */
+    EDGE("jnosql.arangodb.graph.edge"),
+    /**
+     * Edge collection, the source vertex collection and the target vertex collection split by pipe.
+     * It hou,It uses as a prefix.
+     * E.g.: jnosql.arangodb.graph.relationship.1=Person|knows|Person
+     */
+    EDGE_RELATIONSHIP("jnosql.arangodb.graph.relationship"),
+    /**
+     * The vertex collection. It uses as a prefix. E.g.: jnosql.arangodb.graph.vertex.1=vertex
+     */
+    VERTEX("jnosql.arangodb.graph.vertex"),
+    /**
+     * Name of the graph to use.
+     */
+    GRAPH("jnosql.arangodb.graph.graph"),
+    /**
+     * The database host.
+     */
+    HOST("jnosql.arangodb.graph.host"),
+    /**
+     * The user's credential.
+     */
+    USER("jnosql.arangodb.graph.user"),
+    /**
+     * The password's credential.
+     */
+    PASSWORD("jnosql.arangodb.graph.password");
+
+    private final String value;
+
+    ArangoDBGraphConfigurations(String value) {
+        this.value = value;
+    }
+
+    @Override
+    public String get() {
+        return value;
+    }
+}