first commit

Author: MegatronKing

.gitignore

@@ -0,0 +1,6 @@
+.DS_STORE
+/build
+/dist
+/reqable_scripting.egg-info
+__pycache__
+*.cb

README.md

@@ -0,0 +1,12 @@
+# Reqable Scripting Framework
+
+This is the Python scripting framework for Reqable.
+
+# Installation
+```
+pip3 install reqable-scripting
+```
+
+# Documentation
+
+https://reqable.com/docs/capture/addons

reqable/__init__.py

@@ -0,0 +1,4 @@
+import main
+
+if __name__== '__main__':
+  main.main()

reqable/addons.py

@@ -0,0 +1,36 @@
+# API Docs: https://reqable.com/docs/capture/addons
+
+from reqable import *
+
+def onRequest(context, request):
+  # Print url to console
+  # print('request url ' + context.url)
+
+  # Update or add a query parameter
+  # request.queries['foo'] = 'bar'
+
+  # Update or add a http header
+  # request.headers['foo'] = 'bar'
+
+  # Replace http body with a text
+  # request.body = 'Hello World'
+
+  # Map with a local file
+  # request.body.file('~/Desktop/body.json')
+
+  # Convert to dict if the body is a JSON
+  # request.body.jsonify()
+  # Update the JSON content
+  # request.body['foo'] = 'bar'
+
+  # Done
+  return request
+
+def onResponse(context, response):
+  # Update status code
+  # response.code = 404
+
+  # APIs are same as `onRequest`
+
+  # Done
+  return response

reqable/addons_mini.py

@@ -0,0 +1,9 @@
+# API Docs: https://reqable.com/docs/capture/addons
+
+from reqable import *
+
+def onRequest(context, request):
+  return request
+
+def onResponse(context, response):
+  return response

reqable/main.py

@@ -0,0 +1,43 @@
+import sys
+import json
+from reqable import CaptureContext, CaptureHttpRequest, CaptureHttpResponse
+import addons
+
+def main():
+  argv = sys.argv[1:]
+  if len(argv) != 2:
+    raise Exception('Invalid reqable script arguments')
+  type = argv[0]
+  if type == 'request':
+    onRequest(argv[1])
+  elif type == 'response':
+    onResponse(argv[1])
+  else:
+    raise Exception('Unexpected type ' + type)
+
+def onRequest(request):
+  with open(request, 'r', encoding='UTF-8') as content:
+    data = json.load(content)
+    context = CaptureContext(data['context'])
+    result = addons.onRequest(context, CaptureHttpRequest(data['request']))
+    if result is not None:
+      with open(request + '.cb', 'w', encoding='UTF-8') as callback:
+        callback.write(json.dumps({
+          'request': result.serialize(),
+          'shared': context.shared,
+        }))
+
+def onResponse(response):
+  with open(response, 'r', encoding='UTF-8') as content:
+    data = json.load(content)
+    context = CaptureContext(data['context'])
+    result = addons.onResponse(context, CaptureHttpResponse(data['response']))
+    if result is not None:
+      with open(response + '.cb', 'w', encoding='UTF-8') as callback:
+        callback.write(json.dumps({
+          'response': result.serialize(),
+          'shared': context.shared,
+        }))
+
+if __name__== '__main__':
+  main()