import json
import sys

from vectororder.connectors import elasticsearch


def main():
    args_number = len(sys.argv)
    if args_number < 3 or args_number > 4:
        sys.stderr.write('Usage: {} METHOD URL [JSON-BODY-OR-PATH]\n'.format(
            sys.argv[0]))
        sys.exit(1)
    method = sys.argv[1]
    url = sys.argv[2]
    json_body = None
    if args_number > 3:
        json_body = sys.argv[3]
        if json_body.startswith('.') or json_body.startswith('/'):
            # Treat json_body parameter as path.
            with open(json_body) as f:
                json_body = f.read()
    res = elasticsearch.client.transport.perform_request(
        method=method, url=url, body=json_body)
    sys.stdout.write(json.dumps(res) + '\n')


if __name__ == '__main__':
    main()