cuboid 0.0.0 → 0.0.1alpha

data/lib/cuboid/support/mixins/terminal.rb

@@ -0,0 +1,57 @@
+require 'io/console'
+
+module Cuboid
+module Support
+module Mixins
+
+# Terminal manipulation methods.
+module Terminal
+
+    # Clears the line before printing using `puts`.
+    #
+    # @param    [String]    str
+    #   String to output
+    def reputs( str = '' )
+        reprint str + "\n"
+    end
+
+    # Clears the line before printing.
+    #
+    # @param    [String]    str
+    #   String to output.
+    def reprint( str = '' )
+        print restr( str )
+    end
+
+    def restr( str = '' )
+        "\e[0K" + str
+    end
+
+    # Clear the bottom of the screen
+    def clear_screen
+        print "\e[2J"
+    end
+
+    def empty_screen
+        move_to_home
+        rows, cols = $stdin.winsize
+        (rows - 1).times{ puts ' ' * cols }
+        move_to_home
+    end
+
+    # Moves cursor top left to its home
+    def move_to_home
+        print "\e[H"
+    end
+
+    # Flushes the STDOUT buffer
+    def flush
+        $stdout.flush
+    end
+
+    extend self
+end
+
+end
+end
+end

data/lib/cuboid/system.rb

@@ -0,0 +1,119 @@
+require_relative 'system/slots'
+
+module Cuboid
+
+class System
+    include Singleton
+
+    # @return   [Array<Platforms::Base>]
+    attr_reader :platforms
+
+    # @return   [Slots]
+    attr_reader :slots
+
+    def initialize
+        @platforms = []
+        @slots     = Slots.new( self )
+    end
+
+    # @return   [Float]
+    #   System utilization based on slots.
+    #
+    #   * `0.0` => No utilization.
+    #   * `1.0` => Max utilization.
+    def utilization
+        total_slots = System.slots.total
+        return 1.0 if total_slots == 0
+
+        System.slots.used / Float( total_slots )
+    end
+
+    # @return   [Bool]
+    def max_utilization?
+        utilization == 1
+    end
+
+    # @return   [Integer]
+    #   Amount of free RAM in bytes.
+    def memory_free
+        platform.memory_free
+    end
+
+    # @param    [Integer]   pgid
+    #   Process group ID.
+    #
+    # @return   [Integer]
+    #   Amount of RAM in bytes used by the given GPID.
+    def memory_for_process_group( pgid )
+        platform.memory_for_process_group( pgid )
+    end
+
+    # @return   [Integer]
+    #   Amount of free disk space in bytes.
+    def disk_space_free
+        platform.disk_space_free
+    end
+
+    # @return   [String
+    #   Location for temporary file storage.
+    def disk_directory
+        platform.disk_directory
+    end
+
+    # @param    [Integer]   pid
+    #   Process ID.
+    #
+    # @return   [Integer]
+    #   Amount of disk space in bytes used by the given PID.
+    def disk_space_for_process( pid )
+        platform.disk_space_for_process( pid )
+    end
+
+    # @return   [Integer]
+    #   Amount of CPU cores.
+    def cpu_count
+        @cpu_count ||= platform.cpu_count
+    end
+
+    # @return   [Platforms::Base]
+    def platform
+        return @platform if @platform
+
+        platforms.each do |klass|
+            next if !klass.current?
+
+            return @platform = klass.new
+        end
+
+        raise "Unsupported platform: #{RUBY_PLATFORM}"
+    end
+
+    # @private
+    def register_platform( platform )
+        platforms << platform
+    end
+
+    # @private
+    def reset
+        @cpu_count = nil
+        @platform  = nil
+    end
+
+    class <<self
+        def method_missing( sym, *args, &block )
+            if instance.respond_to?( sym )
+                instance.send( sym, *args, &block )
+            else
+                super( sym, *args, &block )
+            end
+        end
+
+        def respond_to?( *args )
+            super || instance.respond_to?( *args )
+        end
+    end
+
+end
+end
+
+require_relative 'system/platforms'

data/lib/cuboid/system/platforms.rb

