# =============================================================================
# CLOUDFLARE : Access : Groups
# =============================================================================

locals {
  # SAML groups from Okta
  saml_groups = {
    contractors          = "Contractors"
    infrastructure_admin = "GL_Users_Infrastructure Admin"
    sales_engineering    = "GL_Users_Sales Engineering"
    sales                = "GL_Users_Sales"
    it_admin             = "GL_Users_IT Admin"
  }

  # Allowed countries
  allowed_countries = ["FR", "DE", "US", "GB"]
  blocked_countries = ["CN", "RU", "AF", "BY", "CD", "CU", "IR", "IQ", "KP", "MM", "SD", "SY", "UA", "ZW"]

  main_countries            = ["FR"]
  europe_countries          = ["AL","AD","AT","AX","BA","BE","BG","BY","CH","CY","CZ","DE","DK","EE","ES","FI","FR","FO","GB","GG","GI","GR","HR","HU","IE","IM","IS","IT","JE","LI","LT","LU","LV","MC","MD","ME","MK","MT","NL","NO","PL","PT","RO","RS","SE","SI","SK","SM","UA","VA"]
  afrique_countries         = ["AO","BF","BI","BJ","BW","CD","CF","CG","CI","CM","CV","DJ","DZ","EG","EH","ER","ET","GA","GH","GM","GN","GQ","GW","KE","KM","LR","LS","LY","MA","MG","ML","MR","MU","MW","MZ","NA","NE","NG","RE","RW","SC","SD","SH","SL","SN","SO","SS","ST","SZ","TD","TF","TG","TN","TZ","UG","YT","ZA","ZM","ZW"]
  america_north_countries   = ["CA","US","MX","BM","PM","GL","UM"]
  america_central_countries = ["AG","AI","AW","BB","BZ","CR","CU","DM","DO","GD","GP","GT","HN","HT","JM","KN","KY","LC","MF","MQ","MS","NI","PA","PR","SV","SX","TC","TT","VC","VG","VI"]
  america_south_countries   = ["AR","BO","BR","CL","CO","EC","FK","GF","GY","PE","PY","SR","UY","VE"]
  asie_countries            = ["AF","AM","AZ","BD","BH","BN","BT","CN","GE","HK","ID","IL","IN","IQ","IR","JO","JP","KG","KH","KP","KR","KW","KZ","LA","LB","LK","MM","MN","MO","MY","NP","OM","PH","PK","PS","QA","SA","SG","SY","TH","TJ","TL","TM","TR","TW","UZ","VN","YE"]
  oceanie_countries         = ["AS","AU","CK","FJ","FM","GU","HM","KI","MH","MP","NC","NF","NR","NU","NZ","PF","PG","PN","PW","SB","TK","TO","TV","UM","VU","WF","WS"]
  antarctique_countries     = ["AQ", "BV"]
  other_countries           = ["IO","GS"]

  # On fusionne *toutes* les zones dans une seule liste
  all_countries = flatten([
    local.europe_countries,
    local.afrique_countries,
    local.america_north_countries,
    local.america_central_countries,
    local.america_south_countries,
    local.asie_countries,
    local.oceanie_countries,
    local.antarctique_countries,
    local.other_countries,
  ])

  # On retire les pays "main"
  blocked_countries_except_main = compact([
    for code in local.all_countries :
    (contains(local.main_countries, code) ? null : code)
  ])

  # On retire les pays "europe"
  blocked_countries_except_europe = compact([
    for code in local.all_countries :
    (contains(local.europe_countries, code) ? null : code)
  ])

  # On retire les pays "afrique"
  blocked_countries_except_afrique= compact([
    for code in local.all_countries :
    (contains(local.afrique_countries, code) ? null : code)
  ])

  # On retire les pays "america_north"
  blocked_countries_except_america_north= compact([
    for code in local.all_countries :
    (contains(local.america_north_countries, code) ? null : code)
  ])
  
  # On retire les pays "america_central"
  blocked_countries_except_america_central= compact([
    for code in local.all_countries :
    (contains(local.america_central_countries, code) ? null : code)
  ])
  
  # On retire les pays "america_south"
  blocked_countries_except_america_south= compact([
    for code in local.all_countries :
    (contains(local.america_south_countries, code) ? null : code)
  ])
  
  # On retire les pays "asie"
  blocked_countries_except_asie= compact([
    for code in local.all_countries :
    (contains(local.asie_countries, code) ? null : code)
  ])
  
  # On retire les pays "oceanie"
  blocked_countries_except_oceanie= compact([
    for code in local.all_countries :
    (contains(local.oceanie_countries, code) ? null : code)
  ])
  
  # On retire les pays "antarctique"
  blocked_countries_except_antarctique= compact([
    for code in local.all_countries :
    (contains(local.antarctique_countries, code) ? null : code)
  ])
  
  # On retire les pays "other"
  blocked_countries_except_other= compact([
    for code in local.all_countries :
    (contains(local.other_countries, code) ? null : code)
  ])

  # OS posture checks
  os_posture_checks = [
    var.cloudflare_linux_posture_id,
    var.cloudflare_macos_posture_id,
    var.cloudflare_windows_posture_id
  ]
}

