-
-
Notifications
You must be signed in to change notification settings - Fork 9.3k
/
PATH.rb
95 lines (78 loc) 路 2.33 KB
/
PATH.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# typed: true
# frozen_string_literal: true
# Representation of a `*PATH` environment variable.
class PATH
include Enumerable
extend Forwardable
delegate each: :@paths
# FIXME: Enable cop again when https://github.com/sorbet/sorbet/issues/3532 is fixed.
# rubocop:disable Style/MutableConstant
Element = T.type_alias { T.nilable(T.any(Pathname, String, PATH)) }
private_constant :Element
Elements = T.type_alias { T.any(Element, T::Array[Element]) }
private_constant :Elements
# rubocop:enable Style/MutableConstant
sig { params(paths: Elements).void }
def initialize(*paths)
@paths = parse(paths)
end
sig { params(paths: Elements).returns(T.self_type) }
def prepend(*paths)
@paths = parse(paths + @paths)
self
end
sig { params(paths: Elements).returns(T.self_type) }
def append(*paths)
@paths = parse(@paths + paths)
self
end
sig { params(index: Integer, paths: Elements).returns(T.self_type) }
def insert(index, *paths)
@paths = parse(@paths.insert(index, *paths))
self
end
sig { params(block: T.proc.params(arg0: String).returns(T::Boolean)).returns(T.self_type) }
def select(&block)
self.class.new(@paths.select(&block))
end
sig { params(block: T.proc.params(arg0: String).returns(T::Boolean)).returns(T.self_type) }
def reject(&block)
self.class.new(@paths.reject(&block))
end
sig { returns(T::Array[String]) }
def to_ary
@paths.dup.to_ary
end
alias to_a to_ary
sig { returns(String) }
def to_str
@paths.join(File::PATH_SEPARATOR)
end
# @!visibility private
sig { returns(String) }
def to_s = to_str
sig { params(other: T.untyped).returns(T::Boolean) }
def ==(other)
(other.respond_to?(:to_ary) && to_ary == other.to_ary) ||
(other.respond_to?(:to_str) && to_str == other.to_str) ||
false
end
sig { returns(T::Boolean) }
def empty?
@paths.empty?
end
sig { returns(T.nilable(T.self_type)) }
def existing
existing_path = select { File.directory?(_1) }
# return nil instead of empty PATH, to unset environment variables
existing_path unless existing_path.empty?
end
private
sig { params(paths: T::Array[Elements]).returns(T::Array[String]) }
def parse(paths)
paths.flatten
.compact
.flat_map { |p| Pathname(p).to_path.split(File::PATH_SEPARATOR) }
.uniq
end
end