require 'cocoapods-meitu-bin/helpers/buildAll/builder'
require 'cocoapods-meitu-bin/helpers/buildAll/podspec_util'
require 'cocoapods-meitu-bin/helpers/buildAll/zip_file_helper'
require 'cocoapods-meitu-bin/helpers/buildAll/bin_helper'
require 'cocoapods-meitu-bin/config/config'
require 'yaml'
require 'digest'
# require 'concurrent'

module Pod
  class Command
    class Bin < Command
      class BuildAll < Bin
        include CBin::BuildAll

        CDN = 'https://cdn.cocoapods.org/'.freeze
        MASTER_HTTP = 'https://github.com/CocoaPods/Specs.git'.freeze
        MASTER_SSH = 'git@github.com:CocoaPods/Specs.git'.freeze

        self.summary = '根据壳工程打包所有依赖组件为静态库（static framework）'
        self.description = <<-DESC
          #{summary}，参数 PODS 可选，如果添加 PODS 则只制作 PODS 包含的库，多个库中间用逗号分隔
        DESC

        self.arguments = [
          CLAide::Argument.new('PODS', false)
        ]

        def self.options
          [
            %w[--clean 全部二进制包制作完成后删除编译临时目录],
            %w[--shell-project 是否依赖壳工程编译产物一次编译生成所有组件的编译产物],
            %w[--clean-single 每制作完一个二进制包就删除该编译临时目录],
            %w[--repo-update 更新Podfile中指定的repo仓库],
            %w[--full-build 是否全量打包],
            %w[--skip-simulator 跳过模拟器编译],
            %w[--configuration=configName 在构建每个目标时使用configName指定构建配置，如：'Debug'、'Release'等]
          ].concat(super).uniq
        end

        def initialize(argv)
          @pods = argv.shift_argument
          @clean = argv.flag?('clean', false)
          @clean_single = argv.flag?('clean-single', false)
          @shell_project = argv.flag?('shell-project', false)
          @repo_update = argv.flag?('repo-update', false)
          @full_build = argv.flag?('full-build', false)
          @skip_simulator = argv.flag?('skip-simulator', false)
          @configuration = argv.option('configuration', 'Debug')
          @base_dir = "#{Pathname.pwd}/build_pods"
          @version_helper = BinHelper.new
          super
        end

        def run
          CBin.config.config_file_exist?
          # 打印提示
          print_tip
          # 开始时间
          @start_time = Time.now.to_i
          # 读取配置文件
          read_config
          # 如果有传入要制作的pod库名
          if @pods
            @write_list = @pods.split(',').map(&:strip)
          end
          # 更新repo仓库
          repo_update
          # 执行pre_build命令
          pre_build
          # 分析依赖
          @analyze_result = analyse
          # 删除编译产物
          clean_build_pods
          # 编译所有pod_targets
          #根据@shell_project 判断是执行 build_shell_project_pod_targets 还是 build_pod_targets
          if @shell_project
            PodUpdateConfig.set_shell_project
          end
          results = @shell_project ? build_shell_project_pod_targets : build_pod_targets
          # 执行post_build命令
          post_build(results)
          # 删除编译产物
          clean_build_pods if @clean
          # 计算耗时
          show_cost_time
        end

        private

        def print_tip
          UI.info '——————————————————————————————————'.blue
          UI.info "|#{' '.center(32)}|".blue
          UI.info "|#{"Configuration:`#{@configuration}`".center(32)}|".blue
          UI.info "|#{' '.center(32)}|".blue
          UI.info '——————————————————————————————————'.blue
        end

        # 打印耗时
        def show_cost_time
          return if @start_time.nil?
          UI.info "总耗时：#{Time.now.to_i - @start_time}s".green
        end

        # 读取配置文件
        def read_config
          UI.title 'Read config from file `BinConfig.yaml`'.green do
            config_file = File.join(Pod::Config.instance.project_root, 'BinConfig.yaml')
            return unless File.exist?(config_file)
            config = YAML.safe_load(File.open(config_file))
            return if config.nil?
            build_config = config['build_config']
            return if build_config.nil?
            @pre_build = build_config['pre_build']
            @post_build = build_config['post_build']
            @black_list = build_config['black_list']
            @write_list = build_config['write_list']
          end
        end

        # 更新repo仓库
        def repo_update
          if @repo_update
            UI.title 'Repo update'.green do
              return if podfile.nil?
              sources_manager = Pod::Config.instance.sources_manager
              podfile.sources.uniq.map do |src|
                UI.message "Update repo: #{src}"
                source = sources_manager.source_with_name_or_url(src)
                source.update(false)
              end
            end
          end
        end

        # 执行pre build
        def pre_build
          if @pre_build
            UI.title 'Execute the command of pre build'.green do
              system(@pre_build)
            end
          end
        end

        # 执行post build
        def post_build(_results)
          if @post_build
            UI.title 'Execute the command of post build'.green do
              system(@post_build)
            end
          end
        end

        # 获取 podfile
        def podfile
          @podfile ||=  Pod::Config.instance.podfile
        end

        # 获取 podfile.lock
        def lockfile
          @lockfile ||= Pod::Config.instance.lockfile
        end

        # 获取 sandbox
        def sandbox
          @sandbox ||=  Pod::Config.instance.sandbox
        end

        # 根据podfile和podfile.lock分析依赖
        def analyse
          UI.title 'Analyze dependencies'.green do
            analyzer = Pod::Installer::Analyzer.new(
              sandbox,
              podfile,
              lockfile
            )
            analyzer.analyze(true)
          end
        end

        # 删除单个Pod的编译中间产物
        def clean_single_build_pod(pod_target)
          UI.title "Clean build pod: `#{pod_target}`".green do
            build_pod_path = File.join(Dir.pwd, 'build_pods', "#{pod_target}")
            FileUtils.rm_rf(build_pod_path) if File.exist?(build_pod_path)
          end
        end

        # 删除编译产物
        def clean_build_pods
          UI.title 'Clean build pods'.green do
            build_path = Dir.pwd + '/build'
            FileUtils.rm_rf(build_path) if File.exist?(build_path)
            build_pods_path = Dir.pwd + '/build_pods'
            FileUtils.rm_rf(build_pods_path) if File.exist?(build_pods_path)
          end
        end

        # 构建所有pod_targets
        def build_pod_targets
          UI.title "Build all pod targets(#{@full_build ? '全量打包' : '非全量打包'})".green do
            pod_targets = @analyze_result.pod_targets.uniq
            success_pods = []
            fail_pods = []
            local_pods = []
            external_pods = []
            binary_pods = []
            created_pods = []
            pod_targets.map do |pod_target|
              begin
                version = @version_helper.version(pod_target.pod_name, pod_target.root_spec.version.to_s, @analyze_result.specifications, @configuration, podfile.include_dependencies?)
                # 黑名单（不分全量和非全量）
                next if skip_build?(pod_target)
                # 白名单（有白名单，只看白名单，不分全量和非全量）
                next if !@write_list.nil? && !@write_list.empty? && !@write_list.include?(pod_target.pod_name)
                # 本地库
                if @sandbox.local?(pod_target.pod_name)
                  local_pods << pod_target.pod_name
                  show_skip_tip("#{pod_target.pod_name} 是本地库")
                  next
                end
                # 外部源（如 git）
                if @sandbox.checkout_sources[pod_target.pod_name]
                  external_pods << pod_target.pod_name
                  show_skip_tip("#{pod_target.pod_name} 以external方式引入")
                  next
                end
                # 无源码
                if !@sandbox.local?(pod_target.pod_name) && !pod_target.should_build?
                  binary_pods << pod_target.pod_name
                  show_skip_tip("#{pod_target.pod_name} 无需编译")
                  next
                end
                # 非全量编译、不在白名单中且已经有相应的二进制版本
                if has_created_binary?(pod_target.pod_name, version)
                  created_pods << pod_target.pod_name
                  show_skip_tip("#{pod_target.pod_name}(#{version}) 已经有二进制版本了")
                  next
                end
                # 构建产物
                builder = Builder.new(pod_target, @sandbox.checkout_sources, @skip_simulator, @configuration)
                result = builder.build
                fail_pods << pod_target.pod_name unless result
                next unless result
                builder.create_binary
                # 压缩并上传zip
                zip_helper = ZipFileHelper.new(pod_target, version, builder.product_dir, builder.build_as_framework?, @configuration)
                result = zip_helper.zip_lib
                fail_pods << pod_target.pod_name unless result
                next unless result
                result = zip_helper.upload_zip_lib
                fail_pods << pod_target.pod_name unless result
                next unless result
                # 生成二进制podspec并上传
                podspec_creator = PodspecUtil.new(pod_target, version, builder.build_as_framework?, @configuration)
                bin_spec = podspec_creator.create_binary_podspec
                bin_spec_file = podspec_creator.write_binary_podspec(bin_spec)
                result = podspec_creator.push_binary_podspec(bin_spec_file)
                fail_pods << pod_target.pod_name unless result
                success_pods << pod_target.pod_name if result
              rescue Pod::StandardError => e
                UI.info "`#{pod_target}`编译失败，原因：#{e}".red
                fail_pods << pod_target.pod_name
                next
              ensure
                clean_single_build_pod(pod_target) if @clean_single
              end
            end
            results = {
              'Total' => pod_targets,
              'Success' => success_pods,
              'Fail' => fail_pods,
              'Local' => local_pods,
              'External' => external_pods,
              'No Source File' => binary_pods,
              'Created Binary' => created_pods,
              'Black List' => @black_list || [],
              'Write List' => @write_list || []
            }
            show_results(results)
            results
          end
        end
        # 基于壳工程编译产物进行个组件的zip压缩 上传和对应二进制podspec生成上传
        def build_shell_project_pod_targets
          UI.title "Build all pod targets(#{@full_build ? '全量打包' : '非全量打包'})".green do
            pod_targets = @analyze_result.pod_targets.uniq
            success_pods = []
            fail_pods = []
            local_pods = []
            external_pods = []
            binary_pods = []
            created_pods = []
            # # 指定线程池的大小（例如，这里创建一个拥有 16 个线程的池）
            # pool = Concurrent::ThreadPoolExecutor.new(max_threads: 16)
            # # 创建一个互斥锁
            # mutex = Mutex.new
            pod_targets.map do |pod_target|
              begin
                version = @version_helper.version(pod_target.pod_name, pod_target.root_spec.version.to_s, @analyze_result.specifications, @configuration, podfile.include_dependencies?)
                # 黑名单（不分全量和非全量）
                next if skip_build?(pod_target)
                # 白名单（有白名单，只看白名单，不分全量和非全量）
                next if !@write_list.nil? && !@write_list.empty? && !@write_list.include?(pod_target.pod_name)
                # 本地库
                if @sandbox.local?(pod_target.pod_name)
                  local_pods << pod_target.pod_name
                  show_skip_tip("#{pod_target.pod_name} 是本地库")
                  next
                end
                # 外部源（如 git）
                if @sandbox.checkout_sources[pod_target.pod_name]
                  external_pods << pod_target.pod_name
                  show_skip_tip("#{pod_target.pod_name} 以external方式引入")
                  next
                end
                # 无源码
                if !@sandbox.local?(pod_target.pod_name) && !pod_target.should_build?
                  binary_pods << pod_target.pod_name
                  show_skip_tip("#{pod_target.pod_name} 无需编译")
                  next
                end
                # 非全量编译、不在白名单中且已经有相应的二进制版本
                if has_created_binary?(pod_target.pod_name, version)
                  created_pods << pod_target.pod_name
                  show_skip_tip("#{pod_target.pod_name}(#{version}) 已经有二进制版本了")
                  next
                end
                # # 提交一个任务给线程池
                # pool.post do
                #   # 在这里执行你的任务逻辑
                #   # 构建产物
                #   builder = Builder.new(pod_target, @sandbox.checkout_sources, @skip_simulator, @configuration)
                #   # result = builder.build
                #   # fail_pods << pod_target.pod_name unless result
                #   # next unless result
                #   builder.create_binary
                #   # 压缩并上传zip
                #   zip_helper = ZipFileHelper.new(pod_target, version, builder.product_dir, builder.build_as_framework?, @configuration)
                #   result = zip_helper.zip_lib
                #   mutex.synchronize do
                #     # 在互斥锁的保护下，安全地修改共享资源
                #     fail_pods << pod_target.pod_name unless result
                #   end
                #   next unless result
                #   # result = zip_helper.upload_zip_lib
                #   # fail_pods << pod_target.pod_name unless result
                #   # next unless result
                #   # 生成二进制podspec并上传
                #   podspec_creator = PodspecUtil.new(pod_target, version, builder.build_as_framework?, @configuration)
                #   bin_spec = podspec_creator.create_binary_podspec
                #   bin_spec_file = podspec_creator.write_binary_podspec(bin_spec)
                #   # result = podspec_creator.push_binary_podspec(bin_spec_file)
                #   # fail_pods << pod_target.pod_name unless result
                #   mutex.synchronize do
                #     # 在互斥锁的保护下，安全地修改共享资源
                #     success_pods << pod_target.pod_name if result
                #   end
                # end
                # 在这里执行你的任务逻辑
                # 构建产物
                builder = Builder.new(pod_target, @sandbox.checkout_sources, @skip_simulator, @configuration)
                # result = builder.build
                # fail_pods << pod_target.pod_name unless result
                # next unless result
                builder.create_binary
                # 压缩并上传zip
                zip_helper = ZipFileHelper.new(pod_target, version, builder.product_dir, builder.build_as_framework?, @configuration)
                result = zip_helper.zip_lib
                fail_pods << pod_target.pod_name unless result
                next unless result
                result = zip_helper.upload_zip_lib
                fail_pods << pod_target.pod_name unless result
                next unless result
                # 生成二进制podspec并上传
                podspec_creator = PodspecUtil.new(pod_target, version, builder.build_as_framework?, @configuration)
                bin_spec = podspec_creator.create_binary_podspec
                bin_spec_file = podspec_creator.write_binary_podspec(bin_spec)
                result = podspec_creator.push_binary_podspec(bin_spec_file)
                fail_pods << pod_target.pod_name unless result
                success_pods << pod_target.pod_name if result
              rescue Pod::StandardError => e
                UI.info "`#{pod_target}`失败，原因：#{e}".red
                fail_pods << pod_target.pod_name
                next
              end
            end
            # 关闭线程池，等待所有任务完成
            # pool.shutdown
            # pool.wait_for_termination
            results = {
              'Total' => pod_targets,
              'Success' => success_pods,
              'Fail' => fail_pods,
              'Local' => local_pods,
              'External' => external_pods,
              'No Source File' => binary_pods,
              'Created Binary' => created_pods,
              'Black List' => @black_list || [],
              'Write List' => @write_list || []
            }

            show_results(results)
            results
          end
        end

        def show_skip_tip(title)
          UI.info title.yellow
        end

        # 是否跳过编译
        def skip_build?(pod_target)
          !@black_list.nil? && !@black_list.empty? && @black_list.include?(pod_target.pod_name)
        end

        # 展示结果
        def show_results(results)
          UI.title '打包结果：'.green do
            UI.info '——————————————————————————————————'.green
            UI.info "|#{'Type'.center(20)}|#{'Count'.center(11)}|".green
            UI.info '——————————————————————————————————'.green
            results.each do |key, value|
              UI.info "|#{key.center(20)}|#{value.size.to_s.center(11)}|".green
            end
            UI.info '——————————————————————————————————'.green

            # 打印出失败的 target
            unless results['Fail'].empty?
              UI.info "\n打包失败的库：#{results['Fail']}".red
            end
          end
        end

        # 是否已经有二进制版本了
        def has_created_binary?(pod_name, version)
          # name 或 version 为nil
          return false if pod_name.nil? || version.nil?
          # 是否全量打包
          return false if @full_build
          # 是否在白名单中
          return false if !@write_list.nil? && !@write_list.empty? && @write_list.include?(pod_name)
          sources_manager = Config.instance.sources_manager
          binary_source = sources_manager.binary_source
          result = false
          begin
            specification = binary_source.specification(pod_name, version)
            result = true unless specification.nil?
          rescue Pod::StandardError => e
            result = false
          end
          result
        end
      end
    end
  end
end