com.qiniu.entry.EntryMain Maven / Gradle / Ivy

Go to download
package com.qiniu.entry;

import com.qiniu.config.ParamsConfig;
import com.qiniu.config.PropertiesFile;
import com.qiniu.interfaces.IDataSource;
import com.qiniu.datasource.InputSource;
import com.qiniu.interfaces.IEntryParam;
import com.qiniu.interfaces.ILineProcess;
import com.qiniu.util.*;

import java.io.File;
import java.io.IOException;
import java.util.*;

public class EntryMain {

    // 如果不希望对危险 process 进行 verify，请将该参数设置为 false
    public static boolean processVerify = true;

    @SuppressWarnings("unchecked")
    public static void main(String[] args) throws Exception {
        Map preSetMap = new HashMap(){{
            put("f", "verify=false");
            put("s", "single=true");
            put("single", "single=true");
            put("line", "single=true");
            put("i", "interactive=true");
            put("interactive", "interactive=true");
            put("d", "default=true"); // for default account setting
            put("getaccount", "getaccount=default"); // for default account query
            put("dis", "displayed=false"); // get default account without displayed secret
            put("key=", ""); // get default account without displayed secret
            put("to-key=", ""); // get default account without displayed secret
        }};
        Map paramsMap = getEntryParams(args, preSetMap);
        IEntryParam entryParam = new ParamsConfig(paramsMap);
        if (paramsMap.containsKey("account")) {
            String accountName = paramsMap.get("account");
            if (!accountName.contains("-")) accountName = String.join("-", "qiniu", accountName);
            AccountUtils.setAccount(entryParam, accountName);
            List keysList = AccountUtils.getAccount(accountName, true);
            for (String[] keys : keysList) {
                System.out.println(keys[2] + ": ");
                System.out.println("id: " + keys[0]);
                System.out.println("secret: " + keys[1]);
            }
            return;
        } else if (paramsMap.containsKey("getaccount")) {
            boolean implicit = !paramsMap.containsKey("displayed");
            String accountName = paramsMap.get("getaccount");
            if ("default".equals(accountName)) {
                if (entryParam.getValue("default", "false").equals("true")) accountName = null;
                else throw new IOException("please set account name or use \"-d\"");
            }
            List keysList = AccountUtils.getAccount(accountName, implicit);
            for (String[] keys : keysList) {
                System.out.println(keys[2] + ": ");
                System.out.println("id: " + keys[0]);
                System.out.println("secret: " + keys[1]);
            }
            return;
        } else if (paramsMap.containsKey("delaccount")) {
            AccountUtils.deleteAccount(paramsMap.get("delaccount"));
            return;
        } else if (paramsMap.containsKey("domainsfrom")) {
            paramsMap.put("single", "true");
            paramsMap.put("process", "domainsofbucket");
            paramsMap.put("bucket", paramsMap.get("domainsfrom"));
        }
        if (paramsMap.containsKey("verify")) processVerify = Boolean.parseBoolean(paramsMap.get("verify"));
        boolean single = paramsMap.containsKey("single") && Boolean.parseBoolean(paramsMap.get("single"));
        if (!single && paramsMap.containsKey("key") && !paramsMap.containsKey("path")) single = true;
        boolean interactive = paramsMap.containsKey("interactive") && Boolean.parseBoolean(paramsMap.get("interactive"));
        CommonParams commonParams = single ? new CommonParams(paramsMap) : new CommonParams(entryParam);
        QSuitsEntry qSuitsEntry = new QSuitsEntry(entryParam, commonParams);
        ILineProcess> processor = single || interactive ? qSuitsEntry.whichNextProcessor(true) :
                qSuitsEntry.getProcessor();
        if (processVerify && processor != null) {
            String process = processor.getProcessName();
            if (processor.getNextProcessor() != null) process = processor.getNextProcessor().getProcessName();
            if (ProcessUtils.isDangerous(process)) {
                if (single) {
                    System.out.println("your process is " + process + ", are you sure? (y/n): ");
                } else {
                    String path = paramsMap.get("path");
                    if (path == null || "".equals(path)) path = "qiniu://" + paramsMap.get("bucket");
                    System.out.println("your last process is " + process + " for " + path + ", are you sure? (y/n): ");
                }
                Scanner scanner = new Scanner(System.in);
                String an = scanner.next();
                if (!an.equalsIgnoreCase("y") && !an.equalsIgnoreCase("yes")) {
                    return;
                }
            }
        }
        if (single) {
            if (processor != null) {
                Map converted = commonParams.getMapLine();
                if ("qupload".equals(processor.getProcessName())) {
                    if (converted.containsKey("filepath")) {
                        converted.put("filepath", FileUtils.convertToRealPath(converted.get("filepath")));
                    } else {
                        converted.put("filepath", FileUtils.convertToRealPath(converted.get("key")));
                    }
                }
                if (paramsMap.containsKey("domainsfrom")) {
                    System.out.println(processor.processLine(converted).replace(",", "\n"));
                } else {
                    System.out.println(processor.processLine(converted));
                }
            }
        } else if (interactive) {
            InputSource inputSource = qSuitsEntry.getInputSource();
            inputSource.export(System.in, processor);
        } else {
            IDataSource dataSource = qSuitsEntry.getDataSource();
            if (dataSource != null) {
                dataSource.setProcessor(processor);
//                dataSource.export();
                dataSource.export(commonParams.getStartDateTime(), commonParams.getPauseDelay(), commonParams.getPauseDuration());
//                dataSource.export(LocalDateTime.now().plusSeconds(5), 0, 19);
            }
        }
        if (processor != null) processor.closeResource();
    }