@@ -0,0 +1,84 @@
+module Cuboid
+
+class System
+module Platforms
+
+class Base
+    class <<self
+
+        # @private
+        def inherited( platform )
+            System.register_platform platform
+        end
+
+        # @return   [Bool]
+        #   `true` if it's the current platform, `false` otherwise.
+        #
+        # @abstract
+        def current?
+            raise 'Missing implementation'
+        end
+    end
+
+    # @return   [Integer]
+    #   Amount of free RAM in bytes.
+    #
+    # @abstract
+    def memory_free
+        raise 'Missing implementation'
+    end
+
+    # @param    [Integer]   pgid
+    #   Process group ID.
+    #
+    # @return   [Integer]
+    #   Amount of RAM in bytes used by the given GPID.
+    #
+    # @abstract
+    def memory_for_process_group( pgid )
+        raise 'Missing implementation'
+    end
+
+    # @return   [Integer]
+    #   Amount of free disk in bytes.
+    #
+    # @abstract
+    def disk_space_free
+        raise 'Missing implementation'
+    end
+
+    # @return   [String
+    #   Location for temporary file storage.
+    def disk_directory
+        Options.paths.os_tmpdir
+    end
+
+    # @param    [Integer]   pid
+    #   Process ID.
+    #
+    # @return   [Integer]
+    #   Amount of disk in bytes used by the given PID.
+    def disk_space_for_process( pid )
+        Support::Database::Base.disk_space_for( pid )
+    end
+
+    # @return   [Integer]
+    #   Amount of CPU cores.
+    def cpu_count
+        Concurrent.processor_count
+    end
+
+    # @private
+    def _exec( cmd )
+        %x(#{cmd})
+    end
+
+end
+
+end
+end
+end
+
+Dir.glob( "#{File.dirname(__FILE__)}/**/*.rb" ).each do |platform|
+    require platform
+end

data/lib/cuboid/system/platforms/linux.rb

@@ -0,0 +1,26 @@
+require_relative 'mixins/unix'
+
+module Cuboid
+
+class System
+module Platforms
+class Linux < Base
+    include Mixins::Unix
+
+    # @return   [Integer]
+    #   Amount of free RAM in bytes.
+    def memory_free
+        memory.available_bytes
+    end
+
+    class <<self
+        def current?
+            Cuboid.linux?
+        end
+    end
+
+end
+
+end
+end
+end

data/lib/cuboid/system/platforms/mixins/unix.rb

@@ -0,0 +1,46 @@
+require 'vmstat'
+
+module Cuboid
+class System
+module Platforms
+module Mixins
+
+module Unix
+
+    # @param    [Integer]   pgid
+    #   Process group ID.
+    #
+    # @return   [Integer]
+    #   Amount of RAM in bytes used by the given GPID.
+    def memory_for_process_group( pgid )
+        rss = 0
+
+        _exec( "ps -o rss -g #{pgid}" ).split("\n")[1..-1].each do |rss_string|
+            rss += rss_string.to_i
+        end
+
+        rss * pagesize
+    end
+
+    # @return   [Integer]
+    #   Amount of free disk in bytes.
+    def disk_space_free
+        Vmstat.disk( disk_directory ).available_bytes
+    end
+
+    private
+
+    def pagesize
+        @pagesize ||= memory.pagesize
+    end
+
+    def memory
+        Vmstat.memory
+    end
+
+end
+
+end
+end
+end
+end

data/lib/cuboid/system/platforms/osx.rb

@@ -0,0 +1,25 @@
+require_relative 'mixins/unix'
+
+module Cuboid
+
+class System
+module Platforms
+class OSX < Base
+    include Mixins::Unix
+
+    # @return   [Integer]
+    #   Amount of free RAM in bytes.
+    def memory_free
+        pagesize * memory.free
+    end
+
+    class <<self
+        def current?
+            Cuboid.mac?
+        end
+    end
+
+end
+end
+end
+end

data/lib/cuboid/system/platforms/windows.rb

@@ -0,0 +1,81 @@
+module Cuboid
+
+class System
+module Platforms
+
+class Windows < Base
+
+    class <<self
+        def current?
+            Cuboid.windows?
+        end
+    end
+
+    # @return   [Integer]
+    #   Amount of free RAM in bytes.
+    def memory_free
+        result = wmi.ExecQuery(
+            'select AvailableBytes from Win32_PerfFormattedData_PerfOS_Memory'
+        )
+
+        memory = nil
+        result.each do |e|
+            memory = e.availableBytes.to_i
+            e.ole_free
+        end
+        result.ole_free
+
+        memory
+    end
+
+    # @param    [Integer]   pgid
+    #   Process group ID.
+    #
+    # @return   [Integer]
+    #   Amount of RAM in bytes used by the given GPID.
+    def memory_for_process_group( pgid )
+        processes = wmi.ExecQuery(
+            "select PrivatePageCount from win32_process where ProcessID='#{pgid}' or ParentProcessID='#{pgid}'"
+        )
+
+        memory = 0
+        processes.each do |process|
+            # Not actually pages but bytes, no idea why.
+            memory += process.privatePageCount.to_i
+            process.ole_free
+        end
+        processes.ole_free
+
+        memory
+    end
+
+    # @return   [Integer]
+    #   Amount of free disk in bytes.
+    def disk_space_free
+        device_id = disk_directory.split( '/' ).first
+
+        drives = wmi.ExecQuery(
+            "select FreeSpace from win32_LogicalDisk where DeviceID='#{device_id}'"
+        )
+
+        space = nil
+        drives.each do |drive|
+            space = drive.freeSpace.to_i
+            drive.ole_free
+        end
+        drives.ole_free
+
+        space
+    end
+
+    private
+
+    def wmi
+        @wmi ||= WIN32OLE.connect( 'winmgmts://' )
+    end
+
+end
+end
+end
+
+end