Class: Chef::Knife::List

Inherits:
ChefFS::Knife show all
Defined in:
lib/chef/knife/list.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods inherited from ChefFS::Knife

#chef_fs, #configure_chef, #create_chef_fs, #create_local_fs, deps, #discover_repo_dir, #format_path, inherited, #local_fs, #parallelize, #pattern_arg_from, #pattern_args, #pattern_args_from

Instance Attribute Details

#exit_code ⇒ Object

Returns the value of attribute exit_code.



62
63
64
 
# File 'lib/chef/knife/list.rb', line 62

def exit_code
  @exit_code
end

Instance Method Details

#add_dir_result(result) ⇒ Object 



124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
 
# File 'lib/chef/knife/list.rb', line 124

def add_dir_result(result)
  begin
    children = result.children.sort_by(&:name)
  rescue Chef::ChefFS::FileSystem::NotFoundError => e
    ui.error "#{format_path(e.entry)}: No such file or directory"
    return []
  end

  result = [ [ result, children ] ]
  if config[:recursive]
    child_dirs = children.select(&:dir?)
    result += parallelize(child_dirs) { |child| add_dir_result(child) }.flatten(1).to_a
  end
  result
end

#maybe_add_slash(path, is_dir) ⇒ Object 



168
169
170
171
172
173
174
 
# File 'lib/chef/knife/list.rb', line 168

def maybe_add_slash(path, is_dir)
  if config[:trailing_slashes] && is_dir
    "#{path}/"
  else
    path
  end
end

#print_result_paths(results, indent = "") ⇒ Object 



140
141
142
 
# File 'lib/chef/knife/list.rb', line 140

def print_result_paths(results, indent = "")
  print_results(results.map { |result| maybe_add_slash(format_path(result), result.dir?) }, indent)
end

#print_results(results, indent) ⇒ Object 



144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
 
# File 'lib/chef/knife/list.rb', line 144

def print_results(results, indent)
  return if results.length == 0

  print_space = results.map(&:length).max + 2
  if config[:one_column] || !stdout.isatty
    columns = 0
  else
    columns = TTY::Screen.columns
  end
  current_line = ""
  results.each do |result|
    if current_line.length > 0 && current_line.length + print_space > columns
      output current_line.rstrip
      current_line = ""
    end
    if current_line.length == 0
      current_line << indent
    end
    current_line << result
    current_line << (" " * (print_space - result.length))
  end
  output current_line.rstrip if current_line.length > 0
end

#run ⇒ Object 



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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
 
# File 'lib/chef/knife/list.rb', line 64

def run
  patterns = name_args.length == 0 ? [""] : name_args

  # Get the top-level matches
  all_results = parallelize(pattern_args_from(patterns)) do |pattern|
    pattern_results = Chef::ChefFS::FileSystem.list(config[:local] ? local_fs : chef_fs, pattern).to_a

    if pattern_results.first && !pattern_results.first.exists? && pattern.exact_path
      ui.error "#{format_path(pattern_results.first)}: No such file or directory"
      self.exit_code = 1
    end
    pattern_results
  end.flatten(1).to_a

  # Process directories
  if !config[:bare_directories]
    dir_results = parallelize(all_results.select(&:dir?)) do |result|
      add_dir_result(result)
    end.flatten(1)

  else
    dir_results = []
  end

  # Process all other results
  results = all_results.select { |result| result.exists? && (!result.dir? || config[:bare_directories]) }.to_a

  # Flatten out directory results if necessary
  if config[:flat]
    dir_results.each do |result, children| # rubocop:disable Style/HashEachMethods
      results += children
    end
    dir_results = []
  end

  # Sort by path for happy output
  results = results.sort_by(&:path)
  dir_results = dir_results.sort_by { |result| result[0].path }

  # Print!
  if results.length == 0 && dir_results.length == 1
    results = dir_results[0][1]
    dir_results = []
  end

  print_result_paths results
  printed_something = results.length > 0
  dir_results.each do |result, children|
    if printed_something
      output ""
    else
      printed_something = true
    end
    output "#{format_path(result)}:"
    print_results(children.map { |child| maybe_add_slash(child.display_name, child.dir?) }.sort, "")
  end

  exit exit_code if exit_code
end