#
# Copyright:: Copyright 2015-2016, Chef Software, Inc.
# License:: Apache License, Version 2.0
#
# 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 "chef/win32/api/registry"
require "chef/win32/unicode"
require "win32/registry"

module Win32
  class Registry

    module API

      extend Chef::ReservedNames::Win32::API::Registry

      module_function

      # ::Win32::Registry#delete_value uses RegDeleteValue which
      # is not an imported function after bug 10820 was solved.  So
      # we overwrite it to call the correct imported function.
      # https://bugs.ruby-lang.org/issues/10820
      # Still a bug in trunk as of March 21, 2016 (Ruby 2.3.0)
      def DeleteValue(hkey, name)
        check RegDeleteValueW(hkey, name.to_wstring)
      end

      # ::Win32::Registry#delete_key uses RegDeleteKeyW. We need to use
      # RegDeleteKeyExW to properly support WOW64 systems.
      # Still a bug in trunk as of March 21, 2016 (Ruby 2.3.0)
      def DeleteKey(hkey, name)
        check RegDeleteKeyExW(hkey, name.to_wstring, 0, 0)
      end

    end

    if RUBY_VERSION =~ /^2\.1/
      # ::Win32::Registry#write does not correctly handle data in Ruby 2.1
      # This bug is _reportedly_ resolved in Ruby 2.1.7 and 2.2.3
      # but fails in appveyor on 2.1.8 unless we keep applying this monkeypatch
      # https://bugs.ruby-lang.org/issues/11439
      def write(name, type, data)
        case type
        when REG_SZ, REG_EXPAND_SZ
          data = data.to_s.encode(WCHAR) + WCHAR_NUL
        when REG_MULTI_SZ
          data = data.to_a.map { |s| s.encode(WCHAR) }.join(WCHAR_NUL) << WCHAR_NUL << WCHAR_NUL
        when REG_BINARY
          data = data.to_s
        when REG_DWORD
          data = API.packdw(data.to_i)
        when REG_DWORD_BIG_ENDIAN
          data = [data.to_i].pack("N")
        when REG_QWORD
          data = API.packqw(data.to_i)
        else
          raise TypeError, "Unsupported type #{type}"
        end
        API.SetValue(@hkey, name, type, data, data.bytesize)
      end
    end
  end
end