require_relative "base_class"

module Oscal
  class Control < Oscal::BaseClass
    KEY = %i(id klass title params props links parts controls)

    attr_accessor *KEY

    attr_serializable *KEY

    def set_value(key_name, val)
      case key_name
      when "params"
        Parameter.wrap(val)
      when "props"
        Property.wrap(val)
      when "links"
        Link.wrap(val)
      when "parts"
        Part.wrap(val)
      when "controls"
        Control.wrap(val)
      else
        val
      end
    end
  end
end