package com.windchat.im.boot.utils;

import java.io.BufferedReader;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintWriter;

import com.windchat.im.boot.config.ConfigHelper;
import com.windchat.im.boot.config.ConfigKey;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.DefaultParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.windchat.common.utils.PrintUtils;
import com.windchat.common.utils.StringHelper;
import com.windchat.im.storage.DataSourceManager;
import com.windchat.im.storage.dao.config.DBConfig;
import com.windchat.im.storage.dao.config.DBType;
import com.windchat.im.storage.dao.sql.SQLConst;
import com.windchat.im.storage.exception.MigrateDatabaseException;
import com.windchat.im.storage.exception.NeedInitMysqlException;
import com.windchat.im.storage.exception.UpgradeDatabaseException;

public class Helper {
	private static final Logger logger = LoggerFactory.getLogger(Helper.class);

	/**
	 * 
	 * @param args
	 * @return true:only print helper message false:start openzaly-server
	 */
	public static boolean startHelper(String[] args) {
		PrintWriter pw = null;
		try {
			Options options = new Options();
			options.addOption("h", false, "help message list");
			options.addOption("help", false, "help message list");
			options.addOption("team", false, "specify openzaly TEAM edition and init database config");
			options.addOption("upgrade", false, "upgrade openzaly server and database");
			options.addOption("migrate", false, "migrate data from sqlite to mysql");
			DefaultParser posixParser = new DefaultParser();
			CommandLine commandLine = posixParser.parse(options, args);

			if (commandLine.hasOption("h") || commandLine.hasOption("help")) {
				pw = new PrintWriter(System.out);
				// 1.print logo
				showAkaxinBanner(pw);
				// 2.print Userage
				HelpFormatter helpFormatter = new HelpFormatter();
				helpFormatter.printHelp("java [<name>=value] -jar openzaly-server.jar [-h|-help]", options, false);
				// 3.print helper message
				printHelperMessage(pw);
				return true;
			} else if (commandLine.hasOption("upgrade")) {
				pw = new PrintWriter(System.out);
				upgrade(pw);
				return true;
			} else if (commandLine.hasOption("team") || commandLine.hasOption("TEAM")
					|| commandLine.hasOption("Team")) {
				pw = new PrintWriter(System.out);
				initMysqlDatabase(pw);
				return true;
			} else if (commandLine.hasOption("migrate")) {
				migrateSqlite2Mysql();
				return true;
			}
			return false;
		} catch (ParseException e) {
			logger.error("print helper with startHelper error", e);
		} finally {
			if (pw != null) {
				pw.close();
			}
		}
		return true;
	}

	public static void showAkaxinBanner(PrintWriter pw) {
		BufferedReader buffer = null;
		try {
			InputStream inputStream = Helper.class.getResourceAsStream("/logo.txt");
			buffer = new BufferedReader(new InputStreamReader(inputStream));
			String line = null;
			while ((line = buffer.readLine()) != null) {
				pw.println(line);
			}
			pw.flush();
			inputStream.close();
		} catch (Exception e) {
			logger.error("read helper message from file error", e);
		} finally {
			// 关闭buffer
			try {
				if (buffer != null) {
					buffer.close();
				}
			} catch (IOException e) {
				logger.error("buffer close error with IOException");
			}
		}
	}

	public static void buildEnvToSystemOut(PrintWriter pwriter) {
		pwriter.println();
		// openzaly version
		String ozVersion = ConfigHelper.getStringConfig(ConfigKey.SITE_VERSION);
		pwriter.println("openzaly-version : " + ozVersion);
		// os
		String osName = System.getProperty("os.name");
		String dataModel = System.getProperty("sun.arch.data.model"); // 32位 or 64位
		pwriter.println("OS Name : " + osName + " " + dataModel + "位");
		// java version
		String javaVersion = System.getProperty("java.version"); // 获取JDK版本
		pwriter.println("java-version : " + javaVersion);

		pwriter.println();
		pwriter.println("[OK] openzaly-server is starting...");
		pwriter.flush();
	}

	public static void startSuccess(PrintWriter pwriter) {
		pwriter.println("[OK] start openzaly-server successfully");
		pwriter.flush();
	}

	public static void startFail(PrintWriter pwriter) {
		pwriter.println("[Error] start openzaly-server failed, server exit...");
		pwriter.println();
		pwriter.flush();
	}

