kanrisuru 0.13.0 → 0.14.0

data/lib/kanrisuru/core/socket/commands/ss.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module Kanrisuru
+  module Core
+    module Socket
+      def ss(opts = {})
+        state = opts[:state]
+        expression = opts[:expression]
+        family = opts[:family]
+
+        command = Kanrisuru::Command.new('ss')
+
+        command.append_flag('-a')
+        command.append_flag('-m')
+
+        command.append_flag('-n', opts[:numeric])
+        command.append_flag('-t', opts[:tcp])
+        command.append_flag('-u', opts[:udp])
+        command.append_flag('-x', opts[:unix])
+        command.append_flag('-w', opts[:raw])
+
+        if Kanrisuru::Util.present?(family)
+          raise ArgumentError, 'invalid family type' unless NETWORK_FAMILIES.include?(family)
+
+          command.append_arg('-f', family)
+        end
+
+        if Kanrisuru::Util.present?(state)
+          raise ArgumentError, 'invalid filter state' if !TCP_STATES.include?(state) && !OTHER_STATES.include?(state)
+
+          command.append_arg('state', state)
+        end
+
+        command << expression if Kanrisuru::Util.present?(expression)
+
+        execute_shell(command)
+
+        Kanrisuru::Result.new(command) do |cmd|
+          Parser::Ss.parse(cmd, state)
+        end
+      end
+    end
+  end
+end

data/lib/kanrisuru/core/socket/commands.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require_relative 'commands/ss'

data/lib/kanrisuru/core/socket/constants.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module Kanrisuru
+  module Core
+    module Socket
+      TCP_STATES = %w[
+        established syn-sent syn-recv
+        fin-wait-1 fin-wait-2 time-wait
+        closed close-wait last-ack listening closing
+      ].freeze
+
+      OTHER_STATES = %w[
+        all connected synchronized bucket syn-recv
+        big
+      ].freeze
+
+      TCP_STATE_ABBR = {
+        'ESTAB' => 'established', 'LISTEN' => 'listening', 'UNCONN' => 'unconnected',
+        'SYN-SENT' => 'syn-sent', 'SYN-RECV' => 'syn-recv', 'FIN-WAIT-1' => 'fin-wait-1',
+        'FIN-WAIT-2' => 'fin-wait-2', 'TIME-WAIT' => 'time-wait', 'CLOSE-WAIT' => 'close-wait',
+        'LAST-ACK' => 'last-ack', 'CLOSING' => 'closing'
+      }.freeze
+
+      NETWORK_FAMILIES = %w[
+        unix inet inet6 link netlink
+      ].freeze
+    end
+  end
+end

data/lib/kanrisuru/core/socket/parser.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require_relative 'parsers/ss'

data/lib/kanrisuru/core/socket/parsers/ss.rb

