DEV: Xenforo importer improvements (#18457)

* Fix: make expressions non-greedy
* Feature: import Xenforo avatars
* Feature: import Xenforo likes
* Feature: import Xenforo private messages
* Feature: Xenforo create permalinks
* Feature: Xenforo migrate view counts
* Fix: Xenforo list regexes
* Fix: Xenforo import all attachments

script/import_scripts/xenforo.rb

@@ -2,6 +2,18 @@
 
 require "mysql2"
 
+begin
+  require 'php_serialize' # https://github.com/jqr/php-serialize
+rescue LoadError
+  puts
+  puts 'php_serialize not found.'
+  puts 'Add to Gemfile, like this: '
+  puts
+  puts "echo gem \\'php-serialize\\' >> Gemfile"
+  puts "bundle install"
+  exit
+end
+
 require File.expand_path(File.dirname(__FILE__) + "/base.rb")
 
 # Call it like this:
@@ -30,6 +42,20 @@ class ImportScripts::XenForo < ImportScripts::Base
     import_users
     import_categories
     import_posts
+    import_private_messages
+    import_likes
+  end
+
+  def import_avatar(id, imported_user)
+    filename = File.join(AVATAR_DIR, 'l', (id / 1000).to_s, "#{id}.jpg")
+    unless File.exist?(filename)
+      return nil
+    end
+    upload = create_upload(imported_user.id, filename, "avatar_#{id}")
+    return if !upload.persisted?
+    imported_user.create_user_avatar
+    imported_user.user_avatar.update(custom_upload_id: upload.id)
+    imported_user.update(uploaded_avatar_id: upload.id)
   end
 
   def import_users
@@ -59,7 +85,10 @@ class ImportScripts::XenForo < ImportScripts::Base
           created_at: Time.zone.at(user['created_at']),
           last_seen_at: Time.zone.at(user['last_visit_time']),
           moderator: user['is_moderator'] == 1 || user['is_staff'] == 1,
-          admin: user['is_admin'] == 1 }
+          admin: user['is_admin'] == 1,
+          post_create_action: proc do |u|
+            import_avatar(user['id'], u)
+          end
       end
     end
   end
@@ -72,6 +101,7 @@ class ImportScripts::XenForo < ImportScripts::Base
                title,
                description,
                parent_node_id,
+               node_name,
                display_order
           FROM #{TABLE_PREFIX}node
       ORDER BY parent_node_id, display_order
@@ -84,7 +114,11 @@ class ImportScripts::XenForo < ImportScripts::Base
         id: c['id'],
         name: c['title'],
         description: c['description'],
-        position: c['display_order']
+        position: c['display_order'],
+        post_create_action: proc do |category|
+          url = "board/#{c['node_name']}"
+          Permalink.find_or_create_by(url: url, category_id: category.id) 
+        end
       }
     end
 
@@ -98,7 +132,11 @@ class ImportScripts::XenForo < ImportScripts::Base
         name: c['title'],
         description: c['description'],
         position: c['display_order'],
-        parent_category_id: category_id_from_imported_category_id(c['parent_node_id'])
+        parent_category_id: category_id_from_imported_category_id(c['parent_node_id']),
+        post_create_action: proc do |category|
+          url = "board/#{c['node_name']}"
+          Permalink.find_or_create_by(url: url, category_id: category.id) 
+        end
       }
     end
 
@@ -149,6 +187,29 @@ class ImportScripts::XenForo < ImportScripts::Base
     @prefix_as_category = true
   end
 
