public GeneratorDriver AddGenerators(ImmutableArray <ISourceGenerator> generators)
{
// set editsFailed true, as we won't be able to apply edits with a new generator
var newState = _state.With(generators: _state.Generators.AddRange(generators), generatorStates: _state.GeneratorStates.AddRange(new GeneratorState[generators.Length]), editsFailed: true);
return(FromState(newState));
}
public GeneratorDriver AddGenerators(ImmutableArray <ISourceGenerator> generators)
{
(var filteredGenerators, var incrementalGenerators) = GetIncrementalGenerators(generators, SourceExtension);
var newState = _state.With(sourceGenerators: _state.Generators.AddRange(filteredGenerators),
incrementalGenerators: _state.IncrementalGenerators.AddRange(incrementalGenerators),
generatorStates: _state.GeneratorStates.AddRange(new GeneratorState[filteredGenerators.Length]));
return(FromState(newState));
}
private GeneratorDriverState ApplyPartialEdit(GeneratorDriverState state, PendingEdit edit, CancellationToken cancellationToken = default)
{
var initialState = state;
// see if any generators accept this particular edit
var stateBuilder = PooledDictionary <ISourceGenerator, GeneratorState> .GetInstance();
for (int i = 0; i < initialState.Generators.Length; i++)
{
var generator = initialState.Generators[i];
var generatorState = initialState.GeneratorStates[i];
if (edit.AcceptedBy(generatorState.Info))
{
// attempt to apply the edit
var previousSources = CreateSourcesCollection();
previousSources.AddRange(generatorState.SourceTexts);
var context = new GeneratorEditContext(previousSources, cancellationToken);
var succeeded = edit.TryApply(generatorState.Info, context);
if (!succeeded)
{
// we couldn't successfully apply this edit
// return the original state noting we failed
return(initialState.With(editsFailed: true));
}
// update the state with the new edits
var additionalSources = previousSources.ToImmutableAndFree();
state = state.With(generatorStates: state.GeneratorStates.SetItem(i, new GeneratorState(generatorState.Info, sourceTexts: additionalSources, trees: ParseAdditionalSources(generator, additionalSources, cancellationToken), diagnostics: ImmutableArray <Diagnostic> .Empty)));
}
}
state = edit.Commit(state);
return(state);
}
private GeneratorDriverState ApplyPartialEdit(GeneratorDriverState state, PendingEdit edit, CancellationToken cancellationToken = default)
{
var initialState = state;
// see if any generators accept this particular edit
var stateBuilder = PooledDictionary <ISourceGenerator, GeneratorState> .GetInstance();
foreach (var(generator, generatorState) in state.GeneratorStates)
{
if (edit.AcceptedBy(generatorState.Info))
{
// attempt to apply the edit
var context = new EditContext(generatorState.Sources.Keys.ToImmutableArray(), cancellationToken);
var succeeded = edit.TryApply(generatorState.Info, context);
if (!succeeded)
{
// we couldn't successfully apply this edit
// return the original state noting we failed
return(initialState.With(editsFailed: true));
}
// update the state with the new edits
state = state.With(generatorStates: state.GeneratorStates.SetItem(generator, generatorState.WithSources(ParseAdditionalSources(context.AdditionalSources.ToImmutableAndFree(), cancellationToken))));
}
}
state = edit.Commit(state);
return(state);
}
private static GeneratorDriverState StateWithPendingEditsApplied(GeneratorDriverState state)
{
if (state.Edits.Length == 0)
{
return(state);
}
foreach (var edit in state.Edits)
{
state = edit.Commit(state);
}
return(state.With(edits: ImmutableArray <PendingEdit> .Empty, editsFailed: false));
}
private GeneratorDriver BuildFinalCompilation(Compilation compilation, out Compilation outputCompilation, GeneratorDriverState state, CancellationToken cancellationToken)
{
ArrayBuilder <SyntaxTree> trees = ArrayBuilder <SyntaxTree> .GetInstance();
foreach (var generatorState in state.GeneratorStates)
{
trees.AddRange(generatorState.Trees);
}
outputCompilation = compilation.AddSyntaxTrees(trees);
trees.Free();
state = state.With(edits: ImmutableArray <PendingEdit> .Empty,
editsFailed: false);
return(FromState(state));
}
internal override GeneratorDriverState Commit(GeneratorDriverState state) => state.With(additionalTexts: state.AdditionalTexts.Add(this.AddedText));