@@ -0,0 +1,109 @@
+# frozen_string_literal: true
+
+module Kanrisuru
+  module Core
+    module Socket
+      module Parser
+        class Ss
+          class << self
+            def parse(command, state)
+              rows = []
+
+              ## New lines with tabs are from text overflow
+              ## on on stdout of SS command.
+              ## Replace the newline and tab chars with a space.
+              string = command.raw_result.join
+              string = string.gsub("\n\t", "\s")
+              lines = string.lines.map(&:strip)
+
+              headers = lines.shift
+
+              lines.each do |line|
+                values = line.split
+                next if values.length < 5
+
+                socket_stats = Kanrisuru::Core::Socket::Statistics.new
+                socket_stats.netid =
+                  if headers.include?('Netid')
+                    values.shift
+                  elsif opts[:tcp]
+                    'tcp'
+                  elsif opts[:udp]
+                    'udp'
+                  elsif opts[:raw]
+                    'raw'
+                  else
+                    ''
+                  end
+
+                socket_stats.state = if headers.include?('State')
+                                       Kanrisuru::Core::Socket::TCP_STATE_ABBR[values.shift]
+                                     else
+                                       state
+                                     end
+
+                socket_stats.receive_queue = values.shift.to_i
+                socket_stats.send_queue = values.shift.to_i
+
+                address, port = parse_address_port(values)
+                socket_stats.local_address = address
+                socket_stats.local_port = port
+
+                address, port = parse_address_port(values)
+                socket_stats.peer_address = address
+                socket_stats.peer_port = port
+
+                socket_stats.memory = parse_memory(values.shift)
+
+                rows << socket_stats
+              end
+
+              rows
+            end
+
+            def parse_memory(string)
+              return if Kanrisuru::Util.blank?(string) ||
+                        Regexp.new(/skmem:\((\S+)\)/).match(string).nil?
+
+              _, string = string.split(/skmem:\((\S+)\)/)
+              values = string.split(',')
+
+              memory = Kanrisuru::Core::Socket::StatisticsMemory.new
+              memory.rmem_alloc  = values[0].split(/(\d+)/)[1].to_i
+              memory.rcv_buf     = values[1].split(/(\d+)/)[1].to_i
+              memory.wmem_alloc  = values[2].split(/(\d+)/)[1].to_i
+              memory.snd_buf     = values[3].split(/(\d+)/)[1].to_i
+              memory.fwd_alloc   = values[4].split(/(\d+)/)[1].to_i
+              memory.wmem_queued = values[5].split(/(\d+)/)[1].to_i
+              memory.ropt_mem    = values[6].split(/(\d+)/)[1].to_i
+              memory.back_log    = values[7].split(/(\d+)/)[1].to_i
+              memory.sock_drop   = values[8].split(/(\d+)/)[1].to_i
+
+              memory
+            end
+
+            def parse_address_port(values)
+              address = values.shift
+              port = nil
+
+              if address == '*' && Regexp.new(/skmem:\((\S+)\)/).match(values[0])
+                port = '*'
+              elsif Regexp.new(/\[(\S+)\]/).match(address)
+                tokens = address.split(/\[(\S+)\]/)
+                address = "[#{tokens[1]}]"
+                _, port = tokens[2].split(/:(\S+)/)
+              elsif Regexp.new(/:\S+/).match(address)
+                address, port = address.split(/:(\S+)/)
+              else
+                port = values.shift
+              end
+
+              port = Kanrisuru::Util.numeric?(port) ? port.to_i : port
+              [address, port]
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kanrisuru/core/socket/types.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Kanrisuru
+  module Core
+    module Socket
+      Statistics = Struct.new(
+        :netid, :state, :receive_queue, :send_queue,
+        :local_address, :local_port, :peer_address, :peer_port,
+        :memory
+      )
+
+      StatisticsMemory = Struct.new(
+        :rmem_alloc, :rcv_buf, :wmem_alloc, :snd_buf,
+        :fwd_alloc, :wmem_queued, :ropt_mem, :back_log, :sock_drop
+      )
+    end
+  end
+end

data/lib/kanrisuru/core/socket.rb

@@ -2,176 +2,17 @@
 
 require 'ipaddr'
 
+require_relative 'socket/constants'
+require_relative 'socket/types'
+require_relative 'socket/parser'
+require_relative 'socket/commands'
+
 module Kanrisuru
   module Core
     module Socket
       extend OsPackage::Define
 
       os_define :linux, :ss
