111 lines
3.5 KiB
Ruby
111 lines
3.5 KiB
Ruby
#-------------------------------------------------------------------------
|
|
# Copyright (c) Microsoft Open Technologies, Inc.
|
|
# All Rights Reserved. Licensed under the MIT License.
|
|
#--------------------------------------------------------------------------
|
|
require "debugger"
|
|
require "json"
|
|
require "vagrant/util/which"
|
|
require "vagrant/util/subprocess"
|
|
|
|
module VagrantPlugins
|
|
module HyperV
|
|
module Driver
|
|
class Base
|
|
attr_reader :vmid
|
|
|
|
def initialize(id=nil)
|
|
@vmid = id
|
|
check_power_shell
|
|
@output = nil
|
|
end
|
|
|
|
def execute(path, options)
|
|
r = execute_powershell(path, options) do |type, data|
|
|
process_output(type, data)
|
|
end
|
|
if success?
|
|
JSON.parse(json_output[:success].join) unless json_output[:success].empty?
|
|
else
|
|
message = json_output[:error].join unless json_output[:error].empty?
|
|
raise Error::SubprocessError, message if message
|
|
end
|
|
end
|
|
|
|
def raw_execute(command)
|
|
command = [command , {notify: [:stdout, :stderr, :stdin]}].flatten
|
|
clear_output_buffer
|
|
Vagrant::Util::Subprocess.execute(*command) do |type, data|
|
|
process_output(type, data)
|
|
end
|
|
end
|
|
|
|
protected
|
|
|
|
def json_output
|
|
return @json_output if @json_output
|
|
json_success_begin = false
|
|
json_error_begin = false
|
|
success = []
|
|
error = []
|
|
@output.split("\n").each do |line|
|
|
json_error_begin = false if line.include?("===End-Error===")
|
|
json_success_begin = false if line.include?("===End-Output===")
|
|
message = ""
|
|
if json_error_begin || json_success_begin
|
|
message = line.gsub("\\'","\"")
|
|
end
|
|
success << message if json_success_begin
|
|
error << message if json_error_begin
|
|
json_success_begin = true if line.include?("===Begin-Output===")
|
|
json_error_begin = true if line.include?("===Begin-Error===")
|
|
end
|
|
@json_output = { :success => success, :error => error }
|
|
end
|
|
|
|
def success?
|
|
@error_messages.empty? && json_output[:error].empty?
|
|
end
|
|
|
|
def process_output(type, data)
|
|
if type == :stdout
|
|
@output = data.gsub("\r\n", "\n")
|
|
end
|
|
if type == :stdin
|
|
# $stdin.gets.chomp || ""
|
|
end
|
|
if type == :stderr
|
|
@error_messages = data.gsub("\r\n", "\n")
|
|
end
|
|
end
|
|
|
|
def clear_output_buffer
|
|
@output = ""
|
|
@error_messages = ""
|
|
@json_output = nil
|
|
end
|
|
|
|
def check_power_shell
|
|
unless Vagrant::Util::Which.which('powershell')
|
|
raise "Power Shell not found"
|
|
end
|
|
end
|
|
|
|
def execute_powershell(path, options, &block)
|
|
lib_path = Pathname.new(File.expand_path("../../scripts", __FILE__))
|
|
path = lib_path.join(path).to_s.gsub("/", "\\")
|
|
options = options || {}
|
|
ps_options = []
|
|
options.each do |key, value|
|
|
ps_options << "-#{key}"
|
|
ps_options << "'#{value}'"
|
|
end
|
|
clear_output_buffer
|
|
command = ["powershell", "-NoProfile", "-ExecutionPolicy",
|
|
"Bypass", path, ps_options, {notify: [:stdout, :stderr, :stdin]}].flatten
|
|
Vagrant::Util::Subprocess.execute(*command, &block)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|