77 lines
1.9 KiB
Ruby
Raw Normal View History

2020-11-20 14:20:38 +01:00
# typed: strict
# frozen_string_literal: true
require "hardware"
require "diagnostic"
require "extend/ENV/shared"
require "extend/ENV/std"
require "extend/ENV/super"
2013-08-19 12:32:57 -05:00
2020-11-20 14:20:38 +01:00
module Kernel
sig { params(env: T.nilable(String)).returns(T::Boolean) }
def superenv?(env)
return false if env == "std"
!Superenv.bin.nil?
end
private :superenv?
2013-08-19 12:32:57 -05:00
end
2023-01-11 18:16:43 -08:00
# @!parse
# # ENV is not actually a class, but this makes `YARD` happy
# # @see https://rubydoc.info/stdlib/core/ENV ENV core documentation
# class ENV; end
2013-08-19 12:32:57 -05:00
module EnvActivation
2020-11-20 14:20:38 +01:00
sig { params(env: T.nilable(String)).void }
def activate_extensions!(env: nil)
if superenv?(env)
2013-08-19 12:32:57 -05:00
extend(Superenv)
else
2013-08-19 12:32:59 -05:00
extend(Stdenv)
2013-08-19 12:32:57 -05:00
end
end
sig {
2020-11-20 14:20:38 +01:00
params(
2022-07-26 19:36:43 +01:00
env: T.nilable(String),
cc: T.nilable(String),
build_bottle: T::Boolean,
bottle_arch: T.nilable(String),
debug_symbols: T.nilable(T::Boolean),
_block: T.proc.returns(T.untyped),
2020-11-20 14:20:38 +01:00
).returns(T.untyped)
}
2022-07-30 11:10:26 +01:00
def with_build_environment(env: nil, cc: nil, build_bottle: false, bottle_arch: nil, debug_symbols: false, &_block)
old_env = to_hash.dup
tmp_env = to_hash.dup.extend(EnvActivation)
2020-11-20 14:20:38 +01:00
T.cast(tmp_env, EnvActivation).activate_extensions!(env: env)
T.cast(tmp_env, T.any(Superenv, Stdenv))
.setup_build_environment(cc: cc, build_bottle: build_bottle, bottle_arch: bottle_arch,
2022-07-30 11:10:26 +01:00
debug_symbols: debug_symbols)
replace(tmp_env)
2020-11-20 14:20:38 +01:00
begin
yield
ensure
replace(old_env)
end
end
2020-11-20 14:20:38 +01:00
sig { params(key: T.any(String, Symbol)).returns(T::Boolean) }
def sensitive?(key)
key.match?(/(cookie|key|token|password|passphrase)/i)
end
2018-09-17 02:45:00 +02:00
2020-11-20 14:20:38 +01:00
sig { returns(T::Hash[String, String]) }
def sensitive_environment
select { |key, _| sensitive?(key) }
end
2020-11-20 14:20:38 +01:00
sig { void }
def clear_sensitive_environment!
each_key { |key| delete key if sensitive?(key) }
end
2013-08-19 12:32:57 -05:00
end
ENV.extend(EnvActivation)