# -*- encoding: utf-8 -*-
#
# Author:: Steven Murawski (<smurawski@chef.io>)
#
# Copyright (C) 2015, Chef Software
#
# Licensed 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.

require "kitchen/provisioner/chef_zero"

module Kitchen
  module Provisioner
    # Chef Zero provisioner to run in scheduled task on Windows
    class ChefZeroScheduledTask < ChefZero
      kitchen_provisioner_api_version 2

      plugin_version Kitchen::VERSION

      default_config :task_username

      default_config :task_password

      def prepare_command
        if windows_os?
          debug("Serializing $env:temp")
          wrap_shell_code(<<-EOH
            "`$env:temp = '$env:temp'" |
                out-file $env:temp/kitchen/env.ps1
          EOH
          )
        end
        super
      end

      def init_command
        if windows_os?
          info("Creating the scheduled task.")
          wrap_shell_code("#{super}\n#{setup_scheduled_task_command}")
        else
          super
        end
      end

      def run_command
        if windows_os?
          script = "$script = @'\n#{scheduled_task_command_script}\n'@\n" \
          "\n$Script | out-file \"$env:temp/kitchen/chef-client-script.ps1\"" \
          "\n#{run_scheduled_task_command}"
          wrap_shell_code(script)
        else
          super
        end
      end

      # private

      def local_state_file
        @local_state_file ||= @instance.diagnose[:state_file]
      end

      def task_username
        if config[:task_username]
          config[:task_username]
        else
          if local_state_file.key?(:username)
            local_state_file[:username]
          else
            @instance.transport[:username]
          end
        end
      end

      def task_password
        unless config[:task_password]
          if local_state_file.key?(:password)
            local_state_file[:password]
          else
            @instance.transport[:password]
          end
        end
      end

      def run_scheduled_task_command
        <<-EOH
        try {
          $npipeServer = new-object System.IO.Pipes.NamedPipeServerStream(
            'task', [System.IO.Pipes.PipeDirection]::In)
          $pipeReader = new-object System.IO.StreamReader($npipeServer)
          schtasks /run /tn "chef-tk"
          $npipeserver.waitforconnection()
          $host.ui.writeline('Connected to the scheduled task.')
          while ($npipeserver.IsConnected) {
            $output = $pipereader.ReadLine()
            if ($output -like 'SCHEDULED_TASK_DONE:*') {
              $exit_code = ($output -replace 'SCHEDULED_TASK_DONE:').trim()
            }
            else { $host.ui.WriteLine($output) } } }
        finally {
          $pipereader.dispose()
          $npipeserver.dispose()
          $host.setshouldexit($exit_code)
        }
        EOH
      end

      def new_scheduled_task_command
        "schtasks /create /tn 'chef-tk' " \
        "/ru '#{task_username}' /rp '#{task_password}' " \
        "/sc daily /st 00:00 /rl HIGHEST /f "
      end

      def new_scheduled_task_command_line_ps
        "/tr $executioncontext.invokecommand.expandstring(" \
        '"powershell -executionpolicy unrestricted -File '
      end

      def setup_scheduled_task_command
        new_scheduled_task_command +
          new_scheduled_task_command_line_ps +
          remote_chef_client_script +
          '")'
      end

      def remote_chef_client_script
        @remote_script_path ||= remote_path_join(
          config[:root_path], "chef-client-script.ps1")
      end

      def scheduled_task_command_script
        <<-EOH
        . $psscriptroot/env.ps1;
        start-sleep -seconds 5;
        $npipeClient = new-object System.IO.Pipes.NamedPipeClientStream(
        $env:ComputerName, 'task', [System.IO.Pipes.PipeDirection]::Out);
        $npipeclient.connect();
        $pipeWriter = new-object System.IO.StreamWriter($npipeClient);
        $pipeWriter.AutoFlush = $true;
        #{client_zero_command} |
          foreach-object {} {$pipewriter.writeline($_)} {
            $pipewriter.writeline("SCHEDULED_TASK_DONE: $LastExitCode");
            $pipewriter.dispose();
            $npipeclient.dispose()
          }
        EOH
      end

      def prepare_client_zero_script
        File.open(File.join(sandbox_path, "chef-client-script.ps1"), "w") do |file|
          file.write(Util.outdent!(scheduled_task_command_script))
        end
      end

      def client_zero_command
        [local_mode_command, *chef_client_args].join(" ")
      end

      def chef_client_args
        level = config[:log_level] == :info ? :info : config[:log_level]
        args = [
          "--config #{remote_path_join(config[:root_path], "client.rb")}",
          "--log_level #{level}",
          "--force-formatter",
          "--no-color"
        ]
        add_optional_chef_client_args!(args)

        args
      end
    end
  end
end