winrm-fs 0.2.3 → 0.3.0

data/lib/winrm-fs/core/upload_orchestrator.rb

@@ -1,119 +0,0 @@
-# encoding: UTF-8
-#
-# Copyright 2015 Shawn Neal <sneal@sneal.net>
-#
-# 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_relative 'temp_zip_file'
-require_relative 'file_uploader'
-require_relative 'command_executor'
-require_relative '../scripts/scripts'
-
-module WinRM
-  module FS
-    module Core
-      # Orchestrates the upload of a file or directory
-      class UploadOrchestrator
-        def initialize(service)
-          @service = service
-          @logger = Logging.logger[self]
-        end
-
-        def upload_file(local_path, remote_path, &block)
-          # If the src has a file extension and the destination does not
-          # we can assume the caller specified the dest as a directory
-          if File.extname(local_path) != '' && File.extname(remote_path) == ''
-            remote_path = File.join(remote_path, File.basename(local_path))
-          end
-          temp_path = temp_file_path(local_path)
-          with_command_executor do |cmd_executor|
-            return 0 unless out_of_date?(cmd_executor, local_path, remote_path)
-            do_file_upload(cmd_executor, local_path, temp_path, remote_path, &block)
-          end
-        end
-
-        def upload_directory(local_path, remote_path, &block)
-          with_local_zip(local_path) do |local_zip|
-            temp_path = temp_file_path(local_zip.path)
-            with_command_executor do |cmd_executor|
-              return 0 unless out_of_date?(cmd_executor, local_zip.path, temp_path)
-              do_file_upload(cmd_executor, local_zip.path, temp_path, remote_path, &block)
-            end
-          end
-        end
-
-        private
-
-        def do_file_upload(cmd_executor, local_path, temp_path, remote_path)
-          file_uploader = WinRM::FS::Core::FileUploader.new(cmd_executor)
-          bytes = file_uploader.upload(local_path, temp_path) do |bytes_copied, total_bytes|
-            yield bytes_copied, total_bytes, local_path, remote_path if block_given?
-          end
-
-          cmd_executor.run_powershell(
-            WinRM::FS::Scripts.render('decode_file', src: temp_path, dest: remote_path))
-
-          bytes
-        end
-
-        def with_command_executor
-          cmd_executor = WinRM::FS::Core::CommandExecutor.new(@service)
-          cmd_executor.open
-          yield cmd_executor
-        ensure
-          cmd_executor.close
-        end
-
-        def with_local_zip(local_path)
-          local_zip = create_temp_zip_file(local_path)
-          yield local_zip
-        ensure
-          local_zip.delete if local_zip
-        end
-
-        def out_of_date?(cmd_executor, local_path, remote_path)
-          local_checksum = local_checksum(local_path)
-          remote_checksum = remote_checksum(cmd_executor, remote_path)
-
-          if remote_checksum == local_checksum
-            @logger.debug("#{remote_path} is up to date")
-            return false
-          end
-          true
-        end
-
-        def remote_checksum(cmd_executor, remote_path)
-          script = WinRM::FS::Scripts.render('checksum', path: remote_path)
-          cmd_executor.run_powershell(script).chomp
-        end
-
-        def local_checksum(local_path)
-          Digest::MD5.file(local_path).hexdigest
-        end
-
-        def temp_file_path(local_path)
-          ext = '.tmp'
-          ext = '.zip' if File.extname(local_path) == '.zip'
-          "$env:TEMP/winrm-upload/#{local_checksum(local_path)}#{ext}"
-        end
-
-        def create_temp_zip_file(local_path)
-          zip = WinRM::FS::Core::TempZipFile.new(local_path, recurse_paths: true)
-          zip.add(local_path)
-          zip.build
-          zip
-        end
-      end
-    end
-  end
-end

data/lib/winrm-fs/scripts/decode_file.ps1.erb

@@ -1,36 +0,0 @@
-$path = $ExecutionContext.SessionState.Path
-$tempFile = $path.GetUnresolvedProviderPathFromPSPath("<%= src %>")
-$dest = $path.GetUnresolvedProviderPathFromPSPath("<%= dest %>")
-
-function Decode-File($encodedFile, $decodedFile) {
-    if (Test-Path $encodedFile) {
-      $base64Content = Get-Content $encodedFile
-    }
-    if ($base64Content -eq $null) {
-        New-Item -ItemType file -Force $decodedFile | Out-Null
-    }
-    else {
-        $bytes = [System.Convert]::FromBase64String($base64Content)
-        [System.IO.File]::WriteAllBytes($decodedFile, $bytes) | Out-Null
-    }
-}
-
-function Ensure-Dir-Exists($path) {
-    # ensure the destination directory exists
-    if (!(Test-Path $path)) {
-        New-Item -ItemType Directory -Force -Path $path | Out-Null
-    }
-}
-
-if ([System.IO.Path]::GetExtension($tempFile) -eq '.zip') {
-    Ensure-Dir-Exists $dest
-    Decode-File $tempFile $tempFile
-    $shellApplication = New-Object -com shell.application
-    $zipPackage = $shellApplication.NameSpace($tempFile)
-    $destinationFolder = $shellApplication.NameSpace($dest)
-    $destinationFolder.CopyHere($zipPackage.Items(), 0x10) | Out-Null
-}
-else {
-    Ensure-Dir-Exists ([System.IO.Path]::GetDirectoryName($dest))
-    Decode-File $tempFile $dest
-}

data/spec/temp_zip_file_spec.rb

@@ -1,55 +0,0 @@
-# encoding: UTF-8
-require_relative '../lib/winrm-fs/core/temp_zip_file'
-
-describe WinRM::FS::Core::TempZipFile, integration: true do
-  let(:winrm_fs_dir) { File.expand_path('../lib/winrm-fs', File.dirname(__FILE__)) }
-  let(:temp_zip_file_spec) { __FILE__ }
-  let(:spec_helper) { File.expand_path('spec_helper.rb', File.dirname(__FILE__)) }
-
-  subject { WinRM::FS::Core::TempZipFile.new }
-
-  context 'temp file creation' do
-    it 'should create a temp file on disk' do
-      expect(File.exist?(subject.path)).to be true
-      subject.delete
-      expect(File.exist?(subject.path)).to be false
-    end
-  end
-
-  context 'create zip' do
-    it 'should raise error when file doesn not exist' do
-      expect { subject.add('/etc/foo/does/not/exist') }.to raise_error
-    end
-
-    it 'should add a file to the zip' do
-      subject.add(temp_zip_file_spec)
-      subject.build
-      expect(subject).to contain_zip_entries('spec/temp_zip_file_spec.rb')
-    end
-
-    it 'should add multiple files to the zip' do
-      subject.add(temp_zip_file_spec)
-      subject.add(spec_helper)
-      subject.build
-      expect(subject).to contain_zip_entries([
-        'spec/temp_zip_file_spec.rb',
-        'spec/spec_helper.rb'])
-    end
-
-    it 'should add all files in directory' do
-      subject.add(winrm_fs_dir)
-      subject.build
-      expect(subject).to contain_zip_entries('lib/winrm-fs/exceptions.rb')
-    end
-
-    it 'should add all files in directory to the zip recursively' do
-      subject = WinRM::FS::Core::TempZipFile.new(Dir.pwd, recurse_paths: true)
-      subject.add(winrm_fs_dir)
-      subject.build
-      expect(subject).to contain_zip_entries([
-        'lib/winrm-fs/exceptions.rb',
-        'lib/winrm-fs/core/temp_zip_file.rb',
-        'lib/winrm-fs/scripts/checksum.ps1.erb'])
-    end
-  end
-end