Merge pull request #553 from loresoft/pacoweb-master

Pacoweb master

Author: pwelter34

src/EntityFrameworkCore.Generator.Core/CodeGenerator.cs

@@ -436,24 +436,28 @@ private void ConfigureMySqlServices(IServiceCollection services)
     {
         var designTimeServices = new Pomelo.EntityFrameworkCore.MySql.Design.Internal.MySqlDesignTimeServices();
         designTimeServices.ConfigureDesignTimeServices(services);
+            services.AddEntityFrameworkMySqlNetTopologySuite();
     }
 
     private void ConfigurePostgresServices(IServiceCollection services)
     {
         var designTimeServices = new Npgsql.EntityFrameworkCore.PostgreSQL.Design.Internal.NpgsqlDesignTimeServices();
         designTimeServices.ConfigureDesignTimeServices(services);
+            services.AddEntityFrameworkNpgsqlNetTopologySuite();
     }
 
     private void ConfigureSqlServerServices(IServiceCollection services)
     {
         var designTimeServices = new Microsoft.EntityFrameworkCore.SqlServer.Design.Internal.SqlServerDesignTimeServices();
         designTimeServices.ConfigureDesignTimeServices(services);
+            services.AddEntityFrameworkSqlServerNetTopologySuite();
     }
 
     private void ConfigureSqliteServices(IServiceCollection services)
     {
         var designTimeServices = new Microsoft.EntityFrameworkCore.Sqlite.Design.Internal.SqliteDesignTimeServices();
         designTimeServices.ConfigureDesignTimeServices(services);
+            services.AddEntityFrameworkSqliteNetTopologySuite();
     }
 
     private void ConfigureOracleServices(IServiceCollection services)

src/EntityFrameworkCore.Generator.Core/EntityFrameworkCore.Generator.Core.csproj

@@ -1,4 +1,4 @@
-<Project Sdk="Microsoft.NET.Sdk">
+<Project Sdk="Microsoft.NET.Sdk">
   <PropertyGroup>
     <TargetFramework>net8.0</TargetFramework>
     <RootNamespace>EntityFrameworkCore.Generator</RootNamespace>
@@ -7,8 +7,12 @@
 
   <ItemGroup>
     <PackageReference Include="Microsoft.CodeAnalysis.CSharp.Scripting" Version="4.5.0" />
+    <PackageReference Include="Microsoft.EntityFrameworkCore.Sqlite.NetTopologySuite" Version="5.0.8" />
+    <PackageReference Include="Microsoft.EntityFrameworkCore.SqlServer.NetTopologySuite" Version="5.0.8" />
     <PackageReference Include="Microsoft.Extensions.Configuration.UserSecrets" Version="8.0.0" />
+    <PackageReference Include="Npgsql.EntityFrameworkCore.PostgreSQL.NetTopologySuite" Version="5.0.7" />
     <PackageReference Include="Microsoft.EntityFrameworkCore.Design" Version="8.0.3" />
+    <PackageReference Include="Pomelo.EntityFrameworkCore.MySql.NetTopologySuite" Version="5.0.1" />
     <PackageReference Include="Microsoft.EntityFrameworkCore.Relational" Version="8.0.3" />
     <PackageReference Include="Microsoft.EntityFrameworkCore.SqlServer" Version="8.0.3" />
     <PackageReference Include="Microsoft.EntityFrameworkCore.Sqlite" Version="8.0.3" />

test/EntityFrameworkCore.Generator.Core.Tests/CodeGeneratorTests.cs

@@ -2,6 +2,7 @@
 using FluentAssertions;
 using FluentCommand.SqlServer.Tests;
 using Microsoft.Extensions.Logging.Abstractions;
+using System.IO;
 using Xunit;
 using Xunit.Abstractions;
 
@@ -26,4 +27,38 @@ public void Generate()
         result.Should().BeTrue();
     }
 
