require "action_view"
require "colorize"
require "fileutils"
require "net/http"
require "open-uri"
require "socket"

# Some important folders to help understand how jets builds a project:
#
# /tmp/jets: build root where different jets projects get built.
# /tmp/jets/project: each jets project gets built in a different subdirectory.
#
# The rest of the folders are subfolders under /tmp/jets/project.
#
class Jets::Builders
  class CodeBuilder
    include Jets::AwsServices
    include Util
    extend Memoist

    attr_reader :full_project_path
    def initialize
      # Expanding to the full path and capture now.
      # Dir.chdir gets called later and we'll lose this info.
      @full_project_path = File.expand_path(Jets.root) + "/"
      @version_purger = Purger.new
    end

    def build
      check_ruby_version
      @version_purger.purge
      cache_check_message

      clean_start
      compile_assets # easier to do before we copy the project because node and yarn has been likely setup in the that dir
      compile_rails_assets
      copy_project
      Dir.chdir("#{stage_area}/code") do
        # These commands run from project root
        code_setup
        package_ruby
        code_finish
      end
    end

    # Resolves the chicken-and-egg problem with md5 checksums. The handlers need
    # to reference files with the md5 checksum.  The files are the:
    #
    #   jets/code/rack-checksum.zip
    #   jets/code/opt-checksum.zip
    #
    # We compute the checksums before we generate the node shim handlers.
    def calculate_md5s
      Md5.compute! # populates Md5.checksums hash
    end

    def generate_node_shims
      headline "Generating shims in the handlers folder."
      # Crucial that the Dir.pwd is in the tmp_code because for
      # Jets::Builders::app_files because Jets.boot set ups
      # autoload_paths and this is how project classes are loaded.
      Jets::Builders::HandlerGenerator.build!
    end

    def create_zip_files
      folders = Md5.stage_folders
      # Md5.stage_folders ["stage/bundled", "stage/code"]
      folders.each do |folder|
        zip = Md5Zip.new(folder)
        if exist_on_s3?(zip.md5_name)
          puts "Already exists: s3://#{s3_bucket}/jets/code/#{zip.md5_name}"
        else
          zip = Md5Zip.new(folder)
          zip.create
        end
      end
    end

    def exist_on_s3?(filename)
      s3_key = "jets/code/#{filename}"
      begin
        s3.head_object(bucket: s3_bucket, key: s3_key)
        true
      rescue Aws::S3::Errors::NotFound
        false
      end
    end

    def code_setup
      reconfigure_development_webpacker
    end

    def code_finish
      # Reconfigure code
      store_s3_base_url
      disable_webpacker_middleware
      copy_internal_jets_code

      # Code prep and zipping
      check_code_size!
      calculate_md5s # must be called before generate_node_shims and create_zip_files
      generate_node_shims
      create_zip_files
    end

    def check_code_size!
      CodeSize.check!
    end

    # We copy the files into the project because we cannot require simple functions
    # directly since they are wrapped by an anonymous class.
    # TODO: Do this with the other files we required the same way.
    def copy_internal_jets_code
      files = []
      files.each do |relative_path|
        src = File.expand_path("../internal/#{relative_path}", File.dirname(__FILE__))
        dest = "#{"#{stage_area}/code"}/#{relative_path}"
        FileUtils.mkdir_p(File.dirname(dest))
        FileUtils.cp(src, dest)
      end
    end

    # Thanks https://stackoverflow.com/questions/9354595/recursively-getting-the-size-of-a-directory
    # Seems to overestimate a little bit but close enough.
    def dir_size(folder)
      Dir.glob(File.join(folder, '**', '*'))
        .select { |f| File.file?(f) }
        .map{ |f| File.size(f) }
        .inject(:+)
    end

    # Store s3 base url is needed for asset serving from s3 later. Need to package this
    # as part of the code so we have a reference to it.
    # At this point the minimal stack exists, so we can grab it with the AWS API.
    # We do not want to grab this as part of the live request because it is slow.
    def store_s3_base_url
      write_s3_base_url("#{stage_area}/code/config/s3_base_url.txt")
      write_s3_base_url("#{stage_area}/rack/config/s3_base_url.txt") if Jets.rack?
    end

    def write_s3_base_url(full_path)
      FileUtils.mkdir_p(File.dirname(full_path))
      IO.write(full_path, s3_base_url)
    end

    def s3_base_url
      # Allow user to set assets.base_url
      #
      #   Jets.application.configure do
      #     config.assets.base_url = "https://cloudfront.com/my/base/path"
      #   end
      #
      return Jets.config.assets.base_url if Jets.config.assets.base_url

      region = Jets.aws.region

      asset_base_url = "https://s3-#{region}.amazonaws.com"
      "#{asset_base_url}/#{s3_bucket}/jets" # s3_base_url
    end

    def s3_bucket
      Jets.aws.s3_bucket
    end

    def disable_webpacker_middleware
      full_path = "#{"#{stage_area}/code"}/config/disable-webpacker-middleware.txt"
      FileUtils.mkdir_p(File.dirname(full_path))
      FileUtils.touch(full_path)
    end

    # This happens in the current app directory not the tmp code for simplicity.
    # This is because the node and yarn has likely been set up correctly there.
    def compile_assets
      if ENV['JETS_SKIP_ASSETS']
        puts "Skip compiling assets".colorize(:yellow) # useful for debugging
        return
      end

      headline "Compling assets in current project directory"
      return unless webpacker_included?

      sh("yarn install")
      webpack_command = File.exist?("#{Jets.root}bin/webpack") ?
          "bin/webpack" :
          `which webpack`.strip
      sh "JETS_ENV=#{Jets.env} #{webpack_command}"
    end

    def webpacker_included?
      # Old code, leaving around for now:
      # Thanks: https://stackoverflow.com/questions/4195735/get-list-of-gems-being-used-by-a-bundler-project
      # webpacker_loaded = Gem.loaded_specs.keys.include?("webpacker")
      # return unless webpacker_loaded

      # Checking this way because when using jets standalone for Afterburner mode we don't want to run into
      # bundler gem collisions.  TODO: figure out the a better way to handle the collisions.
      lines = IO.readlines("#{Jets.root}Gemfile")
      lines.detect { |l| l =~ /webpacker/ }
    end

    # This happens in the current app directory not the tmp code for simplicity
    # This is because the node likely been set up correctly there.
    def compile_rails_assets
      return unless Jets.rack? && rails? && !rails_api?

      if ENV['JETS_SKIP_ASSETS']
        puts "Skip compiling rack assets".colorize(:yellow) # useful for debugging
        return
      end

      # Need to capture JETS_ROOT since can be changed by Turbo mode
      jets_root = Jets.root
      Bundler.with_clean_env do
        # Switch gemfile for Afterburner mode
        gemfile = ENV['BUNDLE_GEMFILE']
        ENV['BUNDLE_GEMFILE'] = "#{jets_root}/rack/Gemfile"
        sh "cd #{jets_root} && bundle install"
        ENV['BUNDLE_GEMFILE'] = gemfile

        rails_assets(:clobber, jets_root: jets_root)
        rails_assets(:precompile, jets_root: jets_root)
      end
    end

    def rails_assets(cmd, jets_root:)
      # rake is available in both rails 4 and 5. rails command only in 5
      command = "bundle exec rake assets:#{cmd} --trace"
      command = "RAILS_ENV=#{Jets.env} #{command}" unless Jets.env.development?
      sh("cd #{jets_root}rack && #{command}")
    end

    # Rudimentary rails detection
    # Duplicated in builders/reconfigure_rails.rb
    def rails?
      config_ru = "#{Jets.root}rack/config.ru"
      return false unless File.exist?(config_ru)
      !IO.readlines(config_ru).grep(/Rails.application/).empty?
    end

    # Rudimentary rails api detection
    # Duplicated in builders/reconfigure_rails.rb
    # Another way of checking is loading a rails console and checking Rails.application.config.api_only
    # Using this way for simplicity.
    def rails_api?
      config_app = "#{Jets.root}rack/config/application.rb"
      return false unless File.exist?(config_app)
      !IO.readlines(config_app).grep(/config.api_only.*=.*true/).empty?
    end

    # Cleans out non-cached files like code-*.zip in Jets.build_root
    # for a clean start. Also ensure that the /tmp/jets/project build root exists.
    #
    # Most files are kept around after the build process for inspection and
    # debugging. So we have to clean out the files. But we only want to clean out
    # some of the files.
    def clean_start
      Dir.glob("#{Jets.build_root}/code/code-*.zip").each { |f| FileUtils.rm_f(f) }
      FileUtils.mkdir_p(Jets.build_root) # /tmp/jets/demo
    end

    # Copy project into temporary directory. Do this so we can keep the project
    # directory untouched and we can also remove a bunch of unnecessary files like
    # logs before zipping it up.
    def copy_project
      headline "Copying current project directory to temporary build area: #{"#{stage_area}/code"}"
      FileUtils.rm_rf("#{build_area}/stage") # clear out from previous build's stage area
      FileUtils.mkdir_p("#{build_area}/stage")
      FileUtils.rm_rf("#{stage_area}/code") # remove current code folder
      move_node_modules(Jets.root, Jets.build_root)
      begin
        # puts "cp -r #{@full_project_path} #{"#{stage_area}/code"}".colorize(:yellow) # uncomment to debug
        Jets::Util.cp_r(@full_project_path, "#{stage_area}/code")
      ensure
        move_node_modules(Jets.build_root, Jets.root) # move node_modules directory back
      end
    end

    # Move the node modules to the tmp build folder to speed up project copying.
    # A little bit risky because a ctrl-c in the middle of the project copying
    # results in a missing node_modules but user can easily rebuild that.
    #
    # Tesing shows 6.623413 vs 0.027754 speed improvement.
    def move_node_modules(source_folder, dest_folder)
      source = "#{source_folder}/node_modules"
      dest = "#{dest_folder}/node_modules"
      if File.exist?(source)
        FileUtils.mv(source, dest)
      end
    end

    # Bit hacky but this saves the user from accidentally forgetting to change this
    # when they deploy a jets project in development mode
    def reconfigure_development_webpacker
      return unless Jets.env.development?
      headline "Reconfiguring webpacker development settings for AWS Lambda."

      webpacker_yml = "#{"#{stage_area}/code"}/config/webpacker.yml"
      return unless File.exist?(webpacker_yml)

      config = YAML.load_file(webpacker_yml)
      config["development"]["compile"] = false # force this to be false for deployment
      new_yaml = YAML.dump(config)
      IO.write(webpacker_yml, new_yaml)
    end

    def ruby_packager
      RubyPackager.new(tmp_code)
    end
    memoize :ruby_packager

    def rack_packager
      RackPackager.new("#{tmp_code}/rack")
    end
    memoize :rack_packager

    def package_ruby
      return if Jets.poly_only?

      check_agree
      ruby_packager.install
      reconfigure_rails # call here after "#{stage_area}/code" is available
      rack_packager.install
      ruby_packager.finish # by this time we have a /tmp/jets/demo/stage/code/vendor/gems
      rack_packager.finish

      build_lambda_layer
    end

    def check_agree
      agree = Jets::Gems::Agree.new
      agree.prompt
    end

    def build_lambda_layer
      return if Jets.poly_only?
      lambda_layer = LambdaLayer.new
      lambda_layer.build
    end

    # TODO: Move logic into plugin instead
    def reconfigure_rails
      ReconfigureRails.new("#{"#{stage_area}/code"}/rack").run
    end

    def cache_check_message
      if File.exist?("#{Jets.build_root}/cache")
        puts "The #{Jets.build_root}/cache folder exists. Incrementally re-building the jets using the cache.  To clear the cache: rm -rf #{Jets.build_root}/cache"
      end
    end

    def check_ruby_version
      unless ruby_version_supported?
        puts "You are using Ruby version #{RUBY_VERSION} which is not supported by Jets."
        ruby_variant = Jets::RUBY_VERSION.split('.')[0..1].join('.') + '.x'
        abort("Jets uses Ruby #{Jets::RUBY_VERSION}.  You should use a variant of Ruby #{ruby_variant}".colorize(:red))
      end
    end

    def ruby_version_supported?
      pattern = /(\d+)\.(\d+)\.(\d+)/
      md = RUBY_VERSION.match(pattern)
      ruby = {major: md[1], minor: md[2]}
      md = Jets::RUBY_VERSION.match(pattern)
      jets = {major: md[1], minor: md[2]}

      ruby[:major] == jets[:major] && ruby[:minor] == jets[:minor]
    end

    # Group all the path settings together here
    def self.tmp_code
      Jets::Commands::Build.tmp_code
    end

    def tmp_code
      self.class.tmp_code
    end
  end
end