armenak / DataDefender

 * Copyright 2014, Armenak Grigoryan, and individual contributors as indicated

 * by the @authors tag. See the copyright.txt in the distribution for a

 * full listing of individual contributors.

 *

 * This is free software; you can redistribute it and/or modify it

 * under the terms of the GNU Lesser General Public License as

 * published by the Free Software Foundation; either version 2.1 of

 * the License, or (at your option) any later version.

 *

 * This software 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

 * Lesser General Public License for more details.

 */

package com.strider.datadefender;

import java.util.List;

import java.util.Map;

import java.util.regex.Matcher;

import java.util.regex.Pattern;

import picocli.CommandLine.Option;

import lombok.Getter;

/**

 * Database configuration options for picocli.

 * 

 * @author Zaahid Bateson

 */

@Getter

public class DbConfig {

    public enum Vendor {

        H2, MYSQL, POSTGRESQL, SQLSERVER, ORACLE;

    }

    private static final Map<String, Vendor> VENDOR_MAP = Map.of(

        "h2", Vendor.H2,

        "mysql", Vendor.MYSQL,

        "mariadb", Vendor.MYSQL,

        "postgresql", Vendor.POSTGRESQL,

        "sqlserver", Vendor.SQLSERVER,

        "mssql", Vendor.SQLSERVER,

        "oracle", Vendor.ORACLE

    );

    private Vendor vendor;

    @Option(names = { "-u", "--user" }, description = "The username to connect with")

    private String username;

    @Option(names = { "-p", "--password" }, description = "The password to connect with", arity = "0..1", interactive = true)

    private String password;

    @Option(names = { "--schema" }, description = "The schema to connect to")

    private String schema;

    @Option(names = { "--no-skip-empty-tables-metadata" }, description = "Include generating metadata for empty tables (defaults to skipping)", defaultValue = "true", negatable = true)

    private boolean skipEmptyTables = true;

    @Option(names = { "--include-table-pattern-metadata" }, description = "Pattern(s) matching table names to include for metadata analysis")

    private List<Pattern> includeTablePatterns;

    @Option(names = { "--exclude-table-pattern-metadata" }, description = "Pattern(s) matching table names to exclude for metadata analysis")

    private List<Pattern> excludeTablePatterns;

    private String url;

    @Option(names = { "--url" }, description = "The datasource URL")

    public void setUrl(String url) {

        Pattern p = Pattern.compile("\\s*jdbc:([^:]+):.*");

        if (vendor == null) {

            Matcher m = p.matcher(url);

            if (m.find()) {

                setVendor(m.group(1));

            }

        }

        this.url = url;

    }

}