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

param(
[Parameter(Mandatory=$true,Position = 0)] [string] $Command="",
[string] $D="",
[switch] $Secure,
[Parameter(ValueFromRemainingArguments = $true)][string[]]$args)

# FUNCTIONS

Function PrintUsage { param ()
    Write-Output  "Usage  : oozie-setup.ps1 COMMAND [OPTIONS]"
    Write-Output  "          prepare-war [-d directory] [-secure] (-d identifies an alternative directory for processing jars"
    Write-Output  "                                                -secure will configure the war file to use HTTPS (SSL))"
    Write-Output  "          sharelib create -fs FS_URI [-locallib SHARED_LIBRARY] [-concurrency CONCURRENCY]"
    Write-Output  "                                                                (create sharelib for oozie,"
    Write-Output  "                                                                FS_URI is the fs.default.name"
    Write-Output  "                                                                for hdfs uri; SHARED_LIBRARY, path to the"
    Write-Output  "                                                                Oozie sharelib to install, it can be a tarball"
    Write-Output  "                                                                or an expanded version of it. If ommited,"
    Write-Output  "                                                                the Oozie sharelib tarball from the Oozie"
    Write-Output  "                                                                installation directory will be used."
    Write-Output  "                                                                CONCURRENCY is a number of threads to be used"
    Write-Output  "                                                                for copy operations."
    Write-Output  "                                                                By default 1 thread will be used)"
    Write-Output  "                                                                (action failes if sharelib is already installed"
    Write-Output  "                                                                in HDFS)"
    Write-Output  "          sharelib upgrade -fs FS_URI [-locallib SHARED_LIBRARY] ([deprecated]"
    Write-Output  "                                                                  [use create command to create new version]"
    Write-Output  "                                                                  upgrade existing sharelib, fails if there"
    Write-Output  "                                                                  is no existing sharelib installed in HDFS)"
    Write-Output  "          db create|upgrade|postupgrade -run [-sqlfile <FILE>] (create, upgrade or postupgrade oozie db with an"
    Write-Output  "                                                                optional sql File)"
    Write-Output  "         EXTJS can be downloaded from http://www.extjs.com/learn/Ext_Version_Archives"
}

function Expand-ZIPFile($file, $destination){
    $shell = new-object -com shell.application
    $zip = $shell.NameSpace($file)
    foreach($item in $zip.items()){
        $shell.Namespace($destination).copyhere($item)
    }
}

# MAIN()

# The script will terminate if any steps fail
$ErrorActionPreference = "Stop"

# Constants
$EXT_SUBDIR = "ext-2.2"
$OOZIE_HOME = (Split-Path $MyInvocation.MyCommand.Path) + "\.."
$OOZIE_HOME = Resolve-Path $OOZIE_HOME

$CATALINA_BASE = ""
if ($env:CATALINA_BASE){
    $CATALINA_BASE = "$env:CATALINA_BASE"
}else{
    $CATALINA_BASE = "$OOZIE_HOME\oozie-server"
}

# Finds JAR.EXE and Java
$JAR_EXE=""
$JAVA_BIN=""
if ($env:JAVA_HOME) {
    $JAR_EXE = "$env:JAVA_HOME\bin\jar.exe"
    $JAVA_BIN = "$env:JAVA_HOME\bin\java.exe"
} else {
    Write-Output "WARN: JAVA_HOME not defined. oozie-setup.ps1 will relay on the PATH environment variable to use JAR.exe"
    $JAR_EXE = "jar.exe"
    $JAVA_BIN = "java.exe"
}