    public static Map getEntryParams(String[] args, Map preSetMap) throws IOException {
        Map paramsMap = args != null && args.length > 0 ? ParamsUtils.toParamsMap(args, preSetMap) : null;
        if (paramsMap != null && paramsMap.containsKey("config")) {
            Map fileConfig = ParamsUtils.toParamsMap(paramsMap.get("config"));
            fileConfig.putAll(paramsMap); // 使用命令行的参数覆盖配置文件中可能存在的同个配置
            return fileConfig;
        } else {
            String configFilePath = null;
            List configFiles = new ArrayList(){{
                add("resources" + System.getProperty("file.separator") + "application.config");
                add("resources" + System.getProperty("file.separator") + "application.properties");
                add("resources" + System.getProperty("file.separator") + ".application.config");
                add("resources" + System.getProperty("file.separator") + ".application.properties");
            }};
            for (int i = configFiles.size() - 1; i >= 0; i--) {
                File file = new File(configFiles.get(i));
                if (file.exists()) {
                    configFilePath = configFiles.get(i);
                    System.out.printf("use default config file: %s\n", configFilePath);
                    break;
                }
            }
            Map fileConfig;
            if (configFilePath == null) {
                if (paramsMap == null) throw new IOException("there is no config file detected.");
                else return paramsMap;
            } else if (configFilePath.endsWith(".properties")) {
                fileConfig = ParamsUtils.toParamsMap(new PropertiesFile(configFilePath).getProperties());
            } else {
                fileConfig = ParamsUtils.toParamsMap(configFilePath);
            }
            if (paramsMap == null) {
                return fileConfig;
            } else {
                fileConfig.putAll(paramsMap);
                return fileConfig;
            }
        }
//        if (args != null && args.length > 0) {
//            Map paramsMap = ParamsUtils.toParamsMap(args, preSetMap);
//            if (paramsMap.containsKey("config")) {
//                return ParamsUtils.toParamsMap(paramsMap.get("config"));
//            } else {
//                return paramsMap;
//            }
//        } else {
//            String configFilePath = null;
//            List configFiles = new ArrayList(){{
//                add("resources" + System.getProperty("file.separator") + "application.config");
//                add("resources" + System.getProperty("file.separator") + ".application.config");
//                add("resources" + System.getProperty("file.separator") + ".application.properties");
//            }};
//            for (int i = configFiles.size() - 1; i >= 0; i--) {
//                File file = new File(configFiles.get(i));
//                if (file.exists()) {
//                    configFilePath = configFiles.get(i);
//                    System.out.printf("use default config file: %s\n", configFilePath);
//                    break;
//                }
//            }
//            if (configFilePath == null) {
//                throw new IOException("there is no config file detected.");
//            } else if (configFilePath.endsWith(".properties")) {
//                return ParamsUtils.toParamsMap(new PropertiesFile(configFilePath).getProperties());
//            } else {
//                return ParamsUtils.toParamsMap(configFilePath);
//            }
//        }
    }
}