sourcekit-lsp/Sources/SKTestSupport/INPUTS/BuildServerBuildSystemTests.testBuildTargets/server.py

#!/usr/bin/env python3

import json
import sys


while True:
    line = sys.stdin.readline()
    if len(line) == 0:
        break

    assert line.startswith('Content-Length:')
    length = int(line[len('Content-Length:'):])
    sys.stdin.readline()
    message = json.loads(sys.stdin.read(length))

    response = None
    if message["method"] == "build/initialize":
        response = {
            "jsonrpc": "2.0",
            "id": message["id"],
            "result": {
                "displayName": "test server",
                "version": "0.1",
                "bspVersion": "2.0",
                "rootUri": "blah",
                "capabilities": {"languageIds": ["objective-c", "swift"]},
                "data": {
                    "indexStorePath": "some/index/store/path"
                }
            }
        }
    elif message["method"] == "build/initialized":
        continue
    elif message["method"] == "build/shutdown":
        response = {
            "jsonrpc": "2.0",
            "id": message["id"],
            "result": None
        }
    elif message["method"] == "build/exit":
        break
    elif message["method"] == "workspace/buildTargets":
        response = {
            "jsonrpc": "2.0",
            "id": message["id"],
            "result": {
                "targets": [
                    {
                        "id": {"uri": "target:first_target"},
                        "displayName": "First Target",
                        "baseDirectory": "file:///some/dir",
                        "tags": ["library", "test"],
                        "capabilities": {
                            "canCompile": True,
                            "canTest": True,
                            "canRun": False
                        },
                        "languageIds": ["objective-c", "swift"],
                        "dependencies": []
                    },
                    {
                        "id": {"uri": "target:second_target"},
                        "displayName": "Second Target",
                        "baseDirectory": "file:///some/dir",
                        "tags": ["library", "test"],
                        "capabilities": {
                            "canCompile": True,
                            "canTest": False,
                            "canRun": False
                        },
                        "languageIds": ["objective-c", "swift"],
                        "dependencies": [{"uri": "target:first_target"}]
                    }
                ]
            }
        }
    # ignore other notifications
    elif "id" in message:
        response = {
            "jsonrpc": "2.0",
            "id": message["id"],
            "error": {
                "code": 123,
                "message": "unhandled method {}".format(message["method"]),
            }
        }

    if response:
        responseStr = json.dumps(response)
        try:
            sys.stdout.write("Content-Length: {}\r\n\r\n{}".format(len(responseStr), responseStr))
            sys.stdout.flush()
        except IOError:
            # stdout closed, time to quit
            break