added exporter

lib/exporter.rb

@@ -0,0 +1,221 @@
+require 'json'
+require 'yaml'
+require 'httparty'
+require 'securerandom'
+
+class Exporter
+  def initialize(user)
+    @user = user
+    @obj = {}
+    @export_dirname = "export_#{@user.screen_name}_#{Time.now.to_i}#{SecureRandom.base64.gsub(/[+=\/]/, '')}"
+    @export_filename = @user.screen_name
+  end
+
+  def export
+    fail "An export is already in progress." if @user.export_processing
+    @user.export_processing = true
+    @user.save validate: false
+    collect_user_info
+    collect_questions
+    collect_answers
+    collect_comments
+    collect_smiles
+    finalize
+    publish
+  rescue => _
+    @user.export_processing = false
+    @user.save validate: false
+  end
+
+  private
+
+  def collect_user_info
+    %i(admin answered_count asked_count ban_reason banned_until bio blogger comment_smiled_count commented_count
+       confirmation_sent_at confirmed_at contributor created_at crop_h crop_h_h crop_h_w crop_h_x crop_h_y
+       crop_w crop_x crop_y current_sign_in_at current_sign_in_ip display_name email follower_count friend_count
+       id last_sign_in_at last_sign_in_ip locale location moderator motivation_header permanently_banned
+       privacy_allow_anonymous_questions privacy_allow_public_timeline privacy_allow_stranger_answers
+       privacy_show_in_search profile_header_content_type profile_header_file_name profile_header_file_size
+       profile_header_updated_at profile_picture_content_type profile_picture_file_name profile_picture_file_size
+       profile_picture_updated_at screen_name show_foreign_themes sign_in_count smiled_count supporter translator
+       updated_at website).each do |f|
+      @obj[f] = @user.send f
+    end
+  end
+
+  def collect_questions
+    @obj[:questions] = []
+    @user.questions.each do |q|
+      @obj[:questions] << process_question(q, include_user: false)
+    end
+  end
+
+  def collect_answers
+    @obj[:answers] = []
+    @user.answers.each do |a|
+      @obj[:answers] << process_answer(a, include_user: false)
+    end
+  end
+
+  def collect_comments
+    @obj[:comments] = []
+    @user.comments.each do |c|
+      @obj[:comments] << process_comment(c, include_user: false, include_answer: true)
+    end
+  end
+
+  def collect_smiles
+    @obj[:smiles] = []
+    @user.smiles.each do |s|
+      @obj[:smiles] << process_smile(s)
+    end
+  end
+
+  def finalize
+    `mkdir -p "#{Rails.root.join "public", "export"}"`
+    `mkdir -p /tmp/rs_export/#{@export_dirname}/pictures`
+
+    if @user.profile_picture_file_name
+      %i(large medium small original).each do |s|
+        url = @user.profile_picture.url(s)
+        target_file = "/tmp/rs_export/#{@export_dirname}/pictures/picture_#{s}_#{@user.profile_picture_file_name}"
+        File.open target_file, 'wb' do |f|
+          f.binmode
+          f.write HTTParty.get(url).parsed_response
+        end
+      end
+    end
+
+    if @user.profile_header_file_name
+      %i(web mobile retina original).each do |s|
+        url = @user.profile_header.url(s)
+        target_file = "/tmp/rs_export/#{@export_dirname}/pictures/header_#{s}_#{@user.profile_header_file_name}"
+        File.open target_file, 'wb' do |f|
+          f.binmode
+          f.write HTTParty.get(url).parsed_response
+        end
+      end
+    end
+
+    File.open "/tmp/rs_export/#{@export_dirname}/#{@export_filename}.json", 'w' do |f|
+      f.puts @obj.to_json
+    end
+
+    File.open "/tmp/rs_export/#{@export_dirname}/#{@export_filename}.yml", 'w' do |f|
+      f.puts @obj.to_yaml
+    end
+
+    File.open "/tmp/rs_export/#{@export_dirname}/#{@export_filename}.xml", 'w' do |f|
+      f.puts @obj.to_xml
+    end
+  end
+
+  def publish
+    `tar czvf #{Rails.root.join "public", "export", "#{@export_dirname}.tar.gz"} -C /tmp/rs_export #{@export_dirname}`
+    url = "https://retrospring.net/export/#{@export_dirname}.tar.gz"
+    @user.export_processing = false
+    @user.export_url = url
+    @user.export_created_at = Time.now
+    @user.save validate: false
+    url
+  end
+
+  def process_question(question, options = {})
+    opts = {
+        include_user: true,
+        include_answers: true
+    }.merge(options)
+
+    qobj = {}
+    %i(answer_count author_is_anonymous content created_at id).each do |f|
+      qobj[f] = q.send f
+    end
+
+    if opts[:include_answers]
+      qobj[:answers] = []
+      qobj.answers.each do |a|
+        qobj[:answers] << process_answer(a, include_question: false)
+      end
+    end
+
+    if opts[:include_user]
+      qobj[:user] = question.author_is_anonymous ? nil : user_stub(question.user)
+    end
+
+    qobj
+  end
+
+  def process_answer(answer, options = {})
+    opts = {
+      include_question: true,
+      include_user: true,
+      include_comments: true
+    }.merge(options)
+
+    aobj = {}
+    %i(comment_count content created_at id smile_count).each do |f|
+      aobj[f] = answer.send f
+    end
+
+    if opts[:include_user]
+      aobj[:user] = user_stub(answer.user)
+    end
+
+    if opts[:include_question]
+      aobj[:question] = process_question(answer.question, include_user: true, include_answers: false)
+    end
+
+    if opts[:include_comments]
+      aobj[:comments] = []
+      answer.comments.each do |c|
+        aobj[:comments] << process_comment(c, include_user: true, include_answer: false)
+      end
+    end
+
+    aobj
+  end
+
+  def process_comment(comment, options = {})
+    opts = {
+        include_user: true,
+        include_answer: false
+    }.merge(options)
+
+    cobj = {}
+    %i(content created_at id).each do |f|
+      cobj[f] = comment.send f
+    end
+
+    if opts[:include_user]
+      cobj[:user] = user_stub(comment.user)
+    end
+
+    if opts[:include_answer]
+      cobj[:answer] = process_answer(comment.answer, include_comments: false)
+    end
+
+    cobj
+  end
+
+  def process_smile(smile)
+    sobj = {}
+
+    %i(id created_at).each do |f|
+      sobj[f] = s.send f
+    end
+
+    sobj[:answer] = process_answer(comment.answer, include_comments: false)
+
+    sobj
+  end
+
+  def user_stub(user)
+    uobj = {}
+    %i(admin answered_count asked_count bio blogger comment_smiled_count commented_count contributor created_at
+       display_name follower_count friend_count id location moderator motivation_header permanently_banned screen_name
+       smiled_count supporter translator website).each do |f|
+      uobj[f] = user.send f
+    end
+    uobj
+  end
+end