@echo off
REM
REM Licensed to the Apache Software Foundation (ASF) under one
REM or more contributor license agreements.  See the NOTICE file
REM distributed with this work for additional information
REM regarding copyright ownership.  The ASF licenses this file
REM to you under the Apache License, Version 2.0 (the
REM "License"); you may not use this file except in compliance
REM with the License.  You may obtain a copy of the License at
REM
REM     http://www.apache.org/licenses/LICENSE-2.0
REM
REM Unless required by applicable law or agreed to in writing,
REM software distributed under the License is distributed on an
REM "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
REM KIND, either express or implied.  See the License for the
REM specific language governing permissions and limitations
REM under the License.
REM

REM ==========================================================
REM Apache IoTDB CLI Launcher
REM ==========================================================

setlocal enabledelayedexpansion

REM -------------------------------
REM Default SQL dialect
if "%DEFAULT_SQL_DIALECT%"=="" set "DEFAULT_SQL_DIALECT=tree"

REM Default connection parameters
set "user_param=-u root"
set "passwd_param=-pw root"
set "host_param=-h 127.0.0.1"
set "port_param=-p 6667"
set "sql_dialect_param=-sql_dialect %DEFAULT_SQL_DIALECT%"
set "PARAMETERS="

REM -------------------------------
REM Normalize script path
pushd "%~dp0" >nul 2>&1
set "SCRIPT_DIR=%CD%"
popd >nul 2>&1

REM -------------------------------
REM Set IOTDB_HOME
if not defined IOTDB_HOME (
    for %%I in ("%~dp0..\..") do set "IOTDB_HOME=%%~fI"
)

REM -------------------------------
REM Parse command-line arguments
:parse_args
if "%~1"=="" goto after_parse

if /I "%~1"=="-u" (
    set "user_param=-u %~2"
    shift
    shift
    goto parse_args
)
if /I "%~1"=="-pw" (
    call :is_known_option "%~2"
    if "%~2"=="" (
        set "passwd_param=-pw"
        shift
    ) else if "!IS_KNOWN_OPTION!"=="true" (
        set "passwd_param=-pw"
        shift
    ) else (
        set "passwd_param=-pw %~2"
        shift
        shift
    )
    goto parse_args
)
if /I "%~1"=="-h" (
    set "host_param=-h %~2"
    shift
    shift
    goto parse_args
)
if /I "%~1"=="-p" (
    set "port_param=-p %~2"
    shift
    shift
    goto parse_args
)
if /I "%~1"=="-sql_dialect" (
    set "sql_dialect_param=-sql_dialect %~2"
    shift
    shift
    goto parse_args
)

set "PARAMETERS=%PARAMETERS% %~1"
shift
goto parse_args

:is_known_option
set "IS_KNOWN_OPTION=false"
if /I "%~1"=="-h" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-p" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-u" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-pw" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-e" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-help" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-sql_dialect" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-disableISO8601" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-c" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-timeout" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-usessl" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-ts" set "IS_KNOWN_OPTION=true"
if /I "%~1"=="-tpw" set "IS_KNOWN_OPTION=true"
exit /b 0

:after_parse
REM Combine all parameters
set "PARAMETERS=%host_param% %port_param% %user_param% %passwd_param% %sql_dialect_param% %PARAMETERS%"

REM CLI configuration
set "IOTDB_CLI_CONF=%IOTDB_HOME%\conf"
set "MAIN_CLASS=org.apache.iotdb.cli.Cli"

REM CLASSPATH setup
set "CLASSPATH=%IOTDB_HOME%\lib\*"

REM JAVA executable
if defined JAVA_HOME (
    if exist "%JAVA_HOME%\bin\java.exe" (
        set "JAVA=%JAVA_HOME%\bin\java.exe"
    ) else (
        set "JAVA=java"
    )
) else (
    set "JAVA=java"
)

REM Detect Java major version (--add-opens only for Java 9+, see JEP 396)
set "JAVA_MAJOR="
set "TEMP_VER=%TEMP%\iotdb_cli_java_ver.txt"
"%JAVA%" -version 2>&1 | findstr /i "version" > "%TEMP_VER%"
for /f "tokens=3" %%a in ('type "%TEMP_VER%"') do for /f "tokens=1,2 delims=." %%x in ("%%~a") do (
    if "%%x"=="1" (set "JAVA_MAJOR=%%y") else (set "JAVA_MAJOR=%%x")
    goto :jvm_opts
)

:jvm_opts
del "%TEMP_VER%" 2>nul
REM JVM options: base options always; --add-opens only for Java 9 and above
set "IOTDB_CLI_PARAMS=-Dlogback.configurationFile=%IOTDB_CLI_CONF%\logback-cli.xml"
set "JVM_OPTS=-Dsun.jnu.encoding=UTF-8 -Dfile.encoding=UTF-8"
if not "%JAVA_MAJOR%"=="8" if not "%JAVA_MAJOR%"=="" (
    set "JVM_OPTS=!JVM_OPTS! --add-opens=java.base/java.lang=ALL-UNNAMED"
)

REM Run CLI
"%JAVA%" %JVM_OPTS% %IOTDB_CLI_PARAMS% -cp "%CLASSPATH%" %MAIN_CLASS% %PARAMETERS%

REM -------------------------------
endlocal

exit /b %ERRORLEVEL%