orbit4-5/app/controllers/admin/import_controller.rb

465 lines
16 KiB
Ruby

# encoding: utf-8
require "net/http"
require "uri"
require 'json'
require 'yaml'
class Admin::ImportController < OrbitAdminController
layout "structure"
before_filter :get_user
@@import_stats = {}
@@thread = nil
def check_url
params['url'] = params['url']+"/" if params['url'].last!="/"
uri = URI.parse(params['url'])
begin
@host = uri.host
http = Net::HTTP.new(uri.host, uri.port)
request = Net::HTTP::Get.new(uri.request_uri)
response = http.request(request)
data = response.body
data = JSON.parse(data)
msg = data['status']
rescue
msg = "Failed to connect to RSS2 (#{uri.to_s})"
end
render :json=>{"status"=>msg}
end
def module_categories
module_app = ModuleApp.find_by(:key=>params['module'])
uri = URI.parse(params['url'])
http = Net::HTTP.new(uri.host, uri.port)
request = Net::HTTP::Get.new(uri.request_uri)
response = http.request(request)
data = response.body
categories = JSON.parse(data)
I18n.locale = :zh_tw
categories.each do |id, category|
old_cate = module_app.categories.where(:title => category["zh_tw"]).first
if old_cate.nil?
cat = Category.new
cat.title_translations = category
cat.save!
module_app.categories << cat
categories[id]['id'] = cat.id.to_s
else
categories[id]['id'] = old_cate.id.to_s
end
end
render :json=>categories
end
def module_data_list
uri = URI.parse(params['url'])
http = Net::HTTP.new(uri.host, uri.port)
request = Net::HTTP::Get.new(uri.request_uri)
response = http.request(request)
data = response.body
data = JSON.parse(data)
render :json=>data
end
def module_data
uri = URI.parse(params['url'])
@host = uri.host
http = Net::HTTP.new(uri.host, uri.port)
request = Net::HTTP::Get.new(uri.request_uri)
response = http.request(request)
data = response.body
data = JSON.parse(data)
case params['module']
when 'archive'
import_archive(data)
when 'announcement'
import_announcement(data)
end
render :json=>['status'=>'ok']
end
def import_archive(file)
archive_file = ArchiveFile.where(:rss2_sn=>file['Sn']).first
if archive_file.nil?
archive_file = ArchiveFile.new
archive_file.rss2_sn = file['Sn']
else
archive_file.archive_file_multiples.destroy_all
end
archive_file.title_translations = file['title']
archive_file.sort_number = file['sort_number']
archive_file.category = Category.find(params["category"])
archive_file.create_user_id = current_user.id
archive_file.update_user_id = current_user.id
archive_file.save
if(file['url'])
archive = ArchiveFileMultiple.new({
:file_title_translations=>file['title'],
:choose_lang => @site_valid_locales,
:remote_file_url => file['url'],
:sort_number=>file['sort_number']
})
archive.save
archive_file.archive_file_multiples << archive
end
if(file['url2'])
archive = ArchiveFileMultiple.new({
:file_title_translations=>file['title'],
:choose_lang => @site_valid_locales,
:remote_file_url => file['url2'],
:sort_number=>file['sort_number']
})
archive.save
archive_file.archive_file_multiples << archive
end
end
def import_announcement(news)
bulletin = Bulletin.where(:rss2_sn=>news['Sn']).first
if bulletin.nil?
bulletin = Bulletin.new
bulletin.rss2_sn = news['Sn']
else
bulletin.bulletin_links.destroy_all
bulletin.bulletin_files.destroy_all
end
bulletin.update_user = current_user
bulletin.title_translations = news['Title']
bulletin.subtitle_translations = news['Summary']
@site_valid_locales.each do |locale|
locale = locale.to_s
if news['Content'][locale]=="" and news['Summary'][locale]!=""
news['Content'][locale] = news['Summary'][locale]
news['Summary'][locale] = " "
end
temp = news['Content'][locale]
urls = Nokogiri::HTML(temp).css("img").map do |link|
if URI.parse(link.attr("src")).host == @host
link.attr("src")
end
end
urls.each do |url|
next if url.nil?
a = Asset.new
a.remote_data_url = url
a.title_translations = {"en" => a.data.filename, "zh_tw" => a.data.filename}
a.save!
@user.assets << a
temp.gsub!(CGI::escapeHTML(url), a.data.to_s)
end
news['Content'][locale] = temp
end
bulletin.text_translations = news['Content']
bulletin.category = Category.find(params["category"])
bulletin.view_count = news["Visits"].blank? ? 0 : news["Visits"]
bulletin.postdate = news["PostDate"]
bulletin.deadline = news['Deadline']
bulletin.remote_image_url = news["Pic"] if news["Pic"]
if news["URL"] && news['URL'] != ""
bl = BulletinLink.new
bl.url = news["URL"]
bl.title_translations = {"en" => "Link", "zh_tw" => "Link"}
bl.save!
bulletin.bulletin_links << bl
end
news['files'].each do |f|
bf = BulletinFile.new
if f['url'].split('title=').size == 1
f['url'] = f['url']+"檔案下載"
end
f['title'] = "檔案下載" if f['title'].blank?
bf.remote_file_url = f['url']
bf.title_translations = {"en" => f['title'], "zh_tw" => f['title']}
bf.save!
# Rename uploaded file
file_ext = File.extname(f['url'].split('&')[0])
file = File.new("tmp/uploads/#{bf.title}#{file_ext}","w+b")
file.write(bf.file.read)
bf.file.cache!(file)
bf.save!
bulletin.bulletin_files << bf
File.delete(file)
end
bulletin.save!
end
def rss2_pages
delete_import_file
uri = URI.parse(params['url'])
# @@thread = Thread.new do
http = Net::HTTP.new(uri.host, uri.port)
request = Net::HTTP::Get.new(uri.request_uri)
response = http.request(request)
data = response.body
data = JSON.parse(data)
pages = data["pages"]
@@import_stats["total_pages"] = data["total_pages"].to_i
@@import_stats["current_status"] = 0
@@import_stats["success"] = true
write_to_file
pages.each do |page|
@@import_stats["current_page"] = page["page_id"]
@@import_stats["current_page_name"] = page["title"]["zh_tw"]
write_to_file
if Page.where(:rss2_id => page["page_id"]).count == 0
page_id = page["page_id"]
number = 0
while page_present?page_id
page_id = page_id + "_" + (number + 1).to_s
end
p = Page.new(:page_id => page_id, :parent_page_id => Page.root.id, :page_type => "page", :url => "/#{page_id}", :module => "page_content", :enabled_for => ["en", "zh_tw"], :menu_enabled_for => ["en", "zh_tw"], :enabled_for_sitemap => ["en","zh_tw"], :name_translations => {"en"=>page["title"]["en"], "zh_tw" => page["title"]["zh_tw"]},:rss2_id => page["page_id"])
p.save
en = smart_downloader(page["content"]["en"],data["site_url"])
zh_tw = smart_downloader(page["content"]["zh_tw"],data["site_url"])
page_context = PageContext.new(:content_translations => {"en" => en,"zh_tw" => zh_tw}, :version => 1, :update_user_id => current_user.id, :page_id => p.id)
page_context.save
else
p = Page.find_by(:rss2_id => page["page_id"])
page_id = p.page_id
end
@@import_stats["current_status"] = @@import_stats["current_status"] + 1
write_to_file
if !page["childpages"].blank?
childpages = page["childpages"]
childpages.each_with_index do |childpage,i|
@@import_stats["current_page"] = childpage["page_id"]
@@import_stats["current_page_name"] = page["title"]["zh_tw"]
write_to_file
if Page.where(:rss2_id => childpage["page_id"]).count == 0
p_id = page_id + "_" + i.to_s
case childpage["type"]
when "page"
cp = Page.new(:page_id => p_id, :parent_page_id => p.id, :page_type => "page", :url => "/#{p.page_id}/#{p_id}", :module => "page_content", "enabled_for" => ["en", "zh_tw"], :menu_enabled_for => ["en", "zh_tw"], :enabled_for_sitemap => ["en","zh_tw"], :name_translations => {"en"=>childpage["title"]["en"], "zh_tw" => childpage["title"]["zh_tw"]},:rss2_id => childpage["page_id"])
cp.save
en = smart_downloader(childpage["content"]["en"],data["site_url"])
zh_tw = smart_downloader(childpage["content"]["zh_tw"],data["site_url"])
page_context = PageContext.new(:content_translations => {"en" => en,"zh_tw" => zh_tw}, :version => 1, :update_user_id => current_user.id, :page_id => cp.id)
page_context.save
when "exturl"
cp = Page.new(:page_id => p_id, :parent_page_id => p.id, :page_type => "link", "url" => "/#{p.page_id}/#{p_id}", :enabled_for => ["en", "zh_tw"], :menu_enabled_for => ["en", "zh_tw"], :enabled_for_sitemap => ["en","zh_tw"], :name_translations => {"en"=>childpage["title"]["en"], "zh_tw" => childpage["title"]["zh_tw"]}, :external_url_translations => {"en" => childpage["url"], "zh_tw" => childpage["url"]},:rss2_id => childpage["page_id"])
cp.save
when "file"
asset = Asset.new
asset.remote_data_url = childpage["filename"]
asset.title_translations = {"en" => childpage["file_title"], "zh_tw" =>childpage["file_title"]}
asset.save!
@user.assets << asset
@user.save
external_url = asset.data.url
cp = Page.new(:page_id => p_id, :parent_page_id => p.id, :page_type => "link", "url" => "/#{p.page_id}/#{p_id}", :enabled_for => ["en", "zh_tw"], :menu_enabled_for => ["en", "zh_tw"], :enabled_for_sitemap => ["en","zh_tw"], :name_translations => {"en"=>childpage["title"]["en"], "zh_tw" => childpage["title"]["zh_tw"]}, :external_url_translations => {"en" => external_url, "zh_tw" => external_url},:rss2_id => childpage["page_id"])
cp.save
end
end
@@import_stats["current_status"] = @@import_stats["current_status"] + 1
write_to_file
end
end
end
# end
# @@thread.join
# @@thread.abort_on_exception = true
render :json => {"success"=>true}.to_json
end
def get_import_status
settings = get_settings_from_import_file
render :json => settings.to_json
end
def rss2_galleries
delete_import_file
uri = URI.parse(params['url'])
# @@thread = Thread.new do
http = Net::HTTP.new(uri.host, uri.port)
request = Net::HTTP::Get.new(uri.request_uri)
response = http.request(request)
data = response.body
data = JSON.parse(data)
@@import_stats["total_images"] = data["total_images"].to_i
@@import_stats["current_status"] = 0
@@import_stats["success"] = true
write_to_file
albums = data["albums"]
current_locale = I18n.locale
I18n.locale = :en
category = Category.where(:title => "RSS2 Gallery").first
I18n.locale = current_locale
if category.nil?
module_app = ModuleApp.find_by_key("gallery")
category = Category.new
category.title_translations = {"en" => "RSS2 Gallery", "zh_tw" => "RSS2畫廊"}
category.save
module_app.categories << category
end
albums.each do |album|
@@import_stats["current_album_id"] = album["albumid"]
write_to_file
if Album.where(:rss2_id => album["albumid"]).count == 0
@@import_stats["current_album_name"] = album["name"]["zh_tw"]
write_to_file
newalbum = Album.new(:name_translations => {"en" => album["name"]["en"], "zh_tw" => album["name"]["zh_tw"]}, :description_translations => {"en" => album["description"]["en"], "zh_tw" => album["description"]["zh_tw"]},"rss2_id" => album["albumid"])
newalbum.category_id = category.id
newalbum.save
images = album["images"]
if !images.blank?
images.each do |image|
if AlbumImage.where(:rss2_id => image["photoid"]).count == 0
img = newalbum.album_images.new
img.description_translations = {"en" => image["description"]["en"], "zh_tw" => image["description"]["zh_tw"]}
img.remote_file_url = image["image"]
img.save
end
@@import_stats["current_status"] = @@import_stats["current_status"] + 1
write_to_file
end
end
end
end
# end
# @@thread.join
# @@thread.abort_on_exception = true
render :json => {"success" => true}.to_json
end
def rss2_links
delete_import_file
uri = URI.parse(params['url'])
# @@thread = Thread.new do
http = Net::HTTP.new(uri.host, uri.port)
request = Net::HTTP::Get.new(uri.request_uri)
response = http.request(request)
data = response.body
data = JSON.parse(data)
@@import_stats["total_links"] = data["total_links"].to_i
@@import_stats["current_import"] = "categories"
@@import_stats["current_status"] = 0
@@import_stats["success"] = true
write_to_file
categories = data["categories"]
current_locale = I18n.locale
I18n.locale = :zh_tw
module_app = ModuleApp.find_by_key("web_resource")
categories.each do |category|
cat = Category.where(:title => category[1]["zh_tw"], :module_app_id => module_app.id).first
if cat.nil?
cat = Category.new
cat.title_translations = {"en" => category[1]["en"], "zh_tw" => category[1]["zh_tw"]}
cat.save
module_app.categories << cat
end
end
@@import_stats["current_import"] = "links"
write_to_file
links = data["links"]
links.each do |link|
l = WebLink.where(:rss2_id => link["link_id"]).first
if l.nil?
@@import_stats["current_link_id"] = link["link_id"]
@@import_stats["current_link_name"] = link["title"]["zh_tw"]
write_to_file
weblink = WebLink.new(:title_translations => {"en" => link["title"]["en"], "zh_tw" => link["title"]["zh_tw"]}, :context_translations => {:en => link["description"]["en"], :zh_tw => link["description"]["zh_tw"]}, :url_translations => {"en" => link["url"], "zh_tw" => link["url"]}, :rss2_id => link["link_id"])
category = Category.where(:title => link["category_name"], :module_app_id => module_app).first
weblink.category_id = category.id
weblink.save
end
@@import_stats["current_status"] = @@import_stats["current_status"] + 1
write_to_file
end
I18n.locale = current_locale
# end
# @@thread.join
# @@thread.abort_on_exception = true
render :json => {"success" => true}.to_json
end
private
def page_present?(page)
Page.where(:page_id => page).count > 0
end
def smart_downloader(data,url)
@data = data
excluded_extensions = ["php","/"]
@links = URI.extract(@data)
@links.each do |link|
if link.include?url
temp = link.gsub(url,"")
extension = temp.split(".").last rescue nil
if extension.nil? || excluded_extensions.include?(extension)
@data = @data.sub(link,"#")
else
a = Asset.new
a.remote_data_url = link
a.title_translations = {"en" => a.data.filename, "zh_tw" => a.data.filename}
a.save!
@user.assets << a
@user.save
@data = @data.sub(link, a.data.url)
end
end
end
@data
end
def write_to_file
File.open("#{Rails.root.to_s}/public/import_status.yml", "w") do |file|
file.write @@import_stats.to_yaml
end
end
def delete_import_file
@@import_stats = {}
f = File.join(Rails.root,"public","import_status")
if File.exists?f
File.delete(f)
end
end
def get_settings_from_import_file
YAML::load_file "#{Rails.root.to_s}/public/import_status.yml"
end
def get_user
@user = User.find(current_user.id)
end
end