+  def import_likes
+    puts '', 'importing likes'
+    total_count = mysql_query("SELECT COUNT(*) AS count FROM #{TABLE_PREFIX}liked_content WHERE content_type = 'post'").first["count"]
+    batches(BATCH_SIZE) do |offset|
+      results = mysql_query(
+        "SELECT like_id, content_id, like_user_id, like_date 
+         FROM #{TABLE_PREFIX}liked_content
+         WHERE content_type = 'post'
+         ORDER BY like_id
+         LIMIT #{BATCH_SIZE}
+         OFFSET #{offset};"
+      )
+      break if results.size < 1
+      create_likes(results, total: total_count, offset: offset) do |row|
+        {
+           post_id: row['content_id'],
+           user_id: row['like_user_id'],
+           created_at: Time.zone.at(row['like_date'])
+        }
+      end
+    end
+  end
+
   def import_posts
     puts "", "creating topics and posts"
 
@@ -160,6 +221,7 @@ class ImportScripts::XenForo < ImportScripts::Base
                #{@prefix_as_category ? 't.prefix_id' : 't.node_id'} category_id,
                t.title title,
                t.first_post_id first_post_id,
+               t.view_count,
                p.user_id user_id,
                p.message raw,
                p.post_date created_at
@@ -194,6 +256,10 @@ class ImportScripts::XenForo < ImportScripts::Base
               @category_mappings[m['category_id']].try(:[], :category_id)
           end
           mapped[:title] = CGI.unescapeHTML(m['title'])
+          mapped[:views] = m['view_count']
+          mapped[:post_create_action] = proc do |pp|
+            Permalink.find_or_create_by(url: "threads/#{m['topic_id']}", topic_id: pp.topic_id)
+          end
         else
           parent = topic_lookup_from_imported_post_id(m['first_post_id'])
           if parent
@@ -226,6 +292,60 @@ class ImportScripts::XenForo < ImportScripts::Base
 
   end
 
+  def import_private_messages
+    puts "", "importing private messages..."
+    post_count = mysql_query("SELECT COUNT(*) count FROM xf_conversation_message").first["count"]
+    batches(BATCH_SIZE) do |offset|
+      posts = mysql_query <<-SQL
+        SELECT c.conversation_id, c.recipients, c.title, m.message, m.user_id, m.message_date, m.message_id, IF(c.first_message_id != m.message_id, c.first_message_id, 0) as topic_id 
+        FROM xf_conversation_master c 
+        LEFT JOIN xf_conversation_message m ON m.conversation_id = c.conversation_id 
+        ORDER BY c.conversation_id, m.message_id 
+        LIMIT #{BATCH_SIZE}
+        OFFSET #{offset}
+      SQL
+      break if posts.size < 1
+      next if all_records_exist? :posts, posts.map { |post| "pm_#{post["message_id"]}" }
+      create_posts(posts, total: post_count, offset: offset) do |post|
+        user_id = user_id_from_imported_user_id(post["user_id"]) || Discourse::SYSTEM_USER_ID
+        title = post["title"]
+        message_id = "pm_#{post["message_id"]}"
+        raw = process_xenforo_post(post["message"], 0)
+        if raw.present?
+          msg = {
+            id: message_id,
+            user_id: user_id,
+            raw: raw,
+            created_at:  Time.zone.at(post["message_date"].to_i),
+            import_mode: true
+          }
+          unless post["topic_id"] > 0
+            msg[:title] = post["title"]
+            msg[:archetype] = Archetype.private_message
+            to_user_array = PHP.unserialize(post['recipients'])
+            if to_user_array.size > 0
+              discourse_user_ids = to_user_array.keys.map { |id| user_id_from_imported_user_id(id) }
+              usernames = User.where(id: [discourse_user_ids]).pluck(:username)
+              msg[:target_usernames] = usernames.join(',')
+            end
+          else
+            topic_id = post["topic_id"]
+            if t = topic_lookup_from_imported_post_id("pm_#{topic_id}")
+              msg[:topic_id] = t[:topic_id]
+            else
+              puts "Topic ID #{topic_id} not found, skipping post #{post['message_id']} from #{post['user_id']}"
+              next
+            end
+          end
+          msg
+        else
+          puts "Empty message, skipping post #{post['message_id']}"
+          next
+        end
+      end
+    end
+  end
+
   def process_xenforo_post(raw, import_id)
     s = raw.dup
 
