ls コマンドを作るオブジェクト指向版 の課題提出

07.ls_object/bin/ls.rb

@@ -0,0 +1,21 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'io/console'
+require 'optparse'
+require_relative '../lib/ls_command'
+
+if __FILE__ == $PROGRAM_NAME
+  opt = OptionParser.new
+
+  params = { dot_match: false }
+  opt.on('-a') { |v| params[:dot_match] = v }
+  opt.on('-r') { |v| params[:reverse] = v }
+  opt.on('-l') { |v| params[:long_format] = v }
+  opt.parse!(ARGV)
+  path = ARGV[0] || '.'
+  width = IO.console.winsize[1]
+
+  ls = LsCommand.new(path, width:, **params)
+  puts ls.formatted_output
+end

07.ls_object/lib/file/file_data.rb

@@ -0,0 +1,82 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+require 'etc'
+require 'pathname'
+
+class FileData
+  attr_reader :name, :file_status
+
+  PERMISSION_START_POSITION = 7
+  PERMISSION_END_POSITION = 9
+
+  FILE_TYPE = {
+    'fifo' => 'p',
+    'characterSpecial' => 'c',
+    'directory' => 'd',
+    'blockSpecial' => 'b',
+    'file' => '-',
+    'link' => 'l',
+    'socket' => 's'
+  }.freeze
+
+  SPECIAL_PERMISSION = {
+    0 => 's',
+    1 => 's',
+    2 => 't'
+  }.freeze
+
+  PERMISSION_TYPE = {
+    '000' => '---',
+    '001' => '--x',
+    '010' => '-w-',
+    '011' => '-wx',
+    '100' => 'r--',
+    '101' => 'r-x',
+    '110' => 'rw-',
+    '111' => 'rwx'
+  }.freeze
+
+  def initialize(name, path)
+    @name = name
+    @full_path = Pathname(File.absolute_path(@name, path))
+    @file_type = File.ftype(@full_path)
+    @file_status = build_file_status(File.lstat(@full_path))
+  end
+
+  private
+
+  def build_file_status(status)
+    {
+      type: FILE_TYPE[status.ftype],
+      mode: mode(status),
+      hardlink_nums: status.nlink.to_s,
+      owner_name: Etc.getpwuid(status.uid).name,
+      group_name: Etc.getgrgid(status.gid).name,
+      bytesize: rdev_or_bytesize(status),
+      latest_modify_datetime: status.mtime.strftime('%_m %e %H:%M'),
+      filename: file_name(status),
+      blocks: status.blocks
+    }
+  end
+
+  def file_name(status)
+    status.symlink? ? "#{@name} -> #{File.readlink(@full_path)}" : @name
+  end
+
+  def mode(status)
+    mode_binary_numbers = status.mode.to_s(2).rjust(16, '0')
+    # SUID、SGID、STICKEYBITの順番で特殊権限をチェック
+    mode_binary_numbers[4..6].each_char.with_index.map do |special_permission, i|
+      range_start = PERMISSION_START_POSITION + (3 * i)
+      range_end = PERMISSION_END_POSITION + (3 * i)
+      permission = PERMISSION_TYPE[mode_binary_numbers[range_start..range_end]].dup
+      permission[2] = permission[2] == 'x' ? SPECIAL_PERMISSION[i] : SPECIAL_PERMISSION[i].upcase if special_permission == '1'
+      permission
+    end.join
+  end
+
+  def rdev_or_bytesize(status)
+    %w[characterSpecial blockSpecial].include?(@file_type) ? format('%#01x', status.rdev.to_s(10)) : status.size.to_s
+  end
+end

07.ls_object/lib/format/formatter.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+class Formatter
+  def initialize(files)
+    @files = files
+  end
+
+  def format; end
+end

07.ls_object/lib/format/long_formatter.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+require_relative 'formatter'
+
+class LongFormatter < Formatter
+  def initialize(files, path)
+    super(files)
+    @path = path
+    @max_sizes = build_max_sizes
+    @total_block = sum_blocks
+    @long_format_data = build_long_format_data
+  end
+
+  def format
+    total = "total #{@total_block}" if File.directory?(@path) || @files.size > 1
+    [total, *@long_format_data].compact.join("\n")
+  end
+
+  private
+
+  def build_long_format_data
+    @files.map { |file| format_row(file.file_status, *@max_sizes) }
+  end
+
+  def build_max_sizes
+    %i[hardlink_nums owner_name group_name bytesize].map do |key|
+      find_max_size(key)
+    end
+  end
+
+  def find_max_size(key)
+    @files.map { |file| file.file_status[key].size }.max
+  end
+
+  def format_row(status, max_nlink, max_user, max_group, max_size)
+    [
+      "#{status[:type]}#{status[:mode]}",
+      "  #{status[:hardlink_nums].rjust(max_nlink)}",
+      " #{status[:owner_name].ljust(max_user)}",
+      "  #{status[:group_name].ljust(max_group)}",
+      "  #{status[:bytesize].rjust(max_size)}",
+      " #{status[:latest_modify_datetime]}",
+      " #{status[:filename]}"
+    ].join
+  end
+
+  def sum_blocks
+    @files.sum { |file| file.file_status[:blocks] }
+  end
+end

07.ls_object/lib/format/short_formatter.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require_relative 'formatter'
+
+class ShortFormatter < Formatter
+  def initialize(files, width)
+    super(files)
+    @width = width
+    @max_file_name = @files.map { |file| File.basename(file.name).size }.max
+    @col_count = calculate_col_count
+    @row_count = calculate_row_count
+    @short_format_data = build_short_format_data
+  end
+
+  def format
+    @short_format_data.join("\n")
+  end
+
+  private
+
+  def build_short_format_data
+    safe_transpose.map { |row_files| build_short_format_row(row_files) }
+  end
+
+  def build_short_format_row(row_files)
+    row_files.map do |file|
+      basename = file ? File.basename(file.name) : ''
+      basename.ljust(@max_file_name + 1)
+    end.join.rstrip
+  end
+
+  def calculate_col_count
+    @width / (@max_file_name + 1)
+  end
+
+  def calculate_row_count
+    @col_count.zero? ? @files.count : (@files.count.to_f / @col_count).ceil
+  end
+
+  def safe_transpose
+    nested_file_names = @files.each_slice(@row_count).to_a
+    nested_file_names[0].zip(*nested_file_names[1..])
+  end
+end

07.ls_object/lib/ls_command.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require_relative 'file/file_data'
+require_relative 'format/long_formatter'
+require_relative 'format/short_formatter'
+
+class LsCommand
+  def initialize(path, width: 80, dot_match: false, reverse: false, long_format: false)
+    @path = path
+    @width = width
+    @dot_match = dot_match
+    @reverse = reverse
+    @long_format = long_format
+    @files = build_dot_match_and_sorted_files
+  end
+
+  def formatted_output
+    @long_format ? LongFormatter.new(@files, @path).format : ShortFormatter.new(@files, @width).format
+  end
+
+  private
+
+  def build_dot_match_and_sorted_files
+    files = generate_files
+    matched_files = @dot_match ? files : files.filter { |file| !/^\./.match?(file.name) }
+    @reverse ? matched_files.sort_by(&:name).reverse : matched_files.sort_by(&:name)
+  end
+
+  def generate_files
+    if File.directory?(@path)
+      Dir.open(@path).entries.map { |name| FileData.new(name, @path) }
+    else
+      [FileData.new(@path, File.dirname(@path))]
+    end
+  end
+end