diff --git a/webservice-balancer/clean.sh b/webservice-balancer/clean.sh new file mode 100755 index 0000000..6554ede --- /dev/null +++ b/webservice-balancer/clean.sh @@ -0,0 +1,9 @@ +#!/bin/bash +chmod +x deploy_webservice.py +echo "### Creating venv is probably a good idea - let's make one..." +python3 -m venv web_venv +source ./web_venv/bin/activate +echo "### Now install hcloud module..." +pip install -q hcloud +echo "### And clear last deployment results..." +python deploy_webservice.py --clean diff --git a/webservice-balancer/deploy.sh b/webservice-balancer/deploy.sh new file mode 100755 index 0000000..c85930c --- /dev/null +++ b/webservice-balancer/deploy.sh @@ -0,0 +1,13 @@ +#!/bin/bash +chmod +x deploy_webservice.py +python3 -m venv web_venv +source ./web_venv/bin/activate +pip install -q hcloud +echo "############################" +echo "Let's clean everything from last run..." +python deploy_webservice.py --clean +echo "--------------------------------------------------------------------------" +echo " Success! I guess..." +echo "############################" +echo "Let's spawn some new machines!" +python deploy_webservice.py $1 --create \ No newline at end of file diff --git a/webservice-balancer/deploy_webservice.py b/webservice-balancer/deploy_webservice.py new file mode 100755 index 0000000..609704b --- /dev/null +++ b/webservice-balancer/deploy_webservice.py @@ -0,0 +1,196 @@ +#!/bin/python3 +from hcloud import Client +from hcloud.images.domain import Image +from hcloud.networks.domain import NetworkSubnet +from hcloud.server_types.domain import ServerType +from hcloud.locations.domain import Location +from hcloud.load_balancer_types.domain import LoadBalancerType +from hcloud.load_balancers.domain import ( + LoadBalancerHealthCheck, + LoadBalancerService, + LoadBalancerHealtCheckHttp, +) +import sys + + +CLIENT_ID = "s478841" +SERVERS_NO = 2 +key_name = f"ssh-{CLIENT_ID}" +net_name = f"webservice-net-{CLIENT_ID}" +server_name = f"webservice-server-{CLIENT_ID}" +load_balancer_name = f"webservice-load-balancer-{CLIENT_ID}" +location = Location("hel1") + +ci_webservice = r"""#cloud-config +runcmd: + - cd /home + - git clone https://git.wmi.amu.edu.pl/s478841/DPZC-Hetzner.git + - cd DPZC-Hetzner/webservice-balancer + - chmod +x ./webservice + - ./webservice +""" + + +def create_client(creds_path): + with open(creds_path, "r") as f: + token = str(f.readline()).strip() + client = Client(token=token) + print("The CLIENT has been CREATED") + return client + + +def load_key(pub_key_path): + print(f"\t\tUsing the {pub_key_path} file") + with open(pub_key_path) as f: + key = f.readline() + print("\t\tThe SSH KEY has been LOADED") + return key + + +def create_key(client, key_name, pub_key_path): + print("The SSH KEY has been...") + key_of_power = load_key(pub_key_path) + key = client.ssh_keys.create(name=key_name, public_key=key_of_power) + print("\tCREATED") + return key + + +def remove_key(client, key_name): + print("The KEY has been...", end="") + try: + action = client.ssh_keys.delete(client.ssh_keys.get_by_name(key_name)) + action.wait_until_finished() + except AttributeError: + pass + print("DELETED") + + +def create_network(client, net_name, ip_range): + print("The NETWORK has been...", end="") + net = client.networks.create( + name=net_name, + ip_range=ip_range, + subnets=[ + NetworkSubnet(ip_range=ip_range, network_zone="eu-central", type="cloud") + ], + ) + print("CREATED") + return net + + +def remove_network(client, net_name): + print("The NETWORK has been...", end="") + try: + action = client.networks.delete(client.networks.get_by_name(net_name)) + action.wait_until_finished() + except AttributeError: + pass + + print("DELETED") + + +def create_server(client, server_name, ssh_key, vnet, location, user_data): + print("The WEBSERVICE has been CREATED:", end=" ") + webserver = client.servers.create( + name=server_name, + server_type=ServerType("cx11"), + image=Image(name="ubuntu-22.04"), + ssh_keys=[ssh_key], + networks=[vnet], + location=location, + user_data=user_data, + ) + webserver.action.wait_until_finished() + print(webserver.action.complete) + return webserver + + +def remove_server(client, server_name): + print("The WEBSERVICE has been...", end="") + try: + action = client.servers.delete(client.servers.get_by_name(server_name)) + action.wait_until_finished() + except AttributeError: + pass + print("DELETED") + + +def create_load_balancer(client, name, location, vnet, servers): + print("The LOAD BALANCER has been CREATED:", end=" ") + lb = client.load_balancers.create( + name=name, + load_balancer_type=LoadBalancerType("lb11"), + location=location, + services=[ + LoadBalancerService( + protocol="http", + listen_port=8080, + destination_port=8080, + proxyprotocol=False, + health_check=LoadBalancerHealthCheck( + protocol="http", + port=8080, + interval=20, + timeout=10, + retries=3, + http=LoadBalancerHealtCheckHttp( + path="/factors/6", status_codes=["2xx", "3xx"], tls=False + ), + ), + ) + ], + targets=servers, + public_interface=True, + network=vnet, + ) + lb.action.wait_until_finished() + print(lb.action.complete) + return lb + + +def remove_load_balancer(client, name): + print("The LOAD BALANCER has been...", end="") + try: + action = client.load_balancers.delete(client.load_balancers.get_by_name(name)) + action.wait_until_finished() + except AttributeError: + pass + print("DELETED") + + +if __name__ == "__main__": + + client = create_client("../tokens/token_file") + + ssh_path = sys.argv[-2] + action = sys.argv[-1] + + if action == "--create": + ssh_key = create_key(client, key_name, ssh_path) + vnet = create_network(client, net_name, "10.10.10.0/24") + + servers = [ + create_server( + client, server_name + "_" + str(i+1), ssh_key, vnet, location, ci_webservice + ) + for i in range(SERVERS_NO) + ] + + # print( + # f"You can access the server (in a few minutes) under the address: http://{server.server.data_model.public_net.ipv4.ip}:3000" + # ) + load_balancer = create_load_balancer( + client, load_balancer_name, location, vnet, servers + ) + print( + f"Your service should be available at {load_balancer.load_balancer.data_model.public_net.ipv4.ip}\n\n\t\tHave fun!" + ) + elif action == "--clean": + remove_load_balancer(client, load_balancer_name) + for i in range(SERVERS_NO): + remove_server(client, server_name + "_" + str(i+1)) + remove_network(client, net_name) + remove_key(client, key_name) + print("\n\n\t\tIt was nice to meet you!") + else: + print(f"\nSorry, don't know what the'{action}' means...bye!") diff --git a/webservice-balancer/webservice b/webservice-balancer/webservice new file mode 100755 index 0000000..afd2204 Binary files /dev/null and b/webservice-balancer/webservice differ