public async Task RunAsync()
{
while (true)
{
var line = await Input.ReadLineAsync();
if (line is not null)
{
NotebookParserServerResponse response = null;
string requestId = null;
try
{
var request = NotebookParseOrSerializeRequest.FromJson(line);
requestId = request.Id;
response = HandleRequest(request);
}
catch (Exception ex)
{
if (requestId is not null)
{
// if no ID could be parsed, there's no point in returning an error since it can't be associated with the request
response = new NotebookErrorResponse(requestId, ex.ToString());
}
}
if (response is not null)
{
var responsePayload = response.ToJson();
await Output.WriteLineAsync(responsePayload);
await Output.FlushAsync();
}
}
}
}
public static NotebookParserServerResponse HandleRequest(NotebookParseOrSerializeRequest request)
{
try
{
switch (request)
{
case NotebookParseRequest parse:
{
using var contentStream = new MemoryStream(parse.RawData);
var document = request.SerializationType switch
{
DocumentSerializationType.Dib => CodeSubmission.Read(contentStream, request.DefaultLanguage, WellKnownKernelNames),
DocumentSerializationType.Ipynb => Notebook.Read(contentStream, WellKnownKernelNames),
_ => throw new NotSupportedException($"Unable to parse an interactive document with type '{request.SerializationType }'"),
};
return(new NotebookParseResponse(request.Id, document));
}
case NotebookSerializeRequest serialize:
{
using var resultStream = new MemoryStream();
switch (request.SerializationType)
{
case DocumentSerializationType.Dib:
CodeSubmission.Write(serialize.Document, serialize.NewLine, resultStream);
break;
case DocumentSerializationType.Ipynb:
Notebook.Write(serialize.Document, serialize.NewLine, resultStream);
break;
default:
throw new NotSupportedException($"Unable to serialize a interactive document of type '{request.SerializationType}'");
}
resultStream.Position = 0;
var resultArray = resultStream.ToArray();
return(new NotebookSerializeResponse(request.Id, resultArray));
}
default:
return(new NotebookErrorResponse(request.Id, $"Unsupported request: {request}"));
}
}
catch (Exception ex)
{
return(new NotebookErrorResponse(request.Id, ex.ToString()));
}
}