openscad-graph-editor

using System.Collections.Generic;
using GodotExt;
using OpenScadGraphEditor.Library;
using OpenScadGraphEditor.Library.IO;
using OpenScadGraphEditor.Utils;

namespace OpenScadGraphEditor.Nodes
{
    public class FunctionReturn : ScadNode, ICannotBeDeleted, IReferToAFunction
    {
        private FunctionDescription _description;

        public override string NodeTitle => "Return value";
        public override string NodeQuickLookup => "";
        public override string NodeDescription => "Returns the given value.";

        public InvokableDescription InvokableDescription => _description;

        public override void SaveInto(SavedNode node)
        {
            node.SetData("function_description_id", _description.Id);
            base.SaveInto(node);
        }

        public override string GetPortDocumentation(PortId portId)
        {
            switch (portId.Port)
            {
                case 0 when portId.IsInput:
                    return _description.ReturnValueDescription;
                default:
                    return "";
            }
        }

        public int GetParameterInputPort(int parameterIndex)
        {
            // no ports refer to parameters.
            return -1;
        }

        public int GetParameterOutputPort(int parameterIndex)
        {
            // no ports refer to parameters.
            return -1;
        }


        public IEnumerable<PortId> GetPortsReferringToReturnValue()
        {
            return new[] {PortId.Input(0)};
        }


        public override void RestorePortDefinitions(SavedNode node, IReferenceResolver referenceResolver)
        {
            _description = referenceResolver.ResolveFunctionReference(node.GetDataString("function_description_id"));
            SetupPorts(_description);
            base.RestorePortDefinitions(node, referenceResolver);
        }

        public void SetupPorts(InvokableDescription description)
        {
            GdAssert.That(description is FunctionDescription, "need a function description");
            InputPorts.Clear();
            OutputPorts.Clear();

            _description = (FunctionDescription) description;

            InputPorts
                .OfType(_description.ReturnTypeHint, "Result");
        }

        public override string Render(ScadGraph context, int portIndex)
        {
            return RenderInput(context, 0).OrUndef();
        }
    }
}