	public static void startFailWithError(PrintWriter pwriter, String errMessage) {
		pwriter.println("[Error] error message:" + errMessage);
		pwriter.println("[Error] start openzaly-server failed, server exit...");
		pwriter.println();
		pwriter.flush();
	}

	public static void printUpgradeWarn(PrintWriter pwriter) {
		pwriter.println("[Error] openzaly-server is an old version, you can execute following command to upgrade:");
		pwriter.println();
		pwriter.println("\t java -jar openzaly-server.jar -upgrade");
		pwriter.println();
		pwriter.flush();
	}

	public static void printInitMysqlWarn(PrintWriter pwriter) {
		pwriter.println("[Error] openzaly-server need to init mysql first, you can execute following command:");
		pwriter.println();
		pwriter.println("\t java -jar openzaly-server.jar -init");
		pwriter.println();
		pwriter.flush();
	}

	private static void printHelperMessage(PrintWriter pw) {
		pw.println();
		pw.println("example:java -Dsite.port=2021 -jar openzaly-server.jar ");
		pw.println();
		pw.println("\t-Dsite.address \t\topenzaly Netty address default:0.0.0.0");
		pw.println("\t-Dsite.port \t\topenzaly Netty port default:2021");
		pw.println("\t-Dpluginapi.address \topenzaly PluginAddress default: 0.0.0.0");
		pw.println("\t-Dpluginapi.port \topenzaly PluginApi port default:8280");
		pw.println("\t-Dsite.admin.port \topenzaly admin port default:8288");
		pw.println("\t-Dsite.uic \topenzaly first uic for admin port default: \"000000\"");
		pw.println("\t-Dsite.baseDir \t\topenzaly openzaly-server root dir default:./");
		pw.println();
		pw.flush();
	}

	private static void upgrade(PrintWriter pw) {
		pw.println();
		pw.println("[INFO] start to upgrade openzaly-server...");
		try {
			String dbDir = ConfigHelper.getStringConfig(ConfigKey.SITE_BASE_DIR);
			String siteVersion = ConfigHelper.getStringConfig(ConfigKey.SITE_VERSION);
			DBConfig config = new DBConfig();
			config.setDbDir(dbDir);
			config.setDb(DBType.PERSONAL);
			// 升级
			int dbUserVersion = DataSourceManager.upgradeDB(config);
			int needVersion = SQLConst.SITE_DB_VERSION_11;
			pw.println("[INFO] upgrade openzaly-server version : " + siteVersion);
			if (needVersion == dbUserVersion) {
				pw.println("[OK] upgrade database user-version : " + dbUserVersion);
				pw.println("[OK] upgrade openzaly-server finish ...");
			} else {
				pw.println("[ERROR] upgrade database user-version : " + dbUserVersion);
				pw.println("[ERROR] upgrade openzaly-server fail ...");
			}
		} catch (UpgradeDatabaseException e) {
			pw.println("[ERROR] upgrade openzaly-server error");
			logger.error("upgrade database error", e);
		}
		pw.println();
		pw.flush();
	}

	private static void initMysqlDatabase(PrintWriter pw) {
		pw.println("[INFO] starting init mysql");
		try {
			DataSourceManager.initMysqlConfig();
			pw.println("[OK] init mysql finish ,please execute command to start openzaly-server");
			pw.println();
			pw.println("\t java -jar openzaly-server.jar");
		} catch (FileNotFoundException e) {
			logger.error("init mysql error,", e);
			pw.println(StringHelper.format("[ERROR] init mysql error:[{}]", e.getMessage()));
		} catch (IOException e) {
			logger.error("init mysql error,", e);
			pw.println(StringHelper.format("[ERROR] init mysql error:[{}]", e.getMessage()));
		}
		pw.println();
		pw.flush();
	}

	/**
	 * 把sqlite数据迁移mysql中
	 * 
	 */
	private static void migrateSqlite2Mysql() {
		PrintUtils.print("[INFO]openzaly is migrating sqlite to mysql...");
		try {
			DataSourceManager.migrateDB();
			PrintUtils.print("[OK] migrate sqlite to mysql finish");
		} catch (MigrateDatabaseException e) {
			PrintUtils.print("[ERROR] migrate sqlite to mysql error,msg={}", e.getMessage());
		} catch (NeedInitMysqlException e) {
			PrintUtils.print("[ERROR] please execute command before migrate database.");
			PrintUtils.print();
			PrintUtils.print("java -jar openzaly-server.jar -init");
			PrintUtils.print();
		}
		PrintUtils.flush();
	}
}