aboutsummaryrefslogtreecommitdiffstats
path: root/src/main/java/org/onap/aai/graphgraph/App.java
blob: 1c6bc61b4a40b98a21e3de7a1d0df8cf7035db2d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
/*
  ============LICENSE_START=======================================================
  org.onap.aai
  ================================================================================
  Copyright © 2019-2020 Orange Intellectual Property. All rights reserved.
  ================================================================================
  Licensed under the Apache License, Version 2.0 (the "License");
  you may not use this file except in compliance with the License.
  You may obtain a copy of the License at

     http://www.apache.org/licenses/LICENSE-2.0

  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.
  ============LICENSE_END=========================================================
 */
package org.onap.aai.graphgraph;

import static org.onap.aai.graphgraph.ModelExporter.exportModel;
import static org.onap.aai.graphgraph.ModelExporter.writeExportedModel;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import org.apache.commons.io.FileUtils;
import org.onap.aai.edges.EdgeIngestor;
import org.onap.aai.introspection.MoxyLoader;
import org.onap.aai.nodes.NodeIngestor;
import org.onap.aai.restclient.PropertyPasswordConfiguration;
import org.onap.aai.setup.SchemaVersion;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.ConfigurableApplicationContext;

@SpringBootApplication
public class App {

    public static EdgeIngestor edgeIngestor;
    public static Map<String, MoxyLoader> moxyLoaders = new HashMap<>();

    // TODO
    // this should be used properly within Spring as this is a 'static' workaround due
    // to some initialization issues. By all means feel free to improve and move it to Spring
    public static void loadSchemes(ConfigurableApplicationContext context) {
        String version;
        for (int i = 10; i < 19; i++) {
            version = "v" + i;
            moxyLoaders.put(version, new MoxyLoader(
                    new SchemaVersion(version), (NodeIngestor) context.getBean("nodeIngestor"))
            );
        }
    }

    public static void main(String[] args) throws IOException {
        ArgumentParser parser = new ArgumentParser().parseArguments(args);

        if (parser.isPrintHelp()) {
            parser.printHelp();
            return;
        }

        SpringApplication app = new SpringApplication(App.class);

        if (parser.isRunLocally()) {
            copyKeystore(app);
        }

        app.addInitializers(new PropertyPasswordConfiguration());
        ConfigurableApplicationContext context = app.run(args);
        loadSchemes(context);
        edgeIngestor = (EdgeIngestor) context.getBean("edgeIngestor");

        if (parser.shouldGenerateUrl()) {
            writeExportedModel(exportModel(parser.getSchemaVersion()));
            System.exit(0);
        }
    }

    private static void copyKeystore(SpringApplication app) throws IOException {
        Path path = Paths.get("etc/auth/aai_keystore");
        if (Files.notExists(path)) {
            FileUtils.copyInputStreamToFile(
                    Objects.requireNonNull(app.getClassLoader().getResourceAsStream("etc/auth/aai_keystore")),
                    path.toFile()
            );
        }
    }
}