lib/brief/briefcase.rb in brief-1.8.9 vs lib/brief/briefcase.rb in brief-1.8.10

@@ -66,14 +66,11 @@
       model_settings[:rendered] = !!(params.key?(:rendered))
       model_settings[:content] = !!(params.key?(:content))
 
       all = all_models.compact
 
-      schema = all.map(&:class).uniq.compact
-                 .map(&:to_schema)
-                 .reduce({}) {|m, k| m[k[:type_alias]] = k; m }
-
+      schema = schema_map
       models = all.map {|m| m.as_json(model_settings) }
 
       {
         views: Brief.views.keys,
         key: briefcase.folder_name.to_s.parameterize,
@@ -93,9 +90,15 @@
 
     def use(module_type=:app, module_id)
       options[:app] = module_id.to_s
 
       run(app_config_path) if app_path.try(&:exist?)
+    end
+
+    def schema_map(include_all=false)
+      list = include_all ? Brief::Model.classes : model_classes
+      list.map(&:to_schema)
+        .reduce({}) {|m, k| m[k[:type_alias]] = k; m }
     end
 
     def data
       @data ||= data!
     end