# SAML Rule Groups
resource "cloudflare_zero_trust_access_group" "saml_groups" {
  for_each   = local.saml_groups
  account_id = local.cloudflare_account_id
  name       = each.value

  include = [{
    saml = {
      identity_provider_id = var.cloudflare_okta_identity_provider_id
      attribute_name       = "groups"
      attribute_value      = each.value
    }
  }]
}

# Geographic Rule Groups
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group" {
  account_id = local.cloudflare_account_id
  name       = "GRP_Localisation_Country Requirements"

  include = [
    for country in local.allowed_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries : {
      geo = {
        country_code = country
      }
    }
  ]
}

#
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group_main" {
  account_id = local.cloudflare_account_id

  name       = "GRP_Localisation Country Requirements : Main"
  include = [
    for country in local.main_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries_except_main : {
      geo = {
        country_code = country
      }
    }
  ]
}

#
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group_europe" {
  account_id = local.cloudflare_account_id

  name       = "GRP_Localisation Country Requirements : Europe"
  include = [
    for country in local.europe_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries_except_europe : {
      geo = {
        country_code = country
      }
    }
  ]
}

#
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group_afrique" {
  account_id = local.cloudflare_account_id

  name       = "GRP_Localisation Country Requirements : Afrique"
  include = [
    for country in local.afrique_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries_except_afrique : {
      geo = {
        country_code = country
      }
    }
  ]
}


#
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group_america_north" {
  account_id = local.cloudflare_account_id

  name       = "GRP_Localisation Country Requirements : America North"
  include = [
    for country in local.america_north_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries_except_america_north : {
      geo = {
        country_code = country
      }
    }
  ]
}

#
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group_america_central" {
  account_id = local.cloudflare_account_id

  name       = "GRP_Localisation Country Requirements : America Central"
  include = [
    for country in local.america_central_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries_except_america_central : {
      geo = {
        country_code = country
      }
    }
  ]
}

#
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group_america_south" {
  account_id = local.cloudflare_account_id

  name       = "GRP_Localisation Country Requirements : America South"
  include = [
    for country in local.america_south_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries_except_america_south : {
      geo = {
        country_code = country
      }
    }
  ]
}

#
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group_asie" {
  account_id = local.cloudflare_account_id

  name       = "GRP_Localisation Country Requirements : asie"
  include = [
    for country in local.asie_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries_except_asie : {
      geo = {
        country_code = country
      }
    }
  ]
}

#
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group_oceanie" {
  account_id = local.cloudflare_account_id

  name       = "GRP_Localisation Country Requirements : oceanie"
  include = [
    for country in local.oceanie_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries_except_oceanie : {
      geo = {
        country_code = country
      }
    }
  ]
}

#
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group_antarctique" {
  account_id = local.cloudflare_account_id

  name       = "GRP_Localisation Country Requirements : antarctique"
  include = [
    for country in local.antarctique_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries_except_antarctique : {
      geo = {
        country_code = country
      }
    }
  ]
}
 
#
resource "cloudflare_zero_trust_access_group" "country_requirements_rule_group_other" {
  account_id = local.cloudflare_account_id

  name       = "GRP_Localisation Country Requirements : other"
  include = [
    for country in local.other_countries : {
      geo = {
        country_code = country
      }
    }
  ]
  exclude = [
    for country in local.blocked_countries_except_other : {
      geo = {
        country_code = country
      }
    }
  ]
}

# Device Posture Rule Groups
resource "cloudflare_zero_trust_access_group" "latest_os_version_requirements_rule_group" {
  account_id = local.cloudflare_account_id
  name       = "GL_OS Version Requirements"

  include = [
    for posture_id in local.os_posture_checks : {
      device_posture = {
        integration_uid = posture_id
      }
    }
  ]
}

# Composite Rule Groups
resource "cloudflare_zero_trust_access_group" "employees_rule_group" {
  account_id = local.cloudflare_account_id
  name       = "GL_Users_Employees"

  include = [
    for group_key in ["it_admin", "sales", "sales_engineering", "infrastructure_admin"] : {
      group = {
        id = cloudflare_zero_trust_access_group.saml_groups[group_key].id
      }
    }
  ]
}

resource "cloudflare_zero_trust_access_group" "sales_team_rule_group" {
  account_id = local.cloudflare_account_id
  name       = "GL_Users_Sales Team"

  include = [
    for group_key in ["sales", "sales_engineering"] : {
      group = {
        id = cloudflare_zero_trust_access_group.saml_groups[group_key].id
      }
    }
  ]
}

resource "cloudflare_zero_trust_access_group" "admins_rule_group" {
  account_id = local.cloudflare_account_id
  name       = "GL_Users_Administrators"

  include = [
    for group_key in ["it_admin", "infrastructure_admin"] : {
      group = {
        id = cloudflare_zero_trust_access_group.saml_groups[group_key].id
      }
    }
  ]
}

resource "cloudflare_zero_trust_access_group" "contractors_rule_group" {
  account_id = local.cloudflare_account_id
  name       = "GL_Users_Contractors Extended"

  include = [
    {
      group = {
        id = cloudflare_zero_trust_access_group.saml_groups["contractors"].id
      }
    },
    {
      email_domain = {
        domain = var.cloudflare_email_domain
      }
    }
  ]
}