com.axelor.apps.erp.Main.java Source code

Introduction

Here is the source code for com.axelor.apps.erp.Main.java

Source

/**
 * Axelor Business Solutions
 *
 * Copyright (C) 2014 Axelor (<http://axelor.com>).
 *
 * This program is free software: you can redistribute it and/or  modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.axelor.apps.erp;

import com.axelor.app.AppModule;
import com.axelor.auth.AuthModule;
import com.axelor.data.Commander;
import com.axelor.data.Listener;
import com.axelor.data.csv.CSVImporter;
import com.axelor.db.JpaModule;
import com.axelor.db.Model;
import com.google.common.base.Strings;
import com.google.inject.AbstractModule;
import com.google.inject.Guice;
import com.google.inject.Injector;
import com.google.inject.name.Names;
import com.google.inject.util.Providers;

/**
 * Main importer class. 
 * Can be launched by the script axelor-data.sh
 * 
 * Use to import only CSV files.
 * 
 * @author axelor
 *
 */
public class Main {

    /**
     * Main method.
     * Can be launched by the script axelor-data.sh
     * 
     * @param args
     *       Arguments
     * 
     * @throws Exception
     */
    public static void main(String[] args) throws Exception {

        final Commander cmd = new Commander();
        try {
            if (args == null || args.length == 0)
                throw new Exception();
            cmd.parse(args);
            if (!cmd.getDataDir().isDirectory())
                throw new Exception("invalid data directory");
            if (!cmd.getConfig().isFile())
                throw new Exception("invalid config file");
        } catch (Exception e) {
            String message = e.getMessage();
            if (!Strings.isNullOrEmpty(message))
                System.err.println(e.getMessage());
            Commander.usage();
            return;
        }

        if (cmd.getShowHelp() == Boolean.TRUE) {
            Commander.usage();
            return;
        }

        final String errorDir = cmd.getErrorDir() == null ? null : cmd.getErrorDir().getPath();

        Injector injector = Guice.createInjector(new AbstractModule() {

            @Override
            protected void configure() {
                install(new JpaModule("persistenceUnit", true, true));
                install(new AuthModule.Simple());
                install(new AppModule());
                bindConstant().annotatedWith(Names.named("axelor.data.config")).to(cmd.getConfig().toString());
                bindConstant().annotatedWith(Names.named("axelor.data.dir")).to(cmd.getDataDir().toString());
                bind(String.class).annotatedWith(Names.named("axelor.error.dir"))
                        .toProvider(Providers.<String>of(errorDir));
            }
        });

        CSVImporter importer = injector.getInstance(CSVImporter.class);

        importer.addListener(new Listener() {

            @Override
            public void imported(Model bean) {
            }

            @Override
            public void imported(Integer total, Integer success) {
                System.out.println("/*** Records total : " + total);
                System.out.println("/*** Records success : " + success);
            }

            @Override
            public void handle(Model bean, Exception e) {

            }
        });

        importer.run(null);
    }

}