openscad-graph-editor

using System.Collections.Generic;
using System.Linq;
using OpenScadGraphEditor.Actions;
using OpenScadGraphEditor.Library;
using OpenScadGraphEditor.Refactorings;
using OpenScadGraphEditor.Utils;
using OpenScadGraphEditor.Widgets.AddDialog;

namespace OpenScadGraphEditor.Nodes
{
    /// <summary>
    /// Factory which adds entries to get/set each variable defined in the project.
    /// </summary>
    public class GetterSetterAddDialogEntryFactory : IAddDialogEntryFactory
    {
        public IEnumerable<IAddDialogEntry> BuildEntries(IEditorContext editorContext)
        {
            var result = new List<IAddDialogEntry>();
            
            
            // for the variables directly in the project
            result.AddRange(
                editorContext.CurrentProject.Variables
                    .Select(it => new SingleNodeBasedEntry(
                        Resources.VariableIcon,
                        () => NodeFactory.Build<SetVariable>(it),
                        editorContext
                    ))
            );

            result.AddRange(
                editorContext.CurrentProject.Variables
                    .Select(it => new SingleNodeBasedEntry(
                        Resources.VariableIcon,
                        () => NodeFactory.Build<GetVariable>(it),
                        editorContext
                    ))
            );
            
            // for the variables in external modules
            var externalVariableDescriptions = editorContext.CurrentProject.ExternalReferences
                .SelectMany(it => it.Variables)
                .ToList();

            result.AddRange(
                externalVariableDescriptions
                    .Select(it => new SingleNodeBasedEntry(
                        Resources.VariableIcon,
                        () => NodeFactory.Build<SetVariable>(it),
                        editorContext
                    ))
            );

            result.AddRange(
                externalVariableDescriptions
                    .Select(it => new SingleNodeBasedEntry(
                        Resources.VariableIcon,
                        () => NodeFactory.Build<GetVariable>(it),
                        editorContext
                    ))
            );

            return result;
        }
    }
}