+        [Fact]
+        public void GenerateSpatial()
+        {
+            var generatorOptions = new GeneratorOptions();
+            generatorOptions.Database.ConnectionString = Database.ConnectionString;
+
+            var generator = new CodeGenerator(NullLoggerFactory.Instance);
+            var result = generator.Generate(generatorOptions);
+
+            result.Should().BeTrue();
+
+            const string spatialTableName = "CitiesSpatial";
+
+            var citiesSpatialEntityFile  = Path.Combine(generatorOptions.Data.Entity.Directory, spatialTableName + ".cs");
+            var citiesSpatialMappingFile = Path.Combine(generatorOptions.Data.Mapping.Directory, spatialTableName + "Map.cs");
+
+            var citiesSpatialEntityContent  = File.ReadAllText(citiesSpatialEntityFile);
+            var citiesSpatialMappingContent = File.ReadAllText(citiesSpatialMappingFile);
+
+            citiesSpatialEntityContent.Contains("public NetTopologySuite.Geometries.Geometry GeometryField { get; set; }").Should().BeTrue();
+            citiesSpatialEntityContent.Contains("public NetTopologySuite.Geometries.Geometry GeographyField { get; set; }").Should().BeTrue();
+
+            citiesSpatialMappingContent.Contains("builder.Property(t => t.GeometryField)" + System.Environment.NewLine +
+"                .IsRequired()" + System.Environment.NewLine +
+"                .HasColumnName(\"GeometryField\")" + System.Environment.NewLine +
+"                .HasColumnType(\"geometry\");").Should().BeTrue();
+
+            citiesSpatialMappingContent.Contains("builder.Property(t => t.GeographyField)" + System.Environment.NewLine +
+"                .IsRequired()" + System.Environment.NewLine +
+"                .HasColumnName(\"GeographyField\")" + System.Environment.NewLine +
+"                .HasColumnType(\"geography\");").Should().BeTrue();
+
+        }
+
 }

test/EntityFrameworkCore.Generator.Core.Tests/Scripts/Script001.Tracker.Schema.sql

@@ -142,6 +142,14 @@ CREATE TABLE [dbo].[UserRole] (
     CONSTRAINT [PK_UserRole] PRIMARY KEY ([UserId], [RoleId])
 );
 
+IF NOT EXISTS (SELECT * FROM sys.objects WHERE object_id = OBJECT_ID(N'[dbo].[CitiesSpatial]') AND type in (N'U'))
+CREATE TABLE [dbo].[CitiesSpatial](
+    [Id] int IDENTITY(1,1) NOT NULL,
+    [Name] [nvarchar](50) NULL,
+    [GeometryField] [geometry] NOT NULL,
+    [GeographyField] [geography] NOT NULL,
+ CONSTRAINT [PK_CitiesSpatial] PRIMARY KEY ([Id])
+) 
 
 -- Foreign Keys
 IF NOT EXISTS (SELECT * FROM sys.foreign_keys WHERE object_id = OBJECT_ID(N'[dbo].[FK_Task_Priority_PriorityId]') AND parent_object_id = OBJECT_ID(N'[dbo].[Task]'))

test/EntityFrameworkCore.Generator.Core.Tests/Scripts/Script002.Tracker.Data.sql

@@ -84,3 +84,10 @@ WHEN MATCHED THEN
     UPDATE SET t.[Name] = s.[Name], t.[Description] = s.[Description]
 OUTPUT $action as [Action];
 
+-- Table [dbo].[CitiesSpatial] data
+
+INSERT INTO [dbo].[CitiesSpatial] VALUES
+('Sydney', 'POINT(151.2 -33.8)', 'POINT(151.2 -33.8)'),
+('Athens', 'POINT(23.7 38)', 'POINT(23.7 38)'),
+('Beijing', 'POINT(116.4 39.9)', 'POINT(116.4 39.9)'),
+('London', 'POINT(-0.15 51.5)', 'POINT(-0.15 51.5)');