engine/lib/locomotive/import/pages.rb

188 lines
5.1 KiB
Ruby

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