-
-      Statistics = Struct.new(
-        :netid, :state, :receive_queue, :send_queue,
-        :local_address, :local_port, :peer_address, :peer_port,
-        :memory
-      )
-
-      StatisticsMemory = Struct.new(
-        :rmem_alloc, :rcv_buf, :wmem_alloc, :snd_buf,
-        :fwd_alloc, :wmem_queued, :ropt_mem, :back_log, :sock_drop
-      )
-
-      TCP_STATES = %w[
-        established syn-sent syn-recv
-        fin-wait-1 fin-wait-2 time-wait
-        closed close-wait last-ack listening closing
-      ].freeze
-
-      OTHER_STATES = %w[
-        all connected synchronized bucket syn-recv
-        big
-      ].freeze
-
-      TCP_STATE_ABBR = {
-        'ESTAB' => 'established', 'LISTEN' => 'listening', 'UNCONN' => 'unconnected',
-        'SYN-SENT' => 'syn-sent', 'SYN-RECV' => 'syn-recv', 'FIN-WAIT-1' => 'fin-wait-1',
-        'FIN-WAIT-2' => 'fin-wait-2', 'TIME-WAIT' => 'time-wait', 'CLOSE-WAIT' => 'close-wait',
-        'LAST-ACK' => 'last-ack', 'CLOSING' => 'closing'
-      }.freeze
-
-      NETWORK_FAMILIES = %w[
-        unix inet inet6 link netlink
-      ].freeze
-
-      def ss(opts = {})
-        state = opts[:state]
-        expression = opts[:expression]
-        family = opts[:family]
-
-        command = Kanrisuru::Command.new('ss')
-
-        command.append_flag('-a')
-        command.append_flag('-m')
-
-        command.append_flag('-n', opts[:numeric])
-        command.append_flag('-t', opts[:tcp])
-        command.append_flag('-u', opts[:udp])
-        command.append_flag('-x', opts[:unix])
-        command.append_flag('-w', opts[:raw])
-
-        if Kanrisuru::Util.present?(family)
-          raise ArgumentError, 'invalid family type' unless NETWORK_FAMILIES.include?(family)
-
-          command.append_arg('-f', family)
-        end
-
-        if Kanrisuru::Util.present?(state)
-          raise ArgumentError, 'invalid filter state' if !TCP_STATES.include?(state) && !OTHER_STATES.include?(state)
-
-          command.append_arg('state', state)
-        end
-
-        command << expression if Kanrisuru::Util.present?(expression)
-
-        execute_shell(command)
-
-        Kanrisuru::Result.new(command) do |cmd|
-          rows = []
-
-          ## New lines with tabs are from text overflow
-          ## on on stdout of SS command.
-          ## Replace the newline and tab chars with a space.
-          string = cmd.raw_result.join
-          string = string.gsub("\n\t", "\s")
-          lines = string.lines.map(&:strip)
-
-          headers = lines.shift
-
-          lines.each do |line|
-            values = line.split
-            next if values.length < 5
-
-            socket_stats = Statistics.new
-            socket_stats.netid =
-              if headers.include?('Netid')
-                values.shift
-              elsif opts[:tcp]
-                'tcp'
-              elsif opts[:udp]
-                'udp'
-              elsif opts[:raw]
-                'raw'
-              else
-                ''
-              end
-
-            socket_stats.state = if headers.include?('State')
-                                   TCP_STATE_ABBR[values.shift]
-                                 else
-                                   state
-                                 end
-
-            socket_stats.receive_queue = values.shift.to_i
-            socket_stats.send_queue = values.shift.to_i
-
-            address, port = parse_address_port(values)
-            socket_stats.local_address = address
-            socket_stats.local_port = port
-
-            address, port = parse_address_port(values)
-            socket_stats.peer_address = address
-            socket_stats.peer_port = port
-
-            socket_stats.memory = parse_memory(values.shift)
-
-            rows << socket_stats
-          end
-
-          rows
-        end
-      end
-
-      private
-
-      def parse_memory(string)
-        return if Kanrisuru::Util.blank?(string) ||
-                  Regexp.new(/skmem:\((\S+)\)/).match(string).nil?
-
-        _, string = string.split(/skmem:\((\S+)\)/)
-        values = string.split(',')
-
-        memory = StatisticsMemory.new
-        memory.rmem_alloc  = values[0].split(/(\d+)/)[1].to_i
-        memory.rcv_buf     = values[1].split(/(\d+)/)[1].to_i
-        memory.wmem_alloc  = values[2].split(/(\d+)/)[1].to_i
-        memory.snd_buf     = values[3].split(/(\d+)/)[1].to_i
-        memory.fwd_alloc   = values[4].split(/(\d+)/)[1].to_i
-        memory.wmem_queued = values[5].split(/(\d+)/)[1].to_i
-        memory.ropt_mem    = values[6].split(/(\d+)/)[1].to_i
-        memory.back_log    = values[7].split(/(\d+)/)[1].to_i
-        memory.sock_drop   = values[8].split(/(\d+)/)[1].to_i
-
-        memory
-      end
-
-      def parse_address_port(values)
-        address = values.shift
-        port = nil
-
-        if address == '*' && Regexp.new(/skmem:\((\S+)\)/).match(values[0])
-          port = '*'
-        elsif Regexp.new(/\[(\S+)\]/).match(address)
-          tokens = address.split(/\[(\S+)\]/)
-          address = "[#{tokens[1]}]"
-          _, port = tokens[2].split(/:(\S+)/)
-        elsif Regexp.new(/:\S+/).match(address)
-          address, port = address.split(/:(\S+)/)
-        else
-          port = values.shift
-        end
-
-        port = Kanrisuru::Util.numeric?(port) ? port.to_i : port
-        [address, port]
-      end
     end
   end
 end

