114 lines
3.3 KiB
Ruby
114 lines
3.3 KiB
Ruby
module Locomotive
|
|
module Import
|
|
module Pages
|
|
|
|
def self.process(context)
|
|
site, pages, theme_path = context[:site], context[:database]['pages'], context[:theme_path]
|
|
|
|
context[:done] = {} # initialize the hash storing pages already processed
|
|
|
|
self.add_index_and_404(context)
|
|
|
|
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, context)
|
|
end
|
|
end
|
|
|
|
def self.add_page(fullpath, context)
|
|
puts "....adding #{fullpath}"
|
|
|
|
page = context[:done][fullpath]
|
|
|
|
return page if page # already added, so skip it
|
|
|
|
site, pages, theme_path = context[:site], context[:database]['site']['pages'], context[:theme_path]
|
|
|
|
template = File.read(File.join(theme_path, 'templates', "#{fullpath}.liquid")) rescue "Unable to find #{fullpath}.liquid"
|
|
|
|
self.build_parent_template(template, context)
|
|
|
|
parent = self.find_parent(fullpath, context)
|
|
|
|
page = site.pages.where(:fullpath => fullpath).first || site.pages.build
|
|
|
|
attributes = {
|
|
:title => fullpath.split('/').last.humanize,
|
|
:slug => fullpath.split('/').last,
|
|
:parent => parent,
|
|
:raw_template => template
|
|
}.merge(pages[fullpath] || {}).symbolize_keys
|
|
|
|
# templatized ?
|
|
if content_type_slug = attributes.delete(:content_type)
|
|
attributes[:content_type] = site.content_types.where(:slug => content_type_slug).first
|
|
end
|
|
|
|
page.attributes = attributes
|
|
|
|
page.save!
|
|
|
|
site.reload
|
|
|
|
context[:done][fullpath] = page
|
|
|
|
page
|
|
end
|
|
|
|
def self.build_parent_template(template, context)
|
|
# 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, context)
|
|
end
|
|
end
|
|
|
|
def self.find_parent(fullpath, context)
|
|
site = context[:site]
|
|
|
|
segments = fullpath.split('/')
|
|
|
|
return site.pages.index.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, context)
|
|
end
|
|
|
|
def self.add_index_and_404(context)
|
|
site, pages, theme_path = context[:site], context[:database]['site']['pages'], context[:theme_path]
|
|
|
|
%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"))
|
|
|
|
page.attributes = { :raw_template => template, :position => position }.merge(pages[slug] || {})
|
|
|
|
page.save! rescue nil # TODO better error handling
|
|
|
|
site.reload
|
|
|
|
context[:done][slug] = page
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|
|
end |