Groovy.cs

Go to the documentation of this file.

using System.Linq;

namespace CodeGen.generators
{
    public class GroovyGenerator : Generator
    {
        private const string ClassFormat = "{0}class {1}{2}{{{3}{4}{5}}}";
        private string Indent { get; set; } = GeneratorConf.GetIndent(true, 4);

        protected override string GenerateClass(Class @class)
        {
            string fields = "", inherits = " ", methods = "", classes = "";
            if (@class.Parent?.Length > 0)
            {
                inherits = " extends " + @class.Parent + ' ';
            }

            fields = @class.Fields?.Aggregate('\n' + fields,
                (current, field) => current + GenerateField(field) + '\n');

            methods = @class.Methods?.Aggregate(methods,
                (current, method) => current + '\n' + GeneratorConf.ShiftCode(GenerateMethod(method), 1, Indent) + '\n');
            
            classes = @class.Classes?.Aggregate(classes,
                (current, cls) => current + '\n' + GeneratorConf.ShiftCode(GenerateClass(cls), 1, Indent) + '\n');
            
            var access = "";
            if (@class.Access?.Length > 0)
            {
                access = @class.Access + " ";
            }
            
            return string.Format(ClassFormat, access, @class.Name, inherits, fields, methods, classes);;
        }

        public override string GenerateField(Field field)
        {
            var result = Indent;
            if (field.Access == "" || field.Access == "default")
            {
                result += "private ";
            }
            else
            {
                result += field.Access + ' ';
            }

            if (field.Static)
            {
                result += "static ";
            }

            switch (field.Type)
            {
                case "string":
                    result += "String ";
                    break;
                default:
                    result += field.Type + ' ';
                    break;
            }

            result += field.Name;
            if (field.Default?.Length > 0)
            {
                result += " = " + field.Default;
            }
            return result;
        }

        public override string GenerateMethod(Method method)
        {
            var result = "";
            if (method.Access == "" || method.Access == "default")
            {
                result += "private ";
            }
            else
            {
                result += method.Access + ' ';
            }

            switch (method.Type)
            {
                case "string":
                    result += "String ";
                    break;
                case "":
                    result += "void ";
                    break;
                default:
                    result += method.Type + ' ';
                    break;
            }

            result += method.Name + '(';

            for (var i = 0; i < method.Parameters?.Length; i++)
            {
                var parameter = method.Parameters[i];
                if (parameter.Type == "string")
                {
                    parameter.Type = "String";
                }

                result += parameter.Type + ' ' + parameter.Name;
                if (i + 1 < method.Parameters.Length)
                {
                    result += ", ";
                }
            }

            result += ") {";

            if (method.Type?.Length > 0)
            {
                result += '\n' + Indent + "return 0\n";
            }

            result += '}';
            return result;
        }
    }
}