update llb-url script

TrekkieCoder · Oct 23, 2024 · 51ab039 · 51ab039
1 parent a3853e5
commit 51ab039
Showing 1 changed file with 19 additions and 5 deletions.
diff --git a/tools/k8s/mkllb-url b/tools/k8s/mkllb-url
@@ -1,11 +1,11 @@
 #!/bin/bash
 
 usage() {
-    echo "Usage: $0 -a <ip-addr> -z <zone> -t <type>"
-    echo "       $0 -a <ip-addr> -z <zone> -t <type> -x"
+    echo "Usage: $0 -a <ip-addr> -z <zone> -t <type> -c <cloud>"
+    echo "       $0 -a <ip-addr> -z <zone> -t <type> -c <cloud> -x"
 }
 
-if [[ $# -gt 7 ]] || [[ $# -lt 6 ]]; then
+if [[ $# -gt 9 ]] || [[ $# -lt 6 ]]; then
    usage
    exit
 fi
@@ -18,30 +18,44 @@ if [[ ! -f /usr/local/sbin/kubectl ]]; then
 fi
 
 addr=""
+cloud="bm"
 zone="llb"
 utype="default"
 cmd="apply"
 
-while getopts a:z:t:x opt 
+while getopts a:z:t:c:x opt
 do
     case "${opt}" in
         a) addr=${OPTARG};;
         z) zone=${OPTARG};;
         t) utype=${OPTARG};;
         x) cmd="delete";;
+        c) cloud=${OPTARG};;
         ?) usage;exit;;
     esac
 done
 
 url="http://${addr}:11111"
 
+laddr=${addr}
+if [[ ${cloud} == "aws" ]]; then
+  if [[ ! -f /usr/local/bin/aws ]]; then
+    apt install -y unzip
+    curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip"
+    unzip awscliv2.zip && ./aws/install
+  fi
+  token=`curl -m 10 -X PUT http://169.254.169.254/latest/api/token -H "X-aws-ec2-metadata-token-ttl-seconds: 21600"` && \
+  addr=`curl -m 10 -H "X-aws-ec2-metadata-token: $token" http://169.254.169.254/latest/meta-data/public-ipv4`
+fi
+
 if [[ ${utype} == "hostcidr" ]]; then
   url=${addr}
 fi
 
 echo "============"
 echo "Applying CRD"
 echo "============"
+echo name llb-${laddr}
 echo url $url
 echo zone $zone
 echo utype $utype
@@ -52,7 +66,7 @@ cat <<EOF | kubectl ${cmd} -f -
 apiVersion: "loxiurl.loxilb.io/v1"
 kind: LoxiURL
 metadata:
-  name: llb-${addr}
+  name: llb-${laddr}
 spec:
   loxiURL: $url
   zone: llb