#!/usr/bin/env python
import socket
import os
import sys

def listen_for_token():
    """
    Listens on port 13747 on localhost for a redirect request by OIDC
    server, parses the response and returns the "access_token" value.
    """
    TCP_IP = '127.0.0.1'
    TCP_PORT = 13747
    BUFFER_SIZE = 1024

    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    s.bind((TCP_IP, TCP_PORT))
    s.listen(1)

    conn, addr = s.accept()
    print 'Connection address:', addr
    data = ""
    sent_resp = False
    while 1:
        try:
            r = conn.recv(BUFFER_SIZE)
        except:
            conn.close()
            break
        if not r: break
        data += r

        if not sent_resp:
            response = "Token has been handled."
            conn.send("""HTTP/1.1 200 OK
Content-Length: %s
Content-Type: text/plain
Connection: Closed

%s""" % (len(response), response))
            conn.close()
            sent_resp = True

    s.close()

    data = data.split("\n")
    for line in data:
        variables = line.split("&")
        for var in variables:
            kv = var.split("=")
            if not len(kv) == 2:
                continue
            if kv[0] == "access_token":
                return kv[1]
    return None

mbs_host = "localhost:5000"
token = None
if len(sys.argv) > 2:
    token = sys.argv[2]
if len(sys.argv) > 1:
    mbs_host = sys.argv[1]

print "Usage: submit_build.py [mbs_host] [oidc_token]"
print ""
if not token:
    print "Provide token as command line argument or visit following URL to obtain the token:"
    print "https://id.stg.fedoraproject.org/openidc/Authorization?response_type=token&response_mode=form_post&nonce=1234&scope=openid%20profile%20email&client_id=mbs-authorizer&state=af0ifjsldkj&redirect_uri=http://localhost:13747/"
    print "We are waiting for you to finish the token generation..."

if not token:
    token = listen_for_token()
if not token:
    print "Failed to get a token from response"
    os._exit(1)

print "Submitting build of ..."
with open("submit-build.json", "r") as build:
    print build.read()
print "Using https://%s/module_build_service/module-builds/" % mbs_host
print "NOTE: You need to be a Fedora packager for this to work"
print

os.system("curl -b 'oidc_token=%s' -k -H 'Content-Type: text/json' --data @submit-build.json https://%s/module-build-service/1/module-builds/ -v" % (token, mbs_host))