data/lib/kanrisuru/core/stat/commands/stat.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+module Kanrisuru
+  module Core
+    module Stat
+      def dir?(path)
+        file_type?(path, 'directory')
+      end
+
+      def file?(path)
+        file_type?(path, 'regular file')
+      end
+
+      def empty_file?(path)
+        file_type?(path, 'regular empty file')
+      end
+
+      def block_device?(path)
+        file_type?(path, 'block special file')
+      end
+
+      def char_device?(path)
+        file_type?(path, 'character special file')
+      end
+
+      def symlink?(path)
+        file_type?(path, 'symbolic link')
+      end
+
+      def inode?(path)
+        command = Kanrisuru::Command.new("stat #{path}")
+        command.append_arg('-c', '%i')
+        execute(command)
+        command.success?
+      end
+
+      def file_type?(path, type)
+        command = Kanrisuru::Command.new("stat #{path}")
+        command.append_arg('-c', '%F')
+
+        execute(command)
+
+        result = Kanrisuru::Result.new(command, &:to_s)
+
+        result.success? ? result.data == type : false
+      end
+
+      def stat(path, opts = {})
+        command = Kanrisuru::Command.new('stat')
+        command.append_flag('-L', opts[:follow])
+        command.append_arg('-c', '%A,%b,%D,%F,%g,%G,%h,%i,%n,%s,%u,%U,%x,%y,%z')
+        command << path
+
+        execute(command)
+
+        Kanrisuru::Result.new(command) do |cmd|
+          Parser::Stat.parse(cmd)
+        end
+      end
+    end
+  end
+end

data/lib/kanrisuru/core/stat/commands.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require_relative 'commands/stat'

data/lib/kanrisuru/core/stat/parser.rb

@@ -0,0 +1,3 @@
+# frozen_string_literal: true
+
+require_relative 'parsers/stat'

