#!/usr/bin/env ruby

require "thor"

class DiscourseCLI < Thor
  class_option :verbose, default: false, aliases: :v

  desc "export", "Export a Discourse backup"
  def export(filename=nil)
    load_rails

    filename = File.expand_path(filename) if filename
    puts 'Starting export...'
    output_filename = Jobs::Exporter.new.execute( format: :json, filename: filename)
    puts 'Export done.'
    puts "Output file is in: #{output_filename}", ''
  end

  desc "import", "Import a Discourse backup"
  def import(filename)
    load_rails

    begin
      filename = File.expand_path(filename)
      puts "Starting import: #{filename}"
      Jobs::Importer.new.execute( format: :json, filename: filename)
      puts 'Import done.'
    rescue Import::FilenameMissingError
      puts '', 'The filename argument was missing.', ''
      usage
    rescue Import::ImportDisabledError
      puts '', 'Imports are not allowed.', 'An admin needs to set allow_import to true in the site settings before imports can be run.', ''
      puts 'Import cancelled.', ''
    end
  end

  desc "migrate", "Make sure all the posts are pointing to the new domain"
  option :from, type: :array, required: true, banner: "http://previous.domain.com"
  option :database, default: "default", aliases: :db
  def migrate
    verbose = options[:verbose]
    database = options[:database]
    from = options[:from].map { |f| schemaless f }

    begin
      puts "loading rails..." if verbose
      load_rails

      puts "connecting to #{database}..." if verbose
      RailsMultisite::ConnectionManagement.establish_connection(db: database)

      base_url = schemaless Discourse.base_url_no_prefix

      puts "updating #{Post.count} posts to #{base_url}" if verbose
      Post.find_each do |post|
        raw, cooked = post.raw.dup, post.cooked.dup
        from.each do |f|
          raw.gsub!(f, base_url)
          cooked.gsub!(f, base_url)
        end
        if raw != post.raw || cooked != post.cooked
          Post.where(id: post.id).update_all(raw: raw, cooked: cooked)
          putc "#" if verbose
        else
          putc "."
        end
      end

    rescue => e
      puts "Cannot connect to database: #{database}"
      puts e
      puts e.backtrace.join("\n")
    end

    puts "", "done!" if verbose
  end

  private

  def load_rails
    require File.expand_path(File.dirname(__FILE__) + "/../config/environment")
  end

  def schemaless(url)
    url.gsub(/^https?:/, "")
  end

end

DiscourseCLI.start(ARGV)