@@ -276,54 +396,70 @@ class ImportScripts::XenForo < ImportScripts::Base
     end
 
     # [URL=...]...[/URL]
-    s.gsub!(/\[url="?(.+?)"?\](.+)\[\/url\]/i) { "[#{$2}](#{$1})" }
+    s.gsub!(/\[url="?(.+?)"?\](.+?)\[\/url\]/i) { "[#{$2}](#{$1})" }
+
+    # [URL]...[/URL]
+    s.gsub!(/\[url\](.+?)\[\/url\]/i) { " #{$1} " }
 
     # [IMG]...[/IMG]
     s.gsub!(/\[\/?img\]/i, "")
 
     # convert list tags to ul and list=1 tags to ol
     # (basically, we're only missing list=a here...)
-    s.gsub!(/\[list\](.*?)\[\/list:u\]/m, '[ul]\1[/ul]')
+    s.gsub!(/\[list\](.*?)\[\/list\]/im, '[ul]\1[/ul]')
-    s.gsub!(/\[list=1\](.*?)\[\/list:o\]/m, '[ol]\1[/ol]')
+    s.gsub!(/\[list=1\](.*?)\[\/list\]/im, '[ol]\1[/ol]')
+    s.gsub!(/\[list\](.*?)\[\/list:u\]/im, '[ul]\1[/ul]')
+    s.gsub!(/\[list=1\](.*?)\[\/list:o\]/im, '[ol]\1[/ol]')
+
     # convert *-tags to li-tags so bbcode-to-md can do its magic on phpBB's lists:
+    s.gsub!(/\[\*\]\n/, '')
     s.gsub!(/\[\*\](.*?)\[\/\*:m\]/, '[li]\1[/li]')
+    s.gsub!(/\[\*\](.*?)\n/, '[li]\1[/li]')
+    s.gsub!(/\[\*=1\]/, '')
 
     # [YOUTUBE]<id>[/YOUTUBE]
     s.gsub!(/\[youtube\](.+?)\[\/youtube\]/i) { "\nhttps://www.youtube.com/watch?v=#{$1}\n" }
 
     # [youtube=425,350]id[/youtube]
-    s.gsub!(/\[youtube="?(.+?)"?\](.+)\[\/youtube\]/i) { "\nhttps://www.youtube.com/watch?v=#{$2}\n" }
+    s.gsub!(/\[youtube="?(.+?)"?\](.+?)\[\/youtube\]/i) { "\nhttps://www.youtube.com/watch?v=#{$2}\n" }
 
     # [MEDIA=youtube]id[/MEDIA]
     s.gsub!(/\[MEDIA=youtube\](.+?)\[\/MEDIA\]/i) { "\nhttps://www.youtube.com/watch?v=#{$1}\n" }
 
     # [ame="youtube_link"]title[/ame]
-    s.gsub!(/\[ame="?(.+?)"?\](.+)\[\/ame\]/i) { "\n#{$1}\n" }
+    s.gsub!(/\[ame="?(.+?)"?\](.+?)\[\/ame\]/i) { "\n#{$1}\n" }
 
     # [VIDEO=youtube;<id>]...[/VIDEO]
     s.gsub!(/\[video=youtube;([^\]]+)\].*?\[\/video\]/i) { "\nhttps://www.youtube.com/watch?v=#{$1}\n" }
 
     # [USER=706]@username[/USER]
