added main.py

HariSekhon · HariSekhon · commit f26a380d667a · 2021-05-24T16:26:56.000+01:00
diff --git a/gcp_cloud_function_proxy/main.py b/gcp_cloud_function_proxy/main.py
@@ -0,0 +1,62 @@
+#!/usr/bin/env python
+#  coding=utf-8
+#  vim:ts=4:sts=4:sw=4:et
+#
+#  Author: Hari Sekhon
+#  Date: 2021-05-24 16:03:30 +0100 (Mon, 24 May 2021)
+#
+#  https://github.com/HariSekhon/pytools
+#
+#  License: see accompanying Hari Sekhon LICENSE file
+#
+#  If you're using my code you're welcome to connect with me on LinkedIn
+#  and optionally send me feedback to help steer this or other code I publish
+#
+#  https://www.linkedin.com/in/HariSekhon
+#
+
+"""
+
+GCP Cloud Function to query ifconfig.co to show our IP information for debugging VPC Connector access
+routing via specified VPC Network to using default NAT Gateway
+
+Example usage:
+
+Check GCF source IP to compare if it's permitted through Cloudflare / Firewall rules
+
+Test request examples:
+
+    { "url": "http://ifconfig.co/json" }
+
+defaults to http:// if not specified:
+
+    { "url": "ifconfig.co/json" }
+
+
+Tested on GCP Cloud Functions with Python 3.9
+
+"""
+
+# https://cloud.google.com/functions/docs/writing/specifying-dependencies-python
+
+import json
+import requests
+
+def main(request):
+    """Responds to any HTTP request.
+    Args:
+        request (flask.Request): HTTP request object.
+    Returns:
+        The response text or any set of values that can be turned into a
+        Response object using
+        `make_response <http://flask.pocoo.org/docs/1.0/api/#flask.Flask.make_response>`.
+    """
+    data = json.loads(request.data)
+    url = data['url']
+    if '://' not in url:
+        url = 'http://' + url
+    req = requests.get(url)
+    status_code = req.status_code
+    status_message = req.reason
+    content = req.text
+    return "{} {}\n\n{}".format(status_code, status_message, content)
