minor

.drone.yml

@@ -25,10 +25,20 @@ steps:
         from_secret: DOCKER_PASSWORD
   commands:
   - echo "$DOCKER_PASSWORD" | docker login -u "$DOCKER_USERNAME" --password-stdin
-  - docker build -t="$DOCKER_USERNAME/consumer" .
-  - docker build -t="$DOCKER_USERNAME/consumer:$DRONE_BUILD_NUMBER" .
-  - docker push "$DOCKER_USERNAME/consumer"
-  - docker push "$DOCKER_USERNAME/consumer:$DRONE_BUILD_NUMBER"
+  - docker build -t="$DOCKER_USERNAME/consumer_api" .
+  - docker build -t="$DOCKER_USERNAME/consumer_api:$DRONE_BUILD_NUMBER" .
+  - docker push "$DOCKER_USERNAME/consumer_api"
+  - docker push "$DOCKER_USERNAME/consumer_api:$DRONE_BUILD_NUMBER"
+
+- name: slack
+  image: plugins/slack
+  settings:
+    webhook:
+      from_secret: slack_webhook
+    channel: alerts-ci
+    icon_url: https://cloudcdn.tormakristof.eu/static/drone.svg
+  when:
+    status: [ failure ]
 
 services:
 - name: docker

Dockerfile

@@ -1,9 +1,13 @@
-FROM python:3
+FROM python:3.8-slim
 
-WORKDIR /app
+ADD consumer_api requirements.txt /consumer_api/
+WORKDIR /consumer_api/
 
-COPY . ./
+ENV PIP_NO_CACHE_DIR=true
+ENV TZ Europe/Budapest
+RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
 
-RUN pip3 install --no-cache-dir -r requirements.txt
+RUN pip3 install -r requirements.txt
 
-CMD ["python3" "app.py"]
+EXPOSE 8000
+CMD ["gunicorn", "-b", "0.0.0.0:8000", "app:app"]

consumer_api/app.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+import sentry_sdk
+from flask import Flask
+import os
+from db import redis_client
+from views import ConsumersView, LogView, SyncView
+
+"""
+Main Flask RESTful API
+"""
+
+__author__ = "@tormakris"
+__copyright__ = "Copyright 2020, GoldenPogácsa Team"
+__module_name__ = "app"
+__version__text__ = "1"
+
+sentry_sdk.init("https://0a106e104e114bc9a3fa47f9cb0db2f4@sentry.kmlabz.com/10")
+
+app = Flask(__name__)
+app.config['REDIS_URL'] = os.environ['REDIS_URL']
+app.config['LOCAL_UUID'] = os.environ['LOCAL_UUID']
+
+redis_client.init_app(app)
+
+for view in [ConsumersView, LogView, SyncView]:
+    view.register(app, trailing_slash=False)
+
+if __name__ == "__main__":
+    app.run(debug=True)

consumer_api/db.py

@@ -0,0 +1,3 @@
+from flask_redis import FlaskRedis
+
+redis_client = FlaskRedis()

consumer_api/views/__init__.py

@@ -0,0 +1,3 @@
+from .consumers_view import ConsumersView
+from .log_view import LogView
+from .sync_view import SyncView

consumer_api/views/consumers_view.py

@@ -0,0 +1,12 @@
+import json
+
+from db import redis_client
+from flask import jsonify
+from flask_classful import FlaskView
+
+
+class ConsumersView(FlaskView):
+
+    def get(self):
+        consumer_list = json.loads((redis_client.get("consumer_list") or b"{}").decode('utf-8'))
+        return jsonify([v['ip'] for k, v in consumer_list.items()])

consumer_api/views/log_view.py

@@ -0,0 +1,9 @@
+from flask import request, current_app, Response
+from flask_classful import FlaskView
+
+
+class LogView(FlaskView):
+
+    def post(self):
+        current_app.logger.info(f"New message: {request.json['message']}")
+        return Response(status = 204)

consumer_api/views/sync_view.py

@@ -0,0 +1,32 @@
+import json
+from flask import request, current_app, jsonify
+from flask_classful import FlaskView
+from db import redis_client  # ez nagyon otvar
+
+
+class SyncView(FlaskView):
+
+    def post(self):
+        remote_uuid = request.json['uuid']
+        remote_ip = request.remote_addr
+
+        consumer_list = json.loads((redis_client.get("consumer_list") or b"{}").decode('utf-8'))
+
+        # Log something about it
+        if remote_uuid not in consumer_list.keys():
+            current_app.logger.info(f"New consumer registered (unknown UUID): {remote_uuid} at {remote_ip}")
+        else:  # known
+            if consumer_list[remote_uuid]['ip'] != remote_ip:
+                current_app.logger.info(f"Address of consumer {remote_uuid} changed to {remote_ip}")
+
+        consumer_list.update(
+            {remote_uuid: {"ip": remote_ip}}
+        )
+
+        redis_client.set("consumer_list", json.dumps(consumer_list).encode('utf-8'))
+
+        response = {
+            "uuid": current_app.config['LOCAL_UUID']
+        }
+
+        return jsonify(response)

requirements.txt

@@ -1 +1,5 @@
 sentry_sdk
+flask
+flask-redis
+flask-classful
+gunicorn