module Locomotive module Import class Pages < Base def process context[:done] = {} # initialize the hash storing pages already processed self.add_index_and_404 Dir[File.join(theme_path, 'templates', '**/*')].each do |template_path| fullpath = template_path.gsub(File.join(theme_path, 'templates'), '').gsub('.liquid', '').gsub(/^\//, '') next if %w(index 404).include?(fullpath) self.add_page(fullpath) end # make sure all the pages were processed (redirection pages without template for instance) self.pages.each { |fullpath, attributes| self.add_page_without_template(fullpath.to_s) } end protected def add_page_without_template(fullpath) page = context[:done][fullpath] return page if page # already added, so skip it self._save_page!(fullpath, nil) end def add_page(fullpath) page = context[:done][fullpath] return page if page # already added, so skip it template = File.read(File.join(theme_path, 'templates', "#{fullpath}.liquid")) rescue "Unable to find #{fullpath}.liquid" self.replace_images!(template) self.build_parent_template(template) self._save_page!(fullpath, template) end def _save_page!(fullpath, template) parent = self.find_parent(fullpath) attributes = { :title => fullpath.split('/').last.humanize, :slug => fullpath.split('/').last, :parent => parent, :raw_template => template, :published => true }.merge(self.pages[fullpath] || {}).symbolize_keys # templatized ? if content_type_slug = attributes.delete(:content_type) attributes.merge!({ :templatized => true, :content_type => site.content_types.where(:slug => content_type_slug).first }) end # redirection page ? attributes[:redirect] = true if attributes[:redirect_url].present? page = site.pages.where(:fullpath => self.sanitize_fullpath(fullpath)).first || site.pages.build page.attributes = attributes page.save! self.log "adding #{page.fullpath} (#{template.blank? ? 'without' : 'with'} template) / #{page.position}" site.reload context[:done][fullpath] = page page end def build_parent_template(template) # just check if the template contains the extends keyword fullpath = template.scan(/\{% extends (\w+) %\}/).flatten.first if fullpath # inheritance detected fullpath.gsub!("'", '') return if fullpath == 'parent' self.add_page(fullpath) end end def find_parent(fullpath) segments = fullpath.split('/') return site.pages.root.first if segments.size == 1 segments.pop parent_fullpath = segments.join('/').gsub(/^\//, '') # look for a local index page in db parent = site.pages.where(:fullpath => parent_fullpath).first parent || self.add_page(parent_fullpath) end def add_index_and_404 %w(index 404).each_with_index do |slug, position| page = site.pages.where({ :slug => slug, :depth => 0 }).first page ||= sites.pages.build(:slug => slug, :parent => nil) template = File.read(File.join(theme_path, 'templates', "#{slug}.liquid")) self.replace_images!(template) page.attributes = { :raw_template => template }.merge(self.pages[slug] || {}) page.position = position page.save! rescue nil # TODO better error handling site.reload context[:done][slug] = page end end def replace_images!(template) return if template.blank? template.gsub!(/\/samples\/(.*\.[a-zA-Z0-9]{3})/) do |match| name = $1 collection = AssetCollection.find_or_create_internal(site) if asset = collection.assets.detect { |a| a.source_filename == name } asset.source.url else match end end end def pages @pages ||= self.retrieve_pages end def retrieve_pages pages = context[:database]['site']['pages'] if pages.is_a?(Array) # ordered list of pages tmp, positions = {}, Hash.new(0) pages.each do |data| position = nil fullpath = data.keys.first.to_s unless %w(index 404).include?(fullpath) (segments = fullpath.split('/')).pop position_key = segments.empty? ? 'index' : segments.join('/') position = positions[position_key] positions[position_key] += 1 end attributes = (data.values.first || {}).merge(:position => position) tmp[fullpath] = attributes end pages = tmp end pages end def sanitize_fullpath(fullpath) fullpath.gsub(/\/template$/, '/content_type_template') end end end end