#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import json
import os
from random import randint
from subprocess import PIPE, Popen
from threading import Thread
from time import sleep, time
from flask import Flask, jsonify, request
from requests import get, post
import rethinkdb as r
app = Flask(__name__)
ADDRESS = {'ftp': None, 'rdb': None, 'http': None}
def run(command):
while True:
process = Popen(command, stdout=PIPE, shell=True)
while True:
try:
line = next(process.stdout).strip()
yield line
except:
break
sleep(0.1)
def update_address(server, port):
name = os.getenv("MUNYAL_NAME")
while True:
for line in run("ssh -R 0:localhost:{} serveo.net".format(port)):
line = line.decode("utf-8")
try:
if "from " in line:
address = str(line).split("from ")[1].split(
"\n")[0].strip()
elif "Allocated port" in line:
address = "serveo.net:" + str(line).split(
"Allocated port ")[1].split(" ")[0].strip()
address = address.strip("'")
ADDRESS[server] = address
response = get("http://localhost:5000/directory",
params={"name": name})
response = json.loads(response.text)
print(
response.get("status") + "\t" + response.get("message") +
"\t" + str(ADDRESS))
if response.get("status") == "ok":
if response.get(server + "_address") != address:
if ADDRESS['ftp'] and ADDRESS['rdb'] and ADDRESS[
'http']:
with open("$HOME/.munyal/config/token", "r") as f:
token = f.read()
post("http://localhost:5000/directory",
data={
'name': name,
'http_address': ADDRESS['http'],
'ftp_address': ADDRESS['ftp'],
'rdb_address': ADDRESS['rdb'],
'token': token
})
elif "servidor se ha registrado" in response.get("message"):
if ADDRESS['ftp'] and ADDRESS['rdb'] and ADDRESS['http']:
r = post("http://localhost:5000/directory",
data={
'name': name,
'http_address': ADDRESS['http'],
'ftp_address': ADDRESS['ftp'],
'rdb_address': ADDRESS['rdb']
})
r = json.loads(r.text)
with open("$HOME/.munyal/config/token", "w") as f:
f.write(r.get('token'))
except:
sleep(1)
sleep(1)
@app.route("/", methods=["GET"])
def index():
return '''
<html>
<head>
<title>Munyal API</title>
</head>
<body>
<h1>Munyal private API</h1>
</body>
</html>
'''
@app.route("/upload", methods=["POST"])
def upload():
try:
r.connect("localhost", 28015).repl()
cursor = r.table("changes")
host = request.form.get("host")
action = request.form.get("action")
route = request.form.get("route")
obj = {
'id': str(time()).split('.')[0] + str(randint(1, 1000000)),
'action': action,
'route': route,
'host': host
}
status = 'ok'
try:
cursor.insert(obj).run()
except:
status = 'error'
except:
status = 'error'
obj['status'] = status
return jsonify(obj)
if __name__ == '__main__':
http_thread = Thread(target=update_address, args=("http", "5010"))
http_thread.setDaemon(True)
http_thread.start()
ftp_thread = Thread(target=update_address, args=("ftp", "80"))
ftp_thread.setDaemon(True)
ftp_thread.start()
rdb_thread = Thread(target=update_address, args=("rdb", "27017"))
rdb_thread.setDaemon(True)
rdb_thread.start()
app.run(debug=True, port=5010)