PipelineResults.cs

using System.Collections.Generic;
 
namespace MPewsey.Common.Pipelines
{
    public class PipelineResults
    {
        public Dictionary<string, object> Inputs { get; } = new Dictionary<string, object>();
 
        public Dictionary<string, object> Outputs { get; } = new Dictionary<string, object>();
 
        public bool Success { get; private set; }
 
        public PipelineResults(Dictionary<string, object> inputs)
        {
            Inputs = new Dictionary<string, object>(inputs);
        }
 
        public T GetArgument<T>(string key)
        {
            if (Outputs.TryGetValue(key, out var value))
            {
                if (value is T outputValue)
                    return outputValue;
                throw new System.InvalidCastException($"Invalid cast for argument: {key}. Attempted to cast {value.GetType()} to {typeof(T)}.");
            }
 
            if (Inputs[key] is T inputValue)
                return inputValue;
            throw new System.InvalidCastException($"Invalid cast for argument: {key}. Attempted to cast {Inputs[key].GetType()} to {typeof(T)}.");
        }
 
        public T GetInput<T>(string key)
        {
            if (Inputs[key] is T value)
                return value;
            throw new System.InvalidCastException($"Invalid cast for argument: {key}. Attempted to cast {Inputs[key].GetType()} to {typeof(T)}.");
        }
 
        public T GetOutput<T>(string key)
        {
            if (Outputs[key] is T value)
                return value;
            throw new System.InvalidCastException($"Invalid cast for argument: {key}. Attempted to cast {Outputs[key].GetType()} to {typeof(T)}.");
        }
 
        public void SetOutput<T>(string key, T value)
        {
            Outputs[key] = value;
        }
 
        public void AddOutput<T>(string key, T value)
        {
            Outputs.Add(key, value);
        }
 
        public void Complete()
        {
            Success = true;
        }
    }
}