Application: gvSIG desktop

package org.gvsig.sqlite.dal;

import java.io.File;

import java.net.URL;

import java.util.ArrayList;

import java.util.List;

import org.apache.commons.io.FileUtils;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DatabaseWorkspaceManager;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.impl.DefaultFeature;

import org.gvsig.fmap.dal.feature.spi.FeatureProvider;

import org.gvsig.fmap.dal.store.jdbc2.JDBCHelper;

import org.gvsig.fmap.dal.store.jdbc2.JDBCServerExplorer;

import org.gvsig.fmap.dal.store.jdbc2.spi.FakeConnectionProvider;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestUtils  {

    public static final Logger LOGGER = LoggerFactory.getLogger(TestUtils.class);

    public static final String PROVIDER_NAME = SQLiteLibrary.NAME;

    public static final String BASE_RESOURCES_FOLDER = "/org/gvsig/sqlite/dal/";

    private static int dbcounter = 1;

    public static SQLiteConnectionParameters buildDBConnection(String dbname) throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        SQLiteConnectionParameters conn = (SQLiteConnectionParameters) 

                dataManager.createServerExplorerParameters(PROVIDER_NAME);

        File dbfile = TestUtils.getResource(String.format(

                "test-dbs/%s-%d-%03d",

                dbname,

                System.currentTimeMillis(), 

                dbcounter++

            )

        );

//        System.out.println("#### dbfile: " + dbfile.getAbsolutePath());

        FileUtils.forceMkdir(dbfile.getParentFile());

        conn.setFile(dbfile);

        return conn;

    }

    public static JDBCServerExplorer openServerExplorer(String dbname) throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        SQLiteConnectionParameters conn = buildDBConnection(dbname);

        JDBCServerExplorer explorer = (JDBCServerExplorer) dataManager.openServerExplorer(

                PROVIDER_NAME, conn

        );

        return explorer;

    }

    public static File getTargetFolder() throws Exception {

        URL url = TestUtils.class.getResource("/");

        File x = new File(url.toURI());

        File target = x.getParentFile();

        return target;

    }

    public static File getResource(String name) throws Exception {

        File x = new File(getTargetFolder(), name);

        return x;

    }

    public static File getResourceAsFile(String pathname) throws Exception {

        URL url = TestUtils.class.getResource(pathname);

        File x = new File(url.toURI());

        return x;

    }

    public static void removeDALResource(String dbname, String name) throws Exception {

        SQLiteConnectionParameters connection = buildDBConnection(dbname);

        DatabaseWorkspaceManager workspace = DALLocator.getDataManager().createDatabaseWorkspaceManager(

                (DataServerExplorerParameters) connection

        );

//        ResourcesStorage resources = workspace.getResourcesStorage(name);

//        resources.remove("dal");

    }

    public static FeatureStore openCountriesStore() throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        File f = getResourceAsFile(BASE_RESOURCES_FOLDER+"countries.csv");

        FeatureStore store = (FeatureStore) dataManager.openStore(

                DataStore.CSV_PROVIDER_NAME, 

                "file=",f,

                "automaticTypesDetection=", false,

                "locale=","en"

        );

        return store;

    }

    public static FeatureStore openSourceStore1() throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        File f = getResourceAsFile(BASE_RESOURCES_FOLDER+"testCreateSource1.csv");

        FeatureStore store = (FeatureStore) dataManager.openStore(

                DataStore.CSV_PROVIDER_NAME, 

                "file=",f,

                "automaticTypesDetection=", false,

                "locale=","en"

        );

        return store;

    }

    public static FeatureStore openStoreTest(String baseFile) throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        File f = getResourceAsFile(BASE_RESOURCES_FOLDER+baseFile);

        FeatureStore store = (FeatureStore) dataManager.openStore(

                DataStore.CSV_PROVIDER_NAME, 

                "file=",f,

                "automaticTypesDetection=", false,

                "locale=","en"

        );

                dataManager.getStoresRepository().add(store.getName(), store.getParameters());

        return store;

    }

    public static FeatureStore openSourceStore2() throws Exception {

        DataManager dataManager = DALLocator.getDataManager();

        File f = getResourceAsFile(BASE_RESOURCES_FOLDER+"testCreateSource2.csv");

        FeatureStore store = (FeatureStore) dataManager.openStore(

                DataStore.CSV_PROVIDER_NAME, 

                "file=",f,

                "automaticTypesDetection=", false,

                "locale=","en"

        );

        return store;

    }

    public static List<String> getSQLs(String name) throws Exception {

      File f = getResourceAsFile(BASE_RESOURCES_FOLDER+name);

      List<String> SQLs = new ArrayList<>();

      List<String> lines = FileUtils.readLines(f);

      StringBuilder sb = new StringBuilder();

      for (String line : lines) {

        line = StringUtils.stripStart(line, null);

        if( line.startsWith("--") ) {

          continue;

        }

        if( line.endsWith(";") ) {

          sb.append(line.substring(0, line.length()-1));

          SQLs.add(sb.toString());

          sb.setLength(0);

        } else {

          sb.append(line);

        }

      }

      return SQLs;

    }

    public static FeatureProvider getFeatureProvider(Feature feature) {

      return ((DefaultFeature)feature).getData();

    }

    public static SQLiteHelper createJDBCHelper() throws Exception {

      SQLiteConnectionParameters params = buildDBConnection("fake");

      SQLiteHelper helper = new SQLiteHelper(params, new FakeConnectionProvider());

      return helper;

    }

}