if (($Command -eq "sharelib") -Or ($Command -eq "db") -Or ($Command -eq "import") -Or ($Command -eq "export")) {
          $OOZIE_OPTS="-Doozie.home.dir=$OOZIE_HOME";
          $OOZIE_OPTS="$OOZIE_OPTS -Doozie.config.dir=$OOZIE_HOME\conf";
          $OOZIE_OPTS="$OOZIE_OPTS -Doozie.log.dir=$OOZIE_HOME\log";
          $OOZIE_OPTS="$OOZIE_OPTS -Doozie.data.dir=$OOZIE_HOME\data";
          $OOZIE_OPTS="$OOZIE_OPTS -Dderby.stream.error.file=$OOZIE_HOME\log\derby.log"

          $OOZIECPPATH=""
          $OOZIECPPATH="$OOZIE_HOME\libtools\*;$OOZIE_HOME\lib\*;$OOZIE_HOME\..\extra_libs\*"

          $COMMAND_OPTS=[string]$args

          if ($Command -eq "sharelib") {
            Start-Process $JAVA_BIN -ArgumentList "$OOZIE_OPTS -cp $OOZIECPPATH org.apache.oozie.tools.OozieSharelibCLI $COMMAND_OPTS" -Wait -NoNewWindow
          } elseif ($Command -eq "db") {
            Start-Process $JAVA_BIN -ArgumentList "$OOZIE_OPTS -cp $OOZIECPPATH org.apache.oozie.tools.OozieDBCLI $COMMAND_OPTS" -Wait -NoNewWindow
          } elseif ($Command -eq "export"){
            Start-Process $JAVA_BIN -ArgumentList "$OOZIE_OPTS -cp $OOZIECPPATH org.apache.oozie.tools.OozieDBExportCLI $COMMAND_OPTS" -Wait -NoNewWindow
          } elseif ($Command -eq "import"){
            Start-Process $JAVA_BIN -ArgumentList "$OOZIE_OPTS -cp $OOZIECPPATH org.apache.oozie.tools.OozieDBImportCLI $COMMAND_OPTS" -Wait -NoNewWindow
          }
          exit 0
}elseif ($Command -eq "prepare-war"){

    $InputWar = "$OOZIE_HOME\oozie.war"
    $OutputWar = "$OOZIE_HOME\oozie-server\webapps\oozie.war"
    $SecureConfigsDir="$CATALINA_BASE\conf\ssl"
    $ExtraLibs = Resolve-Path "$OOZIE_HOME\..\extra_libs"
    $EXTJS = "$ExtraLibs\ext-2.2.zip"

    # Validates the input\output wars
    if (!(Test-Path $InputWar)){
        PrintUsage
        throw "Path '$InputWar' doesn't exist"
    }
    if (!$InputWar.ToLower().EndsWith(".war")){
        PrintUsage
        throw "Invalid input war file '$InputWar'"
    }
    if (!$OutputWar.ToLower().EndsWith(".war")){
        PrintUsage
        throw "Invalid input war file '$OutputWar'"
    }
    if ($OutputWar -ieq $InputWar){
        PrintUsage
        throw "Invalid output\input war file. Both parameters cannot be the same file"
    }
    # Deletes previous output wars.
    if (Test-Path $OutputWar){
        Write-Output "Deleting existing output .war '$OutputWar'"
        Remove-Item -Force -Path $OutputWar
    }

    # Selects\Creates the temp directory
    $OOZIE_TEMP = "$OOZIE_HOME\temp"
    $OOZIE_WEB_INF_LIB = "$OOZIE_TEMP\WEB-INF\lib"

    Write-Output "Creating OOZIE_TEMP directory '$OOZIE_TEMP'"
    if (Test-Path "$OOZIE_TEMP") { Remove-Item "$OOZIE_TEMP" -Force -Recurse }
    $x = New-Item "$OOZIE_TEMP" -type directory

    # Extract the InputWar
    pushd $OOZIE_TEMP
    cmd /c $JAR_EXE xvf $InputWar
    if ($LASTEXITCODE -ne 0) {
        throw "Unable to execute 'jar xvf'. Error ($LASTEXITCODE)"
    }
    popd
    # Copy EXT_JS files
    if ((Test-Path $EXTJS)) {
        $EXTJS_HOME = "$ExtraLibs\ext-2.2"

        if (Test-Path "$EXTJS_HOME") { Remove-Item "$EXTJS_HOME" -Force -Recurse }
        $x = New-Item "$EXTJS_HOME" -type directory

        Expand-ZIPFile -File $EXTJS -Destination $EXTJS_HOME
        cp -r "$EXTJS_HOME\ext-2.2" "$OOZIE_TEMP"
    }else{
        Write-Output "INFO: Oozie webconsole disabled, ExtJS library not specified"
    }

    # Copy additional Jars
    if ($D -ne "") {
        $ExtraLibs = $D
    }
    Write-Output "   Adding JarFiles: $ExtraLibs\*.jar"
    cp -r $ExtraLibs\*.jar $OOZIE_WEB_INF_LIB

    if ($Secure) {
        #Use the SSL version of server.xml in oozie-server
        if (Test-Path $SecureConfigsDir\ssl-server.xml){
            cp $SecureConfigsDir\ssl-server.xml $CATALINA_BASE\conf\server.xml
        }

        #Inject the SSL version of web.xml in oozie war
        if (Test-Path $SecureConfigsDir\ssl-web.xml){
            cp $SecureConfigsDir\ssl-web.xml $OOZIE_TEMP\WEB-INF\web.xml
        }

        Write-Output "INFO: Using secure server.xml and secure web.xml"
    }else{
        #Use the regular version of server.xml in oozie-server
        if (Test-Path $SecureConfigsDir\server.xml){
            cp $SecureConfigsDir\server.xml $CATALINA_BASE\conf\server.xml
        }
        #No need to restore web.xml because its already in the original WAR file
    }

    Write-Output "Recreating the new war file '$OutputWar'..."

    cmd /c $JAR_EXE cvf $OutputWar -C $OOZIE_TEMP '.'
    if ($LASTEXITCODE -ne 0) {
        throw "Unable to execute 'jar uvf'. Error ($LASTEXITCODE)"
    }

    Write-Output "Done! $counter files added"
} else {
    PrintUsage
    exit -1
}