#!/usr/bin/python3

import sys
import json
import requests

headers = { 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36',
 'Cache-Control': 'no-cache',
 'Accpet': '*/*',
 'Content-Type': 'application/json'}

_node = {'ip': '127.0.0.1', 'port': 8500}

def gen_ServiceID(service_name, service_ip, service_port):
 return service_name + '-' + service_ip + ':' + str(service_port)

def register_service(service_name, service_ip, service_port , node=_node):
 service = {}
 service['Name'] = service_name
 service['Address'] = service_ip
 service['Port'] = service_port
 service['ID'] = gen_ServiceID(service_name, service_ip, service_port)
 check = {}
 check['TCP'] = service['Address'] + ':' + str(service['Port'])
 check['Interval'] = '10s'
 check['DeregisterCriticalServiceAfter'] = '30s'
 service['Check'] = check
 data = json.dumps(service)
 url = 'http://' + node['ip'] + ':' + str(node['port'])
 url += '/v1/agent/service/register'
 try:
 res = requests.put(url=url, headers=headers, data=data)
 return res.status_code == 200 and len(res.content) == 0
 except Exception as e:
 print(e)
 return False

def deregister_service(ServiceID, node=_node):
 url = 'http://' + node['ip'] + ':' + str(node['port'])
 url += '/v1/agent/service/deregister/'
 url += ServiceID
 try:
 res = requests.put(url=url, headers=headers)
 return res.status_code == 200 and len(res.content) == 0
 except Exception as e:
 print(e)
 return False

def discover_service(service_name, node=_node):
 url = 'http://' + node['ip'] + ':' + str(node['port'])
 url += '/v1/catalog/service/'
 url += service_name
 try:
 res = requests.get(url=url, headers=headers)
 return res.content
 except Exception as e:
 print(e)
 return 'false'

def main():
 print(sys.argv)
 if len(sys.argv) < 4:
 print('Usage: cmd service_name service_ip service_port node_ip node_port')
 return
 service_name = sys.argv[1]
 service_ip = sys.argv[2]
 service_port = int(sys.argv[3])
 if len(sys.argv) >= 5:
 _node['ip'] = sys.argv[4]
 _node['port'] = int(sys.argv[5])
 ServiceID = gen_ServiceID(service_name, service_ip, service_port)
 if register_service(service_name, service_ip, service_port):
 print('register_service [%s] succeed!' % (ServiceID))
 else:
 print('register_service failed!')

 print(discover_service(service_name))

 '''
 if deregister_service(ServiceID):
 print('deregister_service [%s] succeed!' % (ServiceID))
 else:
 print('deregister_service failed')

 print(discover_service(service_name))
 '''

if __name__ == '__main__':
 main()