-    s.gsub!(/\[user="?(.+?)"?\](.+)\[\/user\]/i) { $2 }
+    s.gsub!(/\[user="?(.+?)"?\](.+?)\[\/user\]/i) { $2 }
 
     # Remove the color tag
     s.gsub!(/\[color=[#a-z0-9]+\]/i, "")
     s.gsub!(/\[\/color\]/i, "")
 
     if Dir.exist? ATTACHMENT_DIR
-      s = process_xf_attachments(:gallery, s)
+      s = process_xf_attachments(:gallery, s, import_id)
-      s = process_xf_attachments(:attachment, s)
+      s = process_xf_attachments(:attachment, s, import_id)
     end
 
     s
   end
 
-  def process_xf_attachments(xf_type, s)
+  def process_xf_attachments(xf_type, s, import_id)
     ids = Set.new
     ids.merge(s.scan(get_xf_regexp(xf_type)).map { |x| x[0].to_i })
+    
+    # not all attachments have an [ATTACH=] tag so we need to get the other ID's from the xf_attachment table
+    if xf_type == :attachment && import_id > 0
+      sql = "SELECT attachment_id FROM #{TABLE_PREFIX}attachment WHERE content_id=#{import_id} and content_type='post';"
+      ids.merge(mysql_query(sql).to_a.map { |v| v["attachment_id"].to_i})
+    end
+    
     ids.each do |id|
       next unless id
-      sql = get_xf_sql(xf_type, id).squish!
+      sql = get_xf_sql(xf_type, id).dup.squish!
       results = mysql_query(sql)
       if results.size < 1
         # Strip attachment
@@ -334,11 +470,13 @@ class ImportScripts::XenForo < ImportScripts::Base
       original_filename = results.first['filename']
       result = results.first
       upload = import_xf_attachment(result['data_id'], result['file_hash'], result['user_id'], original_filename)
-      next unless upload
+      if upload && upload.present? && upload.persisted?
-      if upload.present? && upload.persisted?
+        html = @uploader.html_for_upload(upload, original_filename)
-        s.gsub!(get_xf_regexp(xf_type, id), @uploader.html_for_upload(upload, original_filename))
+        unless s.gsub!(get_xf_regexp(xf_type, id), html)
+          s = s + "\n\n#{html}\n\n"
+        end
       else
-        STDERR.puts "Could not find upload: #{upload.id}. Skipping attachment id #{id}"
+        STDERR.puts "Could not process upload: #{original_filename}. Skipping attachment id #{id}"
       end
     end
     s
@@ -372,18 +510,19 @@ class ImportScripts::XenForo < ImportScripts::Base
     case type
     when :gallery
       <<-SQL
-		SELECT m.media_id, m.media_title, a.attachment_id, a.data_id, d.filename, d.file_hash,d.user_id
+        SELECT m.media_id, m.media_title, a.attachment_id, a.data_id, d.filename, d.file_hash, d.user_id
-		FROM xengallery_media as m
+        FROM xengallery_media AS m
-		INNER JOIN #{TABLE_PREFIX}attachment a on m.attachment_id = a.attachment_id
+        INNER JOIN #{TABLE_PREFIX}attachment a ON (m.attachment_id = a.attachment_id AND a.content_type = 'xengallery_media')
-		INNER JOIN #{TABLE_PREFIX}attachment_data d on a.data_id = d.data_id
+        INNER JOIN #{TABLE_PREFIX}attachment_data d ON a.data_id = d.data_id
-		WHERE media_id = #{id}
+        WHERE media_id = #{id}
       SQL
     when :attachment
       <<-SQL
-		SELECT a.attachment_id, a.data_id, d.filename, d.file_hash, d.user_id
+        SELECT a.attachment_id, a.data_id, d.filename, d.file_hash, d.user_id
-		FROM #{TABLE_PREFIX}attachment AS a
+        FROM #{TABLE_PREFIX}attachment AS a 
-		INNER JOIN #{TABLE_PREFIX}attachment_data d ON a.data_id = d.data_id
+        INNER JOIN #{TABLE_PREFIX}attachment_data d ON a.data_id = d.data_id
-		WHERE attachment_id = #{id}
+        WHERE attachment_id = #{id}
+        AND content_type = 'post'
       SQL
     end
   end