internal PipelineStage CreateNextStage(ArgHook.HookContext context, string[] commandLine, ICommandLineArgumentsDefinitionFactory factory)
{
PipelineStage next;
CommandLineAction inProcAction;
if (Stages.Count == 0)
{
next = new RootPipelineStage(commandLine);
}
else if (commandLine[0].StartsWith(ArgPipeline.PipelineStageActionIndicator) && ArgPipelineActionStage.TryCreateActionStage(commandLine, out next))
{
// do nothing, next is populated
}
else if (TryParseStageAction(context.Definition, commandLine[0], out inProcAction))
{
next = new InProcessPipelineStage(context.Definition, commandLine);
}
else if (ExternalPipelineProvider.TryLoadOutputStage(commandLine, out next) == false)
{
throw new UnexpectedArgException("The pipeline action '" + string.Join(" ", commandLine) + "' is not valid. If you want to support piping between processes, learn how to here (TODO URL)");
}
next.CommandLineDefinitionFactory = factory;
this.AddStage(next);
return(next);
}
internal PipelineStage CreateNextStage(ArgHook.HookContext context, string[] commandLine, ICommandLineArgumentsDefinitionFactory factory)
{
PipelineStage next;
CommandLineAction inProcAction;
if (Stages.Count == 0)
{
next = new RootPipelineStage(commandLine);
}
else if (commandLine[0].StartsWith(ArgPipeline.PipelineStageActionIndicator) && ArgPipelineActionStage.TryCreateActionStage(commandLine, out next))
{
// do nothing, next is populated
}
else if (TryParseStageAction(context.Definition, commandLine[0], out inProcAction))
{
next = new InProcessPipelineStage(context.Definition, commandLine);
}
else if (ExternalPipelineProvider.TryLoadOutputStage(commandLine, out next) == false)
{
throw new UnexpectedArgException("The pipeline action '"+string.Join(" ", commandLine)+"' is not valid. If you want to support piping between processes, learn how to here (TODO URL)");
}
next.CommandLineDefinitionFactory = factory;
this.AddStage(next);
return next;
}
public HttpInputPipelineStage(CommandLineArgumentsDefinition baseDefinition, string[] rawCommandLine)
: base(baseDefinition, CleanCommandLineOfInputPortInfo(rawCommandLine))
{
int port;
if (TryFindPort(rawCommandLine, out port) == false)
{
IsProgramLaunchedByExternalPipeline = false;
return;
}
else
{
IsProgramLaunchedByExternalPipeline = true;
}
Init.InitIfNotAlreadyDone();
interceptor = ConsoleOutInterceptor.Instance;
interceptor.Attach();
PowerLogger.LogLine("Initializing input pipe for command line on port "+port+": "+string.Join(" ", this.CmdLineArgs));
wrappedStage = new InProcessPipelineStage(baseDefinition, this.CmdLineArgs.ToArray());
listener = new HttpPipelineMessageListener(port, TimeSpan.FromSeconds(10));
listener.Timeout += () =>
{
PowerLogger.LogLine("HttpInputPipelineStage listener timed out.");
Process.GetCurrentProcess().Kill();
};
wrappedStage.UnhandledException += (ex) =>
{
QueueException(ex);
};
listener.ListenException += (ex) =>
{
QueueException(ex);
};
listener.MessageReceivedHandler = (message) =>
{
if (message.ControlAction == null)
{
LogLine("Pipe input received");
var pipedObject = DeserializePipedObject(message);
wrappedStage.CommandLineDefinitionFactory = this.CommandLineDefinitionFactory;
wrappedStage.Accept(pipedObject);
LogLine("Pipe input processed");
return new HttpPipelineControlResponse() { StatusCode = HttpStatusCode.Accepted };
}
else if (message.ControlAction == "Drain")
{
Drain();
return new HttpPipelineControlResponse() { StatusCode = HttpStatusCode.Accepted };
}
else if (message.ControlAction == "Poll")
{
LogLine("Poll requested");
string pipedObjectArrayJson = null;
string exceptionInfo = null;
List<ConsoleCharacter> textOutput = null;
lock (outputQueueLock)
{
List<object> batch = new List<object>();
while (outputQueue.Count > 0)
{
batch.Add(outputQueue.Dequeue());
}
if (batch.Count > 0)
{
LogLine("Pipe output sent");
pipedObjectArrayJson = JsonConvert.SerializeObject(batch, HttpPipelineMessage.CommonSettings);
}
while(exceptionQueue.Count > 0)
{
LogLine("Exception output sent");
exceptionInfo = exceptionInfo ?? "";
exceptionInfo += exceptionQueue.Dequeue().ToString() + Environment.NewLine + Environment.NewLine;
}
var interceptedText = interceptor.ReadAndClear();
while(interceptedText.Count > 0)
{
textOutput = textOutput ?? new List<ConsoleCharacter>();
textOutput.Add(interceptedText.Dequeue());
}
while(textOutputQueue.Count > 0)
{
textOutput = textOutput ?? new List<ConsoleCharacter>();
textOutput.Add(textOutputQueue.Dequeue());
}
}
return new HttpPipelineControlResponse() { StatusCode = HttpStatusCode.OK, ExceptionInfo = exceptionInfo, ConsoleOutput = textOutput, PipedObjectArrayJson = pipedObjectArrayJson, Value = wrappedStage.IsDrained + "" };
}
else if (message.ControlAction == "Close")
{
LogLine("Close requested");
ArgPipeline.ObjectExitedPipeline -= outputHandler;
return new HttpPipelineControlResponse() { StatusCode = HttpStatusCode.OK, Close = true };
}
else
{
LogLine("Unrecognized action: " + message.ControlAction);
return new HttpPipelineControlResponse() { StatusCode = HttpStatusCode.BadRequest, ConsoleOutput = new ConsoleString("Unrecognized action: "+message.ControlAction).ToList() };
}
};
outputHandler = (obj) =>
{
LogLine("Pipe output queued");
lock(outputQueueLock)
{
outputQueue.Enqueue(obj);
}
};
ArgPipeline.ObjectExitedPipeline += outputHandler;
listener.Start();
}
public HttpInputPipelineStage(CommandLineArgumentsDefinition baseDefinition, string[] rawCommandLine)
: base(baseDefinition, CleanCommandLineOfInputPortInfo(rawCommandLine))
{
int port;
if (TryFindPort(rawCommandLine, out port) == false)
{
IsProgramLaunchedByExternalPipeline = false;
return;
}
else
{
IsProgramLaunchedByExternalPipeline = true;
}
Init.InitIfNotAlreadyDone();
interceptor = ConsoleOutInterceptor.Instance;
interceptor.Attach();
PowerLogger.LogLine("Initializing input pipe for command line on port " + port + ": " + string.Join(" ", this.CmdLineArgs));
wrappedStage = new InProcessPipelineStage(baseDefinition, this.CmdLineArgs.ToArray());
listener = new HttpPipelineMessageListener(port, TimeSpan.FromSeconds(10));
listener.Timeout += () =>
{
PowerLogger.LogLine("HttpInputPipelineStage listener timed out.");
Process.GetCurrentProcess().Kill();
};
wrappedStage.UnhandledException += (ex) =>
{
QueueException(ex);
};
listener.ListenException += (ex) =>
{
QueueException(ex);
};
listener.MessageReceivedHandler = (message) =>
{
if (message.ControlAction == null)
{
LogLine("Pipe input received");
var pipedObject = DeserializePipedObject(message);
wrappedStage.CommandLineDefinitionFactory = this.CommandLineDefinitionFactory;
wrappedStage.Accept(pipedObject);
LogLine("Pipe input processed");
return(new HttpPipelineControlResponse()
{
StatusCode = HttpStatusCode.Accepted
});
}
else if (message.ControlAction == "Drain")
{
Drain();
return(new HttpPipelineControlResponse()
{
StatusCode = HttpStatusCode.Accepted
});
}
else if (message.ControlAction == "Poll")
{
LogLine("Poll requested");
string pipedObjectArrayJson = null;
string exceptionInfo = null;
List <ConsoleCharacter> textOutput = null;
lock (outputQueueLock)
{
List <object> batch = new List <object>();
while (outputQueue.Count > 0)
{
batch.Add(outputQueue.Dequeue());
}
if (batch.Count > 0)
{
LogLine("Pipe output sent");
pipedObjectArrayJson = JsonConvert.SerializeObject(batch, HttpPipelineMessage.CommonSettings);
}
while (exceptionQueue.Count > 0)
{
LogLine("Exception output sent");
exceptionInfo = exceptionInfo ?? "";
exceptionInfo += exceptionQueue.Dequeue().ToString() + Environment.NewLine + Environment.NewLine;
}
var interceptedText = interceptor.ReadAndClear();
while (interceptedText.Count > 0)
{
textOutput = textOutput ?? new List <ConsoleCharacter>();
textOutput.Add(interceptedText.Dequeue());
}
while (textOutputQueue.Count > 0)
{
textOutput = textOutput ?? new List <ConsoleCharacter>();
textOutput.Add(textOutputQueue.Dequeue());
}
}
return(new HttpPipelineControlResponse()
{
StatusCode = HttpStatusCode.OK, ExceptionInfo = exceptionInfo, ConsoleOutput = textOutput, PipedObjectArrayJson = pipedObjectArrayJson, Value = wrappedStage.IsDrained + ""
});
}
else if (message.ControlAction == "Close")
{
LogLine("Close requested");
ArgPipeline.ObjectExitedPipeline -= outputHandler;
return(new HttpPipelineControlResponse()
{
StatusCode = HttpStatusCode.OK, Close = true
});
}
else
{
LogLine("Unrecognized action: " + message.ControlAction);
return(new HttpPipelineControlResponse()
{
StatusCode = HttpStatusCode.BadRequest, ConsoleOutput = new ConsoleString("Unrecognized action: " + message.ControlAction).ToList()
});
}
};
outputHandler = (obj) =>
{
LogLine("Pipe output queued");
lock (outputQueueLock)
{
outputQueue.Enqueue(obj);
}
};
ArgPipeline.ObjectExitedPipeline += outputHandler;
listener.Start();
}
