#
# 参考文献
# https://github.com/firebase/flutterfire/blob/master/packages/firebase_crashlytics/firebase_crashlytics/ios/crashlytics_add_upload_symbols
# https://github.com/MagicalWater/Base-APP-Env/blob/master/fastlane/actions/xcode_parse.rb
#

require 'xcodeproj'
require 'plist'
require 'optparse'

# Dictionary to hold command line arguments
options_dict = {}

# Parse command line arguments into options_dict
OptionParser.new do |options|
    options.banner = "Setup the Tencent to an Xcode target."

    options.on("-p", "--projectDirectory=DIRECTORY", String, "Directory of the Xcode project") do |dir|
        options_dict[:project_dir] = dir
    end

    options.on("-n", "--projectName=NAME", String, "Name of the Xcode project (ex: Runner.xcodeproj)") do |name|
        options_dict[:project_name] = name
    end

    options.on("-a", "--appId=APPID", String, "App ID for Tencent") do |opts|
        options_dict[:app_id] = opts
    end

    options.on("-u", "--universalLink=UNIVERSALLINK", String, "Universal Link for Tencent") do |opts|
        options_dict[:universal_link] = opts
    end
end.parse!

# Minimum required arguments are a project directory and project name
unless (options_dict[:project_dir] and options_dict[:project_name])
    abort("Must provide a project directory and project name.\n")
end

# Path to the Xcode project to modify
project_path = File.join(options_dict[:project_dir], options_dict[:project_name])

unless (File.exist?(project_path))
   abort("Project at #{project_path} does not exist. Please check paths manually.\n");
end

# Actually open and modify the project
project = Xcodeproj::Project.open(project_path)
project.targets.each do |target|
    if (target.name == "Runner")
        app_id = options_dict[:app_id]
        universal_link = options_dict[:universal_link]
        applinks = "applinks:#{URI.parse(universal_link).host}"

        sectionObject = {}
        project.objects.each do |object|
            if object.uuid == target.uuid
                sectionObject = object
                break
            end
        end
        sectionObject.build_configurations.each do |config|
            infoplist = config.build_settings["INFOPLIST_FILE"]
            if (!infoplist)
                puts("INFOPLIST_FILE is not exist")
                exit(0)
            end
            infoplistFile = File.join(options_dict[:project_dir], infoplist)
            if !File.exist?(infoplistFile)
                puts("#{infoplist} is not exist")
                exit(0)
            end
            result = Plist.parse_xml(infoplistFile, marshal: false)
            if (!result)
                result = {}
            end
            urlTypes = result["CFBundleURLTypes"]
            if (!urlTypes)
                urlTypes = []
                result["CFBundleURLTypes"] = urlTypes
            end
            isUrlTypeExist = urlTypes.any? { |urlType| urlType["CFBundleURLSchemes"] && (urlType["CFBundleURLSchemes"].include? "tencent#{app_id}") }
            if (!isUrlTypeExist)
                urlTypes << {
                    "CFBundleTypeRole": "Editor",
                    "CFBundleURLName": "tencent",
                    "CFBundleURLSchemes": [ "tencent#{app_id}" ]
                }
                File.write(infoplistFile, Plist::Emit.dump(result))
            end
            queriesSchemes = result["LSApplicationQueriesSchemes"]
            if (!queriesSchemes)
                queriesSchemes = []
                result["LSApplicationQueriesSchemes"] = queriesSchemes
            end
            tencentSchemes = [
                "mqq",
                "mqqapi",
                "tim",
                "mqqopensdknopasteboard",
                "mqqopensdknopasteboardios16",
                "mqqopensdkapiV2",
                "mqqgamebindinggroup",
                "mqqopensdkavatar",
                "mqqopensdkfriend",
                "mqqopensdklaunchminiapp",
                "mqzone",
                "tencentapi.qq.reqContent",
                "tencentapi.qzone.reqContent",
                "mqqthirdappgroup",
                "mqqopensdkminiapp",
            ]
            if (tencentSchemes.any? { |scheme| !(queriesSchemes.include? scheme)} )
                tencentSchemes.each do |scheme|
                    if (!(queriesSchemes.include? scheme))
                        queriesSchemes << scheme
                    end
                end
                File.write(infoplistFile, Plist::Emit.dump(result))
            end
            security = result["NSAppTransportSecurity"]
            if (!security)
                security = {}
                result["NSAppTransportSecurity"] = security
            end
            if (security["NSAllowsArbitraryLoads"] != true)
                security["NSAllowsArbitraryLoads"] = true
                File.write(infoplistFile, Plist::Emit.dump(result))
            end
            if (security["NSAllowsArbitraryLoadsInWebContent"] != true)
                security["NSAllowsArbitraryLoadsInWebContent"] = true
                File.write(infoplistFile, Plist::Emit.dump(result))
            end
        end
        sectionObject.build_configurations.each do |config|
            codeSignEntitlements = config.build_settings["CODE_SIGN_ENTITLEMENTS"]
            if (!codeSignEntitlements)
                codeSignEntitlements = "Runner/Runner.entitlements"
                config.build_settings["CODE_SIGN_ENTITLEMENTS"] = codeSignEntitlements
                project.save()
            end
            codeSignEntitlementsFile = File.join(options_dict[:project_dir], codeSignEntitlements)
            if !File.exist?(codeSignEntitlementsFile)
                content = Plist::Emit.dump({})
                File.write(codeSignEntitlementsFile, content)
            end
            runnerTargetMainGroup = project.main_group.find_subpath('Runner', false)
            isRefExist = runnerTargetMainGroup.files.any? { |file| file.path.include? File.basename(codeSignEntitlementsFile) }
            if !isRefExist
                runnerTargetMainGroup.new_reference(File.basename(codeSignEntitlementsFile))
                project.save()
            end
            result = Plist.parse_xml(codeSignEntitlementsFile, marshal: false)
            if (!result)
                result = {}
            end
            domains = result["com.apple.developer.associated-domains"]
            if (!domains)
                domains = []
                result["com.apple.developer.associated-domains"] = domains
            end
            isApplinksExist = domains.include? applinks
            if (!isApplinksExist)
                domains << applinks
                File.write(codeSignEntitlementsFile, Plist::Emit.dump(result))
            end
        end
    end
end