module Souls
  class Sql < Thor
    desc "create_instance", "Create Google Cloud SQL - PostgreSQL13"
    method_option :region, default: "", aliases: "--region", desc: "Google Cloud Platform Region"
    method_option :mysql, type: :boolean, default: false, aliases: "--mysql", desc: "Set Cloud SQL Type to MySQL"
    def create_instance
      prompt = TTY::Prompt.new
      password = prompt.mask("Set DB PassWord:")
      app_name = Souls.configuration.app
      project_id = Souls.configuration.project_id
      instance_name = Souls.configuration.instance_name
      region = Souls.configuration.region if options[:region].blank?
      db_type = options[:mysql] ? "MYSQL_8_0" : "POSTGRES_13"

      zone = "#{region}-b"
      system("gcloud config set project #{project_id}")
      system(
        "gcloud sql instances create #{instance_name} \
              --database-version=#{db_type} --cpu=1 --memory=4096MB --zone=#{zone} \
              --root-password='#{password}' --database-flags cloudsql.iam_authentication=on"
      )
      instance_ip = `gcloud sql instances list | grep #{instance_name} | awk '{print $5}'`.strip
      Dir.chdir(Souls.get_api_path.to_s) do
        file_path = ".env"
        File.open(file_path, "w") do |line|
          line.write(<<~TEXT)
            SOULS_DB_HOST=#{instance_ip}
            SOULS_DB_PW=#{password}
            SOULS_DB_USER=postgres
            SOULS_GCP_PROJECT_ID=#{project_id}
            SOULS_SECRET_KEY_BASE='#{SecureRandom.base64(64)}'
            TZ="#{region_to_timezone(region: region)}"
          TEXT
        end
      end
      Dir.chdir(Souls.get_mother_path.to_s) do
        file_path = ".env.production"
        File.open(file_path, "w") do |line|
          line.write(<<~TEXT)
            SOULS_DB_HOST="/cloudsql/#{project_id}:#{region}:#{instance_name}"
            SOULS_DB_PW=#{password}
            SOULS_DB_USER=postgres
            SOULS_APP_NAME=#{app_name}
            SOULS_GCP_PROJECT_ID=#{project_id}
            SOULS_GCP_REGION=#{region}
            SOULS_GCLOUDSQL_INSTANCE="#{project_id}:#{region}:#{instance_name}"
            SOULS_SECRET_KEY_BASE='#{SecureRandom.base64(64)}'
            TZ="#{region_to_timezone(region: region)}"
          TEXT
        end
      end
      Souls::Github.new
    rescue Thor::Error => e
      raise(Thor::Error, e)
    end

    desc "list", "Show Cloud SQL Instances List"
    def list
      system("gcloud sql instances list")
    rescue Thor::Error => e
      raise(Thor::Error, e)
    end

    desc "setup_private_ip", "Enable Private IP"
    def setup_private_ip
      create_ip_range
      create_vpc_connector
      assign_network
    rescue Thor::Error => e
      raise(Thor::Error, e)
    end

    desc "assign_network", "Assign Network"
    def assign_network
      app_name = Souls.configuration.app
      instance_name = Souls.configuration.instance_name
      project_id = Souls.configuration.project_id
      system("gcloud beta sql instances patch #{instance_name} --project=#{project_id} --network=#{app_name}")
    rescue Thor::Error => e
      raise(Thor::Error, e)
    end

    desc "create_ip_range", "Create VPC Adress Range"
    def create_ip_range
      app_name = Souls.configuration.app
      project_id = Souls.configuration.project_id
      system(
        "
            gcloud compute addresses create #{app_name}-ip-range \
              --global \
              --purpose=VPC_PEERING \
              --prefix-length=16 \
              --description='peering range for SOULs' \
              --network=#{app_name} \
              --project=#{project_id}"
      )
    rescue Thor::Error => e
      raise(Thor::Error, e)
    end

    desc "create_vpc_connector", "Create VPC-PEERING Connect"
    def create_vpc_connector
      app_name = Souls.configuration.app
      project_id = Souls.configuration.project_id
      system(
        "
            gcloud services vpc-peerings connect \
              --service=servicenetworking.googleapis.com \
              --ranges=#{app_name}-ip-range \
              --network=#{app_name} \
              --project=#{project_id}
            "
      )
    rescue Thor::Error => e
      raise(Thor::Error, e)
    end

    desc "assgin_ip", "Add Current Grobal IP to White List"
    def assign_ip(instance_name: "", ip: "")
      ip = `curl inet-ip.info` if ip.blank?
      project_id = Souls.configuration.project_id if instance_name.blank?
      instance_name = Souls.configuration.instance_name if instance_name.blank?
      system(
        "
            gcloud beta sql instances patch #{instance_name} \
              --project=#{project_id} \
              --assign-ip \
              --authorized-networks=#{ip}
            "
      )
    rescue Thor::Error => e
      raise(Thor::Error, e)
    end

    private

    def region_to_timezone(region: "asia-northeast1")
      if region.include?("asia")
        "Asia/Tokyo"
      elsif region.include?("europe")
        "Europe/Amsterdam"
      else
        "America/Los_Angeles"
      end
    end
  end
end