lib/amazing/proc_file.rb

   1 # Copyright 2008 Dag Odenhall <dag.odenhall@gmail.com>
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #    http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 module Amazing
  16
  17   # Parse a /proc file
  18   #
  19   #   cpuinfo = ProcFile.parse_file("cpuinfo")
  20   #   cpuinfo[1]["model name"]
  21   #   #=> "AMD Turion(tm) 64 X2 Mobile Technology TL-50"
  22   class ProcFile
  23     include Enumerable
  24
  25     def self.parse_file(file)
  26       file = File.expand_path(file, "/proc")
  27       new(File.new(file))
  28     end
  29
  30     def initialize(string_or_io)
  31       case string_or_io
  32       when String
  33         content = string_or_io
  34       when IO
  35         content = string_or_io.read
  36         string_or_io.close
  37       end
  38       @list = [{}]
  39       content.each_line do |line|
  40         if sep = line.index(":")
  41           @list[-1][line[0..sep-1].strip] = line[sep+1..-1].strip
  42         else
  43           @list << {}
  44         end
  45       end
  46       @list.pop if @list[-1].empty?
  47     end
  48
  49     def each
  50       @list.each do |section|
  51         yield section
  52       end
  53     end
  54
  55     def [](section)
  56       @list[section]
  57     end
  58   end
  59 end