inspec/lib/verify/specfile.rb

# encoding: utf-8
# copyright: 2015, Dominik Richter
# license: All rights reserved
require 'vulcano/base_rule'
require 'vulcano/log'
require 'verify/dummy'
# the user may use dynamic evaluations via pry
begin
  require 'pry'
rescue LoadError
end

module Vulcano
  class DummyContext
    include Vulcano::DSL

    def initialize path, profile_id
      @path = path
      @profile_id = profile_id
      @rules = {}
      @errors = []
    end

    def only_if *a, &b
      # always ignore
    end

    # DSL methods
    def __register_rule r
      fid = VulcanoBaseRule.full_id r, @profile_id
      if @rules[fid].nil?
        @rules[fid] = r
      else
        @errors.push "Duplicate definition of rule #{fid}."
      end
    end
    def __unregister_rule id
      fid = "#{@profile_id}/#{id}"
      if @rules.key? fid
        @rules.delete(fid)
      else
        @errors.push "Failed to skip rule #{fid}, it isn't defined."
      end
    end

    # redefine require to find files in the context
    # of this profile
    def require sth
      # ignore vulcano includes, we already have those
      lib = File::expand_path( File.join @path, '..', '..', 'lib', "#{sth}.rb" )
      if File::file? lib
        require_relative lib
      end
    end

    def method_missing sth, *args
      @errors.push "Don't understand method #{sth} ( #{args} )."
    end

  end
end


module Vulcano
  class SpecFile
    Log = ::Vulcano::Log.new()

    attr_reader :errors, :rules
    def initialize path, metadata
      @rules = []
      @raw = File::read(path)
      @profile_id = metadata.dict['name']
      @invalid_calls = []

      ctx = DummyContext.new(path, @profile_id)
      ctx.instance_eval(@raw, path, 1)
      @errors = ctx.instance_variable_get(:@errors)
      @rules = ctx.instance_variable_get(:@rules)
    end

    def metadata
      header = @raw.sub(/^[^#].*\Z/m,'')
      {
        "title" => mOr(header.match(/^# title: (.*)$/), 'untitled'),
        "copyright" => mOr(header.match(/^# copyright: (.*)$/), 'All rights reserved'),
        "rules" => rules2dict(@rules)
      }
    end

    def self.from_file path, metadata
      if !File::file?(path)
        Log.error "Can't find spec file in #{path}"
        return nil
      end
      return SpecFile.new(path, metadata)
    end

    private

    def rule2dict(rule)
      d = nil
      d = rule.desc.gsub(/\s*\n\s*/, ' ').strip unless rule.desc.nil?
      {
        "impact" => rule.impact,
        "title"  => rule.title,
        "desc"   => d
      }
    end

    def rules2dict(rules)
      res = {}
      rules.map do |id, rule|
        nu = rule2dict(rule)
        if res[id].nil?
          res[id] = nu
        else
          Log.error(
            "Not redefining rule id #{id}:\n"+
            "-- #{res[id]}\n"+
            "++ #{nu}\n"
          )
        end
      end
      res
    end

    def mOr(m, other)
      (m.nil? || m[1].nil?) ? other : m[1]
    end

  end
end

def os *a, &b
  {}
end