main: Added worker interactions

This commit is contained in:
Nathan Woodburn 2023-08-16 16:59:57 +10:00
parent aa8dc9476e
commit 02457ceddf
Signed by: nathanwoodburn
GPG Key ID: 203B000478AD0EF1
6 changed files with 93 additions and 11 deletions

View File

@ -1 +1,2 @@
LICENCE-API=y6X3N9d4nqXYjJ4x LICENCE-API=y6X3N9d4nqXYjJ4x
WORKER_KEY=N5xKQyt280VBa7wl

2
.gitignore vendored
View File

@ -4,3 +4,5 @@
sites.txt sites.txt
licence_key.txt licence_key.txt
workers.txt

View File

@ -25,3 +25,9 @@ Install prerequisites:
chmod +x install.sh chmod +x install.sh
./install.sh ./install.sh
``` ```
Add worker to master server:
```
curl -X POST http://master-server-ip:5000/add-worker?worker=worker-name&ip=worker-server-ip -H "key: api-key"
```

76
main.py
View File

@ -1,6 +1,7 @@
from flask import Flask, request, jsonify from flask import Flask, request, jsonify
import dotenv import dotenv
import os import os
import requests
dotenv.load_dotenv() dotenv.load_dotenv()
@ -44,10 +45,37 @@ def new_site():
if site_exists(domain): if site_exists(domain):
return jsonify({'error': 'Domain already exists', 'success': 'false'}) return jsonify({'error': 'Domain already exists', 'success': 'false'})
# Check if domain contains http:// or https://
if domain.startswith("http://") or domain.startswith("https://"):
return jsonify({'error': 'Domain should not contain http:// or https://', 'success': 'false'})
# Check if worker file exists
workers = None
try:
worker_file = open('workers.txt', 'r')
workers = worker_file.readlines()
worker_file.close()
except FileNotFoundError:
return jsonify({'error': 'No workers available', 'success': 'false'})
# Get a worker that has available slots
worker = None
for line in workers:
ip = line.split(':')[1].strip('\n')
resp=requests.get("http://"+ip + ":5000/status",timeout=2)
if (resp.status_code == 200):
if resp.json()['availability'] == True:
worker = line
break
if worker == None:
return jsonify({'error': 'No workers available', 'success': 'false'})
# Add domain to file # Add domain to file
sites_file = open('sites.txt', 'a') sites_file = open('sites.txt', 'a')
sites_file.write(domain + '\n') sites_file.write(domain + ':' + worker.split(':')[0] + '\n')
sites_file.close() sites_file.close()
# Use key # Use key
@ -60,6 +88,9 @@ def new_site():
key_file.write(line) key_file.write(line)
key_file.close() key_file.close()
# Send worker request
requests.post("http://"+ worker.split(':')[1].strip('\n') + ":5000/new-site?domain=" + domain)
return jsonify({'success': 'true', 'domain': domain, 'status': "creating"}) return jsonify({'success': 'true', 'domain': domain, 'status': "creating"})
@ -67,10 +98,11 @@ def new_site():
@app.route('/add-worker', methods=['POST']) @app.route('/add-worker', methods=['POST'])
def add_worker(): def add_worker():
worker=request.args.get('worker') worker=request.args.get('worker')
worker_IP=request.args.get('ip')
# Get API header # Get API header
api_key = request.headers.get('key') api_key = request.headers.get('key')
if api_key == None or worker == None: if api_key == None or worker == None or worker_IP == None:
return jsonify({'error': 'Invalid API key or worker', 'success': 'false'}) return jsonify({'error': 'Invalid API key or worker info', 'success': 'false'})
if api_key != os.getenv('WORKER_KEY'): if api_key != os.getenv('WORKER_KEY'):
return jsonify({'error': 'Invalid API key', 'success': 'false'}) return jsonify({'error': 'Invalid API key', 'success': 'false'})
@ -90,9 +122,15 @@ def add_worker():
# Add worker to file # Add worker to file
workers_file = open('workers.txt', 'a') workers_file = open('workers.txt', 'a')
workers_file.write(worker + '\n') workers_file.write(worker + ":" + worker_IP + '\n')
workers_file.close() workers_file.close()
return jsonify({'success': 'true', 'worker': worker, 'status': "ready to join"})
online=True
resp=requests.get("http://"+worker_IP + ":5000/ping",timeout=2)
if (resp.status_code != 200):
online=False
return jsonify({'success': 'true', 'worker': worker, 'online': online})
def get_sites_count(): def get_sites_count():
@ -118,13 +156,33 @@ def site_exists(domain):
sites_file = open('sites.txt', 'r') sites_file = open('sites.txt', 'r')
contains_site = False contains_site = False
# Check if domain is in file for line in sites_file.readlines():
if domain in sites_file.read(): if domain == line.split(':')[0]:
contains_site = True contains_site = True
break
sites_file.close() sites_file.close()
return contains_site return contains_site
def site_worker(domain):
# If file doesn't exist, create it
try:
sites_file = open('sites.txt', 'r')
except FileNotFoundError:
sites_file = open('sites.txt', 'w')
sites_file.close()
sites_file = open('sites.txt', 'r')
worker = None
for line in sites_file.readlines():
if domain == line.split(':')[0]:
worker = line.split(':')[1].strip('\n')
break
sites_file.close()
return worker
# Start the server # Start the server
if __name__ == '__main__': if __name__ == '__main__':
app.run(debug=False, port=5000) app.run(debug=False, port=4000)

1
worker/.env.example Normal file
View File

@ -0,0 +1 @@
MAX_SITES=4

View File

@ -26,6 +26,20 @@ def new_site():
# Return the domain and the number of sites # Return the domain and the number of sites
return jsonify({'domain': domain, 'count': count}) return jsonify({'domain': domain, 'count': count})
# Return status
@app.route('/status', methods=['GET'])
def status():
num_Sites = get_sites_count()
availability=(num_Sites < int(os.getenv('MAX_SITES')))
return jsonify({'availability': availability, 'num_sites': num_Sites})
# Ping status
@app.route('/ping')
def ping():
return 'pong'
def get_sites_count(): def get_sites_count():
# If file doesn't exist, create it # If file doesn't exist, create it
try: try:
@ -34,7 +48,7 @@ def get_sites_count():
sites_file = open('sites.txt', 'w') sites_file = open('sites.txt', 'w')
sites_file.close() sites_file.close()
sites_file = open('sites.txt', 'r') sites_file = open('sites.txt', 'r')
print(sites_file.readlines())
# Return number of lines in file # Return number of lines in file
return len(sites_file.readlines()) return len(sites_file.readlines())