From b0b690a7515b08e8ce47f20047a40644c7257732 Mon Sep 17 00:00:00 2001
From: Marius Storhaug <marstor@hotmail.com>
Date: Fri, 8 Sep 2023 20:46:36 +0200
Subject: [PATCH] Update Get-PublicIP

---
 src/PublicIP/PublicIP.psm1 | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/src/PublicIP/PublicIP.psm1 b/src/PublicIP/PublicIP.psm1
index 9dc0942..20321ca 100644
--- a/src/PublicIP/PublicIP.psm1
+++ b/src/PublicIP/PublicIP.psm1
@@ -1,6 +1,16 @@
-Function Get-PublicIP {
-    #Invoke-RestMethod -Uri https://api.myip.com/
-    Invoke-RestMethod -Uri https://ipinfo.io/json
+$script:providerMap = @{
+    MyIP   = 'https://api.myip.com/'
+    IPInfo = 'https://ipinfo.io/json'
+}
+
+Function Get-PublicIP {
+    param(
+        [Parameter(Mandatory = $false)]
+        [ValidateScript({$providerMap.Keys})]
+        $Provider = 'IPInfo'
+    )
+
+    Invoke-RestMethod -Uri $providerMap[$Provider]
 }
 
 Function Get-IPConfig {