diff options
Diffstat (limited to 'core/lib/retriever/field_generator.rb')
-rw-r--r-- | core/lib/retriever/field_generator.rb | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/core/lib/retriever/field_generator.rb b/core/lib/retriever/field_generator.rb new file mode 100644 index 00000000..134f6218 --- /dev/null +++ b/core/lib/retriever/field_generator.rb @@ -0,0 +1,39 @@ +# -*- coding: utf-8 -*- + +class Retriever::FieldGenerator + def initialize(model_klass) + @model_klass = model_klass + end + + def int(field_name, required: false) + @model_klass.add_field(field_name, type: :int, required: required) + end + + def string(field_name, required: false) + @model_klass.add_field(field_name, type: :string, required: required) + end + + def bool(field_name, required: false) + @model_klass.add_field(field_name, type: :bool, required: required) + end + + def time(field_name, required: false) + @model_klass.add_field(field_name, type: :time, required: required) + end + + def uri(field_name, required: false) + @model_klass.add_field(field_name, type: :uri, required: required) + end + + def has(field_name, type, required: false) + @model_klass.add_field(field_name, type: type, required: required) + end +end + + + + + + + + |