data/lib/kanrisuru/core/stat/parsers/stat.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module Kanrisuru
+  module Core
+    module Stat
+      module Parser
+        class Stat
+          def self.parse(command)
+            string = command.to_s
+            values = string.split(',')
+
+            Kanrisuru::Core::Stat::FileStat.new(
+              Kanrisuru::Mode.new(values[0]),
+              values[1].to_i,
+              values[2],
+              values[3],
+              values[4].to_i,
+              values[5],
+              values[6].to_i,
+              values[7].to_i,
+              values[8],
+              values[9].to_i,
+              values[10].to_i,
+              values[11],
+              values[12] ? DateTime.parse(values[12]) : nil,
+              values[13] ? DateTime.parse(values[13]) : nil,
+              values[14] ? DateTime.parse(values[14]) : nil
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kanrisuru/core/stat/types.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module Kanrisuru
+  module Core
+    module Stat
+      FileStat = Struct.new(
+        :mode, :blocks, :device, :file_type,
+        :gid, :group, :links, :inode,
+        :path, :fsize, :uid, :user,
+        :last_access, :last_modified, :last_changed
+      )
+    end
+  end
+end

data/lib/kanrisuru/core/stat.rb

@@ -2,6 +2,10 @@
 
 require 'date'
 
+require_relative 'stat/types'
+require_relative 'stat/parser'
+require_relative 'stat/commands'
+
 module Kanrisuru
   module Core
     module Stat
@@ -15,87 +19,6 @@ module Kanrisuru
       os_define :linux, :file_type?
       os_define :linux, :inode?
       os_define :linux, :stat
-
-      FileStat = Struct.new(
-        :mode, :blocks, :device, :file_type,
-        :gid, :group, :links, :inode,
-        :path, :fsize, :uid, :user,
-        :last_access, :last_modified, :last_changed
-      )
-
-      def dir?(path)
-        file_type?(path, 'directory')
-      end
-
-      def file?(path)
-        file_type?(path, 'regular file')
-      end
-
-      def empty_file?(path)
-        file_type?(path, 'regular empty file')
-      end
-
-      def block_device?(path)
-        file_type?(path, 'block special file')
-      end
-
-      def char_device?(path)
-        file_type?(path, 'character special file')
-      end
-
-      def symlink?(path)
-        file_type?(path, 'symbolic link')
-      end
-
-      def inode?(path)
-        command = Kanrisuru::Command.new("stat #{path}")
-        command.append_arg('-c', '%i')
-        execute(command)
-        command.success?
-      end
-
-      def file_type?(path, type)
-        command = Kanrisuru::Command.new("stat #{path}")
-        command.append_arg('-c', '%F')
-
-        execute(command)
-
-        result = Kanrisuru::Result.new(command, &:to_s)
-
-        result.success? ? result.data == type : false
-      end
-
-      def stat(path, opts = {})
-        command = Kanrisuru::Command.new('stat')
-        command.append_flag('-L', opts[:follow])
-        command.append_arg('-c', '%A,%b,%D,%F,%g,%G,%h,%i,%n,%s,%u,%U,%x,%y,%z')
-        command << path
-
-        execute(command)
-
-        Kanrisuru::Result.new(command) do |cmd|
-          string = cmd.to_s
-          values = string.split(',')
-
-          FileStat.new(
-            Kanrisuru::Mode.new(values[0]),
-            values[1].to_i,
-            values[2],
-            values[3],
-            values[4].to_i,
-            values[5],
-            values[6].to_i,
-            values[7].to_i,
-            values[8],
-            values[9].to_i,
-            values[10].to_i,
-            values[11],
-            values[12] ? DateTime.parse(values[12]) : nil,
-            values[13] ? DateTime.parse(values[13]) : nil,
-            values[14] ? DateTime.parse(values[14]) : nil
-          )
-        end
-      end
     end
   end
 end

data/lib/kanrisuru/core/stream/commands/cat.rb

@@ -0,0 +1,25 @@
+# frozen_string_literal: true
+
+module Kanrisuru
+  module Core
+    module Stream
+      def cat(files, opts = {})
+        command = Kanrisuru::Command.new('cat')
+        command.append_flag('-T', opts[:show_tabs])
+        command.append_flag('-n', opts[:number])
+        command.append_flag('-s', opts[:squeeze_blank])
+        command.append_flag('-v', opts[:show_nonprinting])
+        command.append_flag('-E', opts[:show_ends])
+        command.append_flag('-b', opts[:number_nonblank])
+        command.append_flag('-A', opts[:show_all])
+
+        command.append_array(files)
+
+        append_file(command, opts)
+        execute_shell(command)
+
+        Kanrisuru::Result.new(command, &:to_a)
+      end
+    end
+  end
+end

data/lib/kanrisuru/core/stream/commands/echo.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+module Kanrisuru
+  module Core
+    module Stream
+      def echo(text, opts = {})
+        command = Kanrisuru::Command.new('echo')
+        command.append_flag('-e', opts[:backslash])
+        command << "'#{text}'"
+
+        append_file(command, opts)
+        execute_shell(command)
+
+        Kanrisuru::Result.new(command) do |cmd|
+          Parser::Echo.parse(cmd, opts)
+        end
+      end
+    end
+  end
+end

data/lib/kanrisuru/core/stream/commands/head.rb

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module Kanrisuru
+  module Core
+    module Stream
+      def head(path, opts = {})
+        command = Kanrisuru::Command.new('head')
+        command.append_arg('-c', opts[:bytes])
+        command.append_arg('-n', opts[:lines])
+        command << path
+
+        execute_shell(command)
+
+        Kanrisuru::Result.new(command, &:to_a)
+      end
+    end
+  end
+end