Example #1
0
        /// <summary>
        /// Create the template output
        /// </summary>
        public virtual string TransformText()
        {
            var code = new VBCodeHelper();
            var edm  = new EdmHelper(code);

            if (EntitySet == null)
            {
                throw new ArgumentNullException("EntitySet");
            }

            if (Model == null)
            {
                throw new ArgumentNullException("Model");
            }

            var entityType = EntitySet.ElementType;

            this.Write("Imports System\r\nImports System.Collections.Generic\r\nImports System.ComponentModel" +
                       ".DataAnnotations\r\nImports System.ComponentModel.DataAnnotations.Schema\r\nImports " +
                       "System.Data.Entity.Spatial\r\n\r\n");

            var typeConfigurations = edm.GetConfigurations(EntitySet, Model).OfType <IAttributeConfiguration>();

            foreach (var typeConfiguration in typeConfigurations)
            {
                this.Write(this.ToStringHelper.ToStringWithCulture(code.Attribute(typeConfiguration)));
                this.Write("\r\n");
            }

            this.Write("Partial Public Class ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(entityType)));
            this.Write("\r\n");

            var collectionProperties = from p in entityType.NavigationProperties
                                       where p.ToEndMember.RelationshipMultiplicity == RelationshipMultiplicity.Many
                                       select p;

            if (collectionProperties.Any())
            {
                this.Write("    Public Sub New()\r\n");

                foreach (var collectionProperty in collectionProperties)
                {
                    this.Write("        ");
                    this.Write(this.ToStringHelper.ToStringWithCulture(code.Property(collectionProperty)));
                    this.Write(" = New HashSet(Of ");
                    this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(collectionProperty.ToEndMember.GetEntityType())));
                    this.Write(")()\r\n");
                }

                this.Write("    End Sub\r\n\r\n");
            }

            var first = true;

            foreach (var property in entityType.Properties)
            {
                if (!first)
                {
                    WriteLine(string.Empty);
                }
                else
                {
                    first = false;
                }

                var propertyConfigurations = edm.GetConfigurations(property, Model).OfType <IAttributeConfiguration>();

                foreach (var propertyConfiguration in propertyConfigurations)
                {
                    this.Write("    ");
                    this.Write(this.ToStringHelper.ToStringWithCulture(code.Attribute(propertyConfiguration)));
                    this.Write("\r\n");
                }

                this.Write("    Public Property ");
                this.Write(this.ToStringHelper.ToStringWithCulture(code.Property(property)));
                this.Write(" As ");
                this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(property)));
                this.Write("\r\n");
            }

            foreach (var navigationProperty in entityType.NavigationProperties)
            {
                if (!first)
                {
                    WriteLine(string.Empty);
                }
                else
                {
                    first = false;
                }


                this.Write("    Public Overridable Property ");
                this.Write(this.ToStringHelper.ToStringWithCulture(code.Property(navigationProperty)));
                this.Write(" As ");
                this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(navigationProperty)));
                this.Write("\r\n");
            }

            this.Write("End Class\r\n");
            return(this.GenerationEnvironment.ToString());
        }
        /// <summary>
        /// Create the template output
        /// </summary>
        public virtual string TransformText()
        {

    var code = new VBCodeHelper();
    var edm = new EdmHelper(code);

    if (EntitySet == null)
    {
        throw new ArgumentNullException("EntitySet");
    }

    if (Model == null)
    {
        throw new ArgumentNullException("Model");
    }

    var entityType = EntitySet.ElementType;

            this.Write("Imports System\r\nImports System.Collections.Generic\r\nImports System.ComponentModel" +
                    ".DataAnnotations\r\nImports System.ComponentModel.DataAnnotations.Schema\r\nImports " +
                    "System.Data.Entity.Spatial\r\n\r\n");

    var typeConfigurations = edm.GetConfigurations(EntitySet, Model).OfType<IAttributeConfiguration>();

    foreach (var typeConfiguration in typeConfigurations)
    {

            this.Write(this.ToStringHelper.ToStringWithCulture(code.Attribute(typeConfiguration)));
            this.Write("\r\n");

    }

            this.Write("Partial Public Class ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(entityType)));
            this.Write("\r\n");

    var collectionProperties = from p in entityType.NavigationProperties
                               where p.ToEndMember.RelationshipMultiplicity == RelationshipMultiplicity.Many
                               select p;

    if (collectionProperties.Any())
    {

            this.Write("    Public Sub New()\r\n");

    foreach (var collectionProperty in collectionProperties)
    {

            this.Write("        ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Property(collectionProperty)));
            this.Write(" = New HashSet(Of ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(collectionProperty.ToEndMember.GetEntityType())));
            this.Write(")()\r\n");

    }

            this.Write("    End Sub\r\n\r\n");

    }

    var first = true;

    foreach (var property in entityType.Properties)
    {
        if (!first)
        {
            WriteLine(string.Empty);
        }
        else
        {
            first = false;
        }

        var propertyConfigurations = edm.GetConfigurations(property, Model).OfType<IAttributeConfiguration>();

        foreach (var propertyConfiguration in propertyConfigurations)
        {

            this.Write("    ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Attribute(propertyConfiguration)));
            this.Write("\r\n");

        }

            this.Write("    Public Property ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Property(property)));
            this.Write(" As ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(property)));
            this.Write("\r\n");

    }

    foreach (var navigationProperty in entityType.NavigationProperties)
    {
        if (!first)
        {
            WriteLine(string.Empty);
        }
        else
        {
            first = false;
        }


            this.Write("    Public Overridable Property ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Property(navigationProperty)));
            this.Write(" As ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(navigationProperty)));
            this.Write("\r\n");

    }

            this.Write("End Class\r\n");
            return this.GenerationEnvironment.ToString();
        }
        /// <summary>
        /// Create the template output
        /// </summary>
        public virtual string TransformText()
        {

    var code = new VBCodeHelper();
    var edm = new EdmHelper(code);

    if (Model == null)
    {
        throw new ArgumentNullException("Model");
    }

    if (Namespace == null)
    {
        throw new ArgumentNullException("Namespace");
    }

    if (ContextClassName == null)
    {
        throw new ArgumentNullException("ContextClassName");
    }

    if (ConnectionStringName == null)
    {
        throw new ArgumentNullException("ConnectionStringName");
    }

            this.Write("Imports System\r\nImports System.Data.Entity\r\nImports System.ComponentModel.DataAnn" +
                    "otations.Schema\r\nImports System.Linq\r\n\r\nPartial Public Class ");
            this.Write(this.ToStringHelper.ToStringWithCulture(ContextClassName));
            this.Write("\r\n    Inherits DbContext\r\n\r\n    Public Sub New()\r\n        MyBase.New(\"name=");
            this.Write(this.ToStringHelper.ToStringWithCulture(ConnectionStringName));
            this.Write("\")        \r\n    End Sub\r\n\r\n");

    foreach (var entitySet in Model.ConceptualModel.Container.EntitySets)
    {

            this.Write("    Public Overridable Property ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Property(entitySet)));
            this.Write(" As DbSet(Of ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(entitySet.ElementType)));
            this.Write(")\r\n");

    }

            this.Write("\r\n    Protected Overrides Sub OnModelCreating(ByVal modelBuilder As DbModelBuilde" +
                    "r)\r\n");

    var anyConfiguration = false;

    foreach (var entitySet in Model.ConceptualModel.Container.EntitySets)
    {
        var typeConfigurations = edm.GetConfigurations(entitySet, Model).OfType<IFluentConfiguration>()
            .Where(c => !(c is IAttributeConfiguration || c is KeyConfiguration));

        var firstTypeConfiguration = true;
        foreach (var typeConfiguration in typeConfigurations)
        {
            if (firstTypeConfiguration)
            {
                firstTypeConfiguration = false;

                if (anyConfiguration)
                {
                    WriteLine(string.Empty);
                }
                else
                {
                    anyConfiguration = true;
                }


            this.Write("        modelBuilder.Entity(Of ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(entitySet.ElementType)));
            this.Write(")() _\r\n");

            }
            else
            {
                WriteLine(" _");
            }

            Write("            " + code.MethodChain(typeConfiguration));
        }

        if (!firstTypeConfiguration)
        {
            WriteLine(string.Empty);
        }

        foreach (var property in entitySet.ElementType.Properties)
        {
            var propertyConfigurations = edm.GetConfigurations(property, Model).OfType<IFluentConfiguration>()
                .Where(c => !(c is IAttributeConfiguration));

            var firstPropertyConfiguration = true;
            foreach (var propertyConfiguration in propertyConfigurations)
            {
                var columnConfiguration = propertyConfiguration as ColumnConfiguration;
                if (columnConfiguration != null)
                {
                    // Unset this since it is implied in the key configuration calls themselves
                    columnConfiguration.Order = null;

                    if (columnConfiguration.Name == null && columnConfiguration.TypeName == null)
                    {
                        // Nothing left to configure
                        continue;
                    }
                }

                if (firstPropertyConfiguration)
                {
                    firstPropertyConfiguration = false;
                    
                    if (anyConfiguration)
                    {
                        WriteLine(string.Empty);
                    }
                    else
                    {
                        anyConfiguration = true;
                    }


            this.Write("        modelBuilder.Entity(Of ");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(entitySet.ElementType)));
            this.Write(")() _\r\n            .Property(Function(e) e.");
            this.Write(this.ToStringHelper.ToStringWithCulture(code.Property(property)));
            this.Write(") _\r\n");

                }
                else
                {
                    WriteLine(" _");
                }

                Write("            " + code.MethodChain(propertyConfiguration));
            }

            if (!firstPropertyConfiguration)
            {
                WriteLine(string.Empty);
            }
        }

        foreach (var navigationProperty in entitySet.ElementType.NavigationProperties)
        {
            // Only configure relationships from one end
            if (navigationProperty.RelationshipType.RelationshipEndMembers.First() != navigationProperty.FromEndMember)
            {
                continue;
            }

            bool isDefaultMultiplicity;
            var navigationPropertyMultiplicityConfiguration = edm.GetMultiplicityConfiguration(navigationProperty, out isDefaultMultiplicity);
            var navigationPropertyConfigurations = edm.GetConfigurations(navigationProperty, Model);

            var firstNavigationPropertyConfiguration = true;
            foreach (var navigationPropertyConfiguration in navigationPropertyConfigurations)
            {
                if (firstNavigationPropertyConfiguration)
                {
                    firstNavigationPropertyConfiguration = false;
                    
                    if (anyConfiguration)
                    {
                        WriteLine(string.Empty);
                    }
                    else
                    {
                        anyConfiguration = true;
                    }


            this.Write("        modelBuilder");
            this.Write(this.ToStringHelper.ToStringWithCulture(ApplyVBFixup(code.MethodChain(navigationPropertyMultiplicityConfiguration))));
            this.Write(" _\r\n");

                }
                else
                {
                    WriteLine(" _");
                }

                Write("            " + code.MethodChain(navigationPropertyConfiguration));
            }

            if (!firstNavigationPropertyConfiguration)
            {
                WriteLine(string.Empty);
            }
            else if (!isDefaultMultiplicity)
            {            
                if (anyConfiguration)
                {
                    WriteLine(string.Empty);
                }
                else
                {
                    anyConfiguration = true;
                }

            this.Write("        modelBuilder");
            this.Write(this.ToStringHelper.ToStringWithCulture(ApplyVBFixup(code.MethodChain(navigationPropertyMultiplicityConfiguration))));
            this.Write("\r\n");

            }
        }
    }

            this.Write("    End Sub\r\nEnd Class\r\n");
            return this.GenerationEnvironment.ToString();
        }
Example #4
0
        /// <summary>
        /// Create the template output
        /// </summary>
        public virtual string TransformText()
        {
            var code = new VBCodeHelper();
            var edm  = new EdmHelper(code);

            if (Model == null)
            {
                throw new ArgumentNullException("Model");
            }

            if (Namespace == null)
            {
                throw new ArgumentNullException("Namespace");
            }

            if (ContextClassName == null)
            {
                throw new ArgumentNullException("ContextClassName");
            }

            if (ConnectionStringName == null)
            {
                throw new ArgumentNullException("ConnectionStringName");
            }

            this.Write("Imports System\r\nImports System.Data.Entity\r\nImports System.ComponentModel.DataAnn" +
                       "otations.Schema\r\nImports System.Linq\r\n\r\nPartial Public Class ");
            this.Write(this.ToStringHelper.ToStringWithCulture(ContextClassName));
            this.Write("\r\n    Inherits DbContext\r\n\r\n    Public Sub New()\r\n        MyBase.New(\"name=");
            this.Write(this.ToStringHelper.ToStringWithCulture(ConnectionStringName));
            this.Write("\")        \r\n    End Sub\r\n\r\n");

            foreach (var entitySet in Model.ConceptualModel.Container.EntitySets)
            {
                this.Write("    Public Overridable Property ");
                this.Write(this.ToStringHelper.ToStringWithCulture(code.Property(entitySet)));
                this.Write(" As DbSet(Of ");
                this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(entitySet.ElementType)));
                this.Write(")\r\n");
            }

            this.Write("\r\n    Protected Overrides Sub OnModelCreating(ByVal modelBuilder As DbModelBuilde" +
                       "r)\r\n");

            var anyConfiguration = false;

            foreach (var entitySet in Model.ConceptualModel.Container.EntitySets)
            {
                var typeConfigurations = edm.GetConfigurations(entitySet, Model).OfType <IFluentConfiguration>()
                                         .Where(c => !(c is IAttributeConfiguration || c is KeyConfiguration));

                var firstTypeConfiguration = true;
                foreach (var typeConfiguration in typeConfigurations)
                {
                    if (firstTypeConfiguration)
                    {
                        firstTypeConfiguration = false;

                        if (anyConfiguration)
                        {
                            WriteLine(string.Empty);
                        }
                        else
                        {
                            anyConfiguration = true;
                        }


                        this.Write("        modelBuilder.Entity(Of ");
                        this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(entitySet.ElementType)));
                        this.Write(")() _\r\n");
                    }
                    else
                    {
                        WriteLine(" _");
                    }

                    Write("            " + code.MethodChain(typeConfiguration));
                }

                if (!firstTypeConfiguration)
                {
                    WriteLine(string.Empty);
                }

                foreach (var property in entitySet.ElementType.Properties)
                {
                    var propertyConfigurations = edm.GetConfigurations(property, Model).OfType <IFluentConfiguration>()
                                                 .Where(c => !(c is IAttributeConfiguration));

                    var firstPropertyConfiguration = true;
                    foreach (var propertyConfiguration in propertyConfigurations)
                    {
                        var columnConfiguration = propertyConfiguration as ColumnConfiguration;
                        if (columnConfiguration != null)
                        {
                            // Unset this since it is implied in the key configuration calls themselves
                            columnConfiguration.Order = null;

                            if (columnConfiguration.Name == null && columnConfiguration.TypeName == null)
                            {
                                // Nothing left to configure
                                continue;
                            }
                        }

                        if (firstPropertyConfiguration)
                        {
                            firstPropertyConfiguration = false;

                            if (anyConfiguration)
                            {
                                WriteLine(string.Empty);
                            }
                            else
                            {
                                anyConfiguration = true;
                            }


                            this.Write("        modelBuilder.Entity(Of ");
                            this.Write(this.ToStringHelper.ToStringWithCulture(code.Type(entitySet.ElementType)));
                            this.Write(")() _\r\n            .Property(Function(e) e.");
                            this.Write(this.ToStringHelper.ToStringWithCulture(code.Property(property)));
                            this.Write(") _\r\n");
                        }
                        else
                        {
                            WriteLine(" _");
                        }

                        Write("            " + code.MethodChain(propertyConfiguration));
                    }

                    if (!firstPropertyConfiguration)
                    {
                        WriteLine(string.Empty);
                    }
                }

                foreach (var navigationProperty in entitySet.ElementType.NavigationProperties)
                {
                    // Only configure relationships from one end
                    if (navigationProperty.RelationshipType.RelationshipEndMembers.First() != navigationProperty.FromEndMember)
                    {
                        continue;
                    }

                    bool isDefaultMultiplicity;
                    var  navigationPropertyMultiplicityConfiguration = edm.GetMultiplicityConfiguration(navigationProperty, out isDefaultMultiplicity);
                    var  navigationPropertyConfigurations            = edm.GetConfigurations(navigationProperty, Model);

                    var firstNavigationPropertyConfiguration = true;
                    foreach (var navigationPropertyConfiguration in navigationPropertyConfigurations)
                    {
                        if (firstNavigationPropertyConfiguration)
                        {
                            firstNavigationPropertyConfiguration = false;

                            if (anyConfiguration)
                            {
                                WriteLine(string.Empty);
                            }
                            else
                            {
                                anyConfiguration = true;
                            }


                            this.Write("        modelBuilder");
                            this.Write(this.ToStringHelper.ToStringWithCulture(ApplyVBFixup(code.MethodChain(navigationPropertyMultiplicityConfiguration))));
                            this.Write(" _\r\n");
                        }
                        else
                        {
                            WriteLine(" _");
                        }

                        Write("            " + code.MethodChain(navigationPropertyConfiguration));
                    }

                    if (!firstNavigationPropertyConfiguration)
                    {
                        WriteLine(string.Empty);
                    }
                    else if (!isDefaultMultiplicity)
                    {
                        if (anyConfiguration)
                        {
                            WriteLine(string.Empty);
                        }
                        else
                        {
                            anyConfiguration = true;
                        }

                        this.Write("        modelBuilder");
                        this.Write(this.ToStringHelper.ToStringWithCulture(ApplyVBFixup(code.MethodChain(navigationPropertyMultiplicityConfiguration))));
                        this.Write("\r\n");
                    }
                }
            }

            this.Write("    End Sub\r\nEnd Class\r\n");
            return(this.GenerationEnvironment.ToString());
        }