80 lines
		
	
	
		
			1.8 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			80 lines
		
	
	
		
			1.8 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
| # typed: true
 | |
| # frozen_string_literal: true
 | |
| 
 | |
| module Homebrew
 | |
|   # Helper module for simulating different system configurations.
 | |
|   #
 | |
|   # @api private
 | |
|   class SimulateSystem
 | |
|     class << self
 | |
|       attr_reader :arch, :os
 | |
| 
 | |
|       sig {
 | |
|         type_parameters(:U).params(
 | |
|           os:     Symbol,
 | |
|           arch:   Symbol,
 | |
|           _block: T.proc.returns(T.type_parameter(:U)),
 | |
|         ).returns(T.type_parameter(:U))
 | |
|       }
 | |
|       def with(os: T.unsafe(nil), arch: T.unsafe(nil), &_block)
 | |
|         raise ArgumentError, "At least one of `os` or `arch` must be specified." if !os && !arch
 | |
| 
 | |
|         old_os = self.os
 | |
|         old_arch = self.arch
 | |
| 
 | |
|         begin
 | |
|           self.os = os if os && os != current_os
 | |
|           self.arch = arch if arch && arch != current_arch
 | |
| 
 | |
|           yield
 | |
|         ensure
 | |
|           @os = old_os
 | |
|           @arch = old_arch
 | |
|         end
 | |
|       end
 | |
| 
 | |
|       sig { params(new_os: Symbol).void }
 | |
|       def os=(new_os)
 | |
|         os_options = [:macos, :linux, *MacOSVersion::SYMBOLS.keys]
 | |
|         raise "Unknown OS: #{new_os}" unless os_options.include?(new_os)
 | |
| 
 | |
|         @os = new_os
 | |
|       end
 | |
| 
 | |
|       sig { params(new_arch: Symbol).void }
 | |
|       def arch=(new_arch)
 | |
|         raise "New arch must be :arm or :intel" unless OnSystem::ARCH_OPTIONS.include?(new_arch)
 | |
| 
 | |
|         @arch = new_arch
 | |
|       end
 | |
| 
 | |
|       sig { void }
 | |
|       def clear
 | |
|         @os = @arch = nil
 | |
|       end
 | |
| 
 | |
|       sig { returns(T::Boolean) }
 | |
|       def simulating_or_running_on_macos?
 | |
|         [:macos, *MacOSVersion::SYMBOLS.keys].include?(os)
 | |
|       end
 | |
| 
 | |
|       sig { returns(T::Boolean) }
 | |
|       def simulating_or_running_on_linux?
 | |
|         os == :linux
 | |
|       end
 | |
| 
 | |
|       sig { returns(Symbol) }
 | |
|       def current_arch
 | |
|         @arch || Hardware::CPU.type
 | |
|       end
 | |
| 
 | |
|       sig { returns(Symbol) }
 | |
|       def current_os
 | |
|         os || :generic
 | |
|       end
 | |
|     end
 | |
|   end
 | |
| end
 | |
| 
 | |
| require "extend/os/simulate_system"
 | 
