package.rb in epub-parser-0.1.1

- old
+ new
@@ -1,42 +1,45 @@
-%w[ metadata manifest spine guide bindings ].each { |f| require "epub/publication/package/#{f}" }
-
 module EPUB
   module Publication
     class Package
-      attr_accessor :book,
-                    :version, :unique_identifier_id, :prefix, :xml_lang, :dir, :id
-      attr_reader :metadata, :manifest, :spine, :guide, :bindings
-      alias lang  xml_lang
-      alias lang= xml_lang=
+      CONTENT_MODELS = [:metadata, :manifest, :spine, :guide, :bindings]
+      RESERVED_VOCABULARY_PREFIXES = {
+        ''        => 'http://idpf.org/epub/vocab/package/#',
+        'dcterms' => 'http://purl.org/dc/terms/',
+        'marc'    => 'http://id.loc.gov/vocabulary/',
+        'media'   => 'http://www.idpf.org/epub/vocab/overlays/#',
+        'onix'    => 'http://www.editeur.org/ONIX/book/codelists/current.html#',
+        'xsd'     => 'http://www.w3.org/2001/XMLSchema#'
+      }
 
-      def metadata=(metadata)
-        metadata.package = self
-        @metadata = metadata
-      end
 
-      def manifest=(manifest)
-        manifest.package = self
-        @manifest = manifest
+      class << self
+        def define_content_model(model_name)
+          define_method "#{model_name}=" do |model|
+            current_model = __send__(model_name)
+            current_model.package = nil if current_model
+            model.package = self
+            instance_variable_set "@#{model_name}", model
+          end
+        end
       end
 
-      def spine=(spine)
-        spine.package = self
-        @spine = spine
-      end
+      attr_accessor :book,
+                    :version, :prefix, :xml_lang, :dir, :id
+      attr_reader *CONTENT_MODELS
+      alias lang  xml_lang
+      alias lang= xml_lang=
 
-      def guide=(guide)
-        guide.package = self
-        @guide = guide
+      CONTENT_MODELS.each do |model|
+        define_content_model model
       end
 
-      def bindings=(bindings)
-        bindings.package = self
-        @buindings = bindings
-      end
-
       def unique_identifier
         @metadata.unique_identifier
       end
     end
   end
+end
+
+EPUB::Publication::Package::CONTENT_MODELS.each do |f|
+  require_relative "package/#{f}"
 end