webrtc_server.py

#!/usr/bin/python3

import sys,gi,json,argparse,datetime,threading,time
gi.require_version('GLib', '2.0')
gi.require_version('Gst',  '1.0')
gi.require_version('Soup', '2.4')
gi.require_version('GstWebRTC', '1.0')
gi.require_version('GstSdp', '1.0')
from gi.repository import GLib, Gst, Soup, GstWebRTC, GstSdp

from gst_helpers import *
from webrtc_peer import *
from client import *

args = None
mutex = threading.Lock()

# get address and port from client
def get_client_address(client):
    addr = client.get_remote_address()
    return addr.get_address().to_string()+"_"+str(addr.get_port())

# incoming HTTP(S) request
def http_handler(server,msg,path,query,client,user_data):
    logging.info("HTTP(S) request for: "+path)
    if path == "/":
        path = "/index.html"
    #flags[get_client_address(client)] = query
    content_type = "text/html"
    try:
        data = open("webclient"+path,"rb").read()
        if path.endswith(".js"):
            content_type = "text/javascript"
        if path.endswith(".jpg"):
            content_type = "image/jpeg"
        if path.endswith(".css"):
            content_type = "text/css"
        msg.set_status(Soup.Status.OK)
    except:
        msg.set_status(Soup.Status.NOT_FOUND)
        data=(path+" not found").encode("utf-8")
        if path == "/quit":
            logging.info("Well... bye.")
            GLib.timeout_add(100,quit_mainloop)
            data=b"Server exiting/restarting..."

    msg.response_headers.append("Content-Type",content_type)
    msg.response_headers.append("Cache-Control","no-store")
    #msg.response_headers.append("Access-Control-Allow-Origin","*")
    msg.response_body.append(data)

# Websocket connection was closed by remote
def ws_close_handler(connection, client):
    logging.info("WebSocket closed by remote.")
    client.remove()

# incoming Websocket connection
def ws_conn_handler(server, connection, path, client, user_data):

    time.sleep(1)

    source = get_client_address(client)
    logging.info("New WebSocket connection from: "+source)

    #mutex.acquire()

    wrb = WebRTCDecoder(connection,source,args.stun)
    new_client = Client(source,wrb,args.size[0],args.size[1],mutex)
    connection.connect("closed",ws_close_handler,new_client)

# "main"
print("\nSurfaceCast backend mixing server v0.2.2 - https://github.com/floe/surfacecast\n")
print("Note: any GStreamer-WARNINGs about pipeline loops can be safely ignored.\n")

parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
outfile = datetime.datetime.now().strftime("surfacecast-%Y%m%d-%H%M%S.mkv")

parser.add_argument("-d","--debug", help="more debug output (-dd=max)",  action="count",default=1 )
parser.add_argument("-s","--sink",  help="save all streams to MKV file", action="store_true"      )
parser.add_argument("-p","--port",  help="server HTTPS listening port",  default=8080             )
parser.add_argument("-o","--out",   help="MP4 output filename", default=outfile                   )
parser.add_argument("-u","--stun",  help="STUN server", default="stun://stun.l.google.com:19302"  )
parser.add_argument(     "--size",  help="surface stream output size", default="1280x720"         )
parser.add_argument(     "--fps",   help="frames per second", default=15                          )

args = parser.parse_args()
args.size = [ int(n) for n in args.size.split("x") ]
print("Option",args,"\n")

init_pipeline(on_element_added,args.debug)

frontsrc   = "filesrc location=assets/front.png ! pngdec ! videoconvert ! imagefreeze is-live=true ! queue"
surfacesrc = "videotestsrc is-live=true pattern=solid-color foreground-color=0" #ball motion=sweep background-color=0
audiosrc   = "audiotestsrc is-live=true wave=silence"

add_test_sources(frontsrc,surfacesrc,audiosrc,fake=True,bgcol=0xFFFF00FF,wave="sine",sw=args.size[0],sh=args.size[1],fps=args.fps)
create_frontmixer_queue(args.fps)

server = Soup.Server()
server.add_handler("/",http_handler,None)
server.add_websocket_handler("/ws",None,None,ws_conn_handler,None)
server.set_ssl_cert_file("assets/tls-cert.pem","assets/tls-key.pem")
server.listen_all(int(args.port),Soup.ServerListenOptions.HTTPS)

if args.sink:
    logging.info("Adding file sink client...")
    sink = StreamSink("file_sink",args.out)
    client = Client("file_sink",sink)
    client.link_new_client()

run_mainloop()