323 lines
14 KiB
C#
323 lines
14 KiB
C#
using Microsoft.AspNetCore.Identity;
|
||
using Server.Models;
|
||
using Utils;
|
||
using Route = Server.Models.Route;
|
||
|
||
namespace Server.Data;
|
||
|
||
public class SeedData
|
||
{
|
||
public static async Task Initialize(IServiceProvider serviceProvider)
|
||
{
|
||
var userManager = (UserManager<User>)serviceProvider.GetService(typeof(UserManager<User>))!;
|
||
userManager.UserValidators.Clear();
|
||
|
||
var roleManager = (RoleManager<IdentityRole>)serviceProvider.GetService(typeof(RoleManager<IdentityRole>))!;
|
||
var dbContext = (ApplicationDbContext)serviceProvider.GetService(typeof(ApplicationDbContext))!;
|
||
|
||
// Seed Roles
|
||
foreach (var role in Enum.GetValues(typeof(Identity.Roles)))
|
||
{
|
||
await roleManager.CreateAsync(new IdentityRole(role.ToString()));
|
||
}
|
||
|
||
User? companyUser;
|
||
User driverUser;
|
||
if (!dbContext.Users.Any())
|
||
{
|
||
// Seed Administrator user
|
||
|
||
var adminUser = new User
|
||
{
|
||
FirstName = "user", LastName = "user", Patronymic = "user",
|
||
Email = "admin@autobus.com", EmailConfirmed = true
|
||
};
|
||
|
||
await userManager.CreateAsync(adminUser, Identity.DefaultPassword);
|
||
await userManager.AddToRoleAsync(adminUser, Identity.Roles.Administrator.ToString());
|
||
|
||
// Seed Company user
|
||
|
||
companyUser = new User
|
||
{
|
||
FirstName = "Firstname", LastName = "Lastname", Patronymic = "Patronymic",
|
||
Email = "company@autobus.com",
|
||
EmailConfirmed = true
|
||
};
|
||
|
||
await userManager.CreateAsync(companyUser, Identity.DefaultPassword);
|
||
await userManager.AddToRoleAsync(companyUser, Identity.Roles.Company.ToString());
|
||
|
||
// Seed Driver user
|
||
|
||
driverUser = new User
|
||
{
|
||
FirstName = "user", LastName = "user", Patronymic = "user",
|
||
Email = "driver@autobus.com", EmailConfirmed = true,
|
||
PhoneNumber = "+380951935723", PhoneNumberConfirmed = true,
|
||
BirthDate = DateTime.Now.ToUniversalTime(),
|
||
Document = Identity.Document.Passport, DocumentDetails = "Some details",
|
||
Gender = Identity.Gender.Male
|
||
};
|
||
|
||
await userManager.CreateAsync(driverUser, Identity.DefaultPassword);
|
||
await userManager.AddToRoleAsync(driverUser, Identity.Roles.Driver.ToString());
|
||
|
||
// Seed Default user
|
||
|
||
var defaultUser = new User
|
||
{
|
||
FirstName = "user", LastName = "user", Patronymic = "user",
|
||
Email = "user@autobus.com", EmailConfirmed = true
|
||
};
|
||
|
||
await userManager.CreateAsync(defaultUser, Identity.DefaultPassword);
|
||
await userManager.AddToRoleAsync(defaultUser, Identity.Roles.User.ToString());
|
||
}
|
||
|
||
companyUser = await userManager.FindByEmailAsync("company@autobus.com");
|
||
driverUser = await userManager.FindByEmailAsync("driver@autobus.com");
|
||
|
||
if (!dbContext.Companies.Any())
|
||
{
|
||
// Seed County - State - City - Address relations
|
||
|
||
await dbContext.Countries.AddRangeAsync(new Country { Name = "Ukraine", Code = "UA" });
|
||
await dbContext.SaveChangesAsync();
|
||
|
||
await dbContext.States.AddRangeAsync(new State { CountryId = 1, Name = "Kharkiv Oblast" });
|
||
await dbContext.SaveChangesAsync();
|
||
|
||
await dbContext.Cities.AddRangeAsync(new City { StateId = 1, Name = "Osykovyi Hai" },
|
||
new City { StateId = 1, Name = "Malynivka" },
|
||
new City { StateId = 1, Name = "Chuhuiv" },
|
||
new City { StateId = 1, Name = "Kam'yana Yaruha" },
|
||
new City { StateId = 1, Name = "Rohan'" },
|
||
new City { StateId = 1, Name = "Kharkiv" },
|
||
new City { StateId = 1, Name = "Korobochkyne" },
|
||
new City { StateId = 1, Name = "Kochetok" },
|
||
new City { StateId = 1, Name = "Kytsivka" });
|
||
await dbContext.SaveChangesAsync();
|
||
|
||
await dbContext.Addresses.AddRangeAsync(new Address { CityId = 1, Name = "station near E40 road", Latitude = 36.778044, Longitude = 49.777235 },
|
||
new Address { CityId = 2, Name = "v. Malynivka", Latitude = 36.732271, Longitude = 49.813638 },
|
||
new Address { CityId = 3, Name = "Chuhuyivsʹka Avtostantsiya, Vulytsya Kharkivsʹka, 133", Latitude = 36.687534, Longitude = 49.838337 },
|
||
new Address { CityId = 4, Name = "Kulynychi", Latitude = 36.595877, Longitude = 49.880585 },
|
||
new Address { CityId = 5, Name = "v. Rohan", Latitude = 36.495910, Longitude = 49.907080 },
|
||
new Address { CityId = 6, Name = "Kharkiv Parking Near Mcdonalds", Latitude = 36.207202, Longitude = 49.987560 },
|
||
new Address { CityId = 7, Name = "Ukrposhta 63540, Chuhuivs'ka St, 2Б", Latitude = 36.812743, Longitude = 49.776519 },
|
||
new Address { CityId = 8, Name = "Pivnyk", Latitude = 36.712386, Longitude = 49.867814 },
|
||
new Address { CityId = 9, Name = "v. Kytsivka", Latitude = 36.819010, Longitude = 49.869401 });
|
||
await dbContext.SaveChangesAsync();
|
||
|
||
// Seed Route - RouteAddresses relations
|
||
|
||
await dbContext.Routes.AddRangeAsync(
|
||
new Route
|
||
{
|
||
Type = "default",
|
||
RouteAddresses = new List<RouteAddress>
|
||
{
|
||
new RouteAddress { Order = 1, AddressId = 1},
|
||
new RouteAddress { Order = 2, AddressId = 2},
|
||
new RouteAddress { Order = 3, AddressId = 3},
|
||
new RouteAddress { Order = 4, AddressId = 4},
|
||
new RouteAddress { Order = 5, AddressId = 5},
|
||
new RouteAddress { Order = 6, AddressId = 6},
|
||
}
|
||
},
|
||
new Route
|
||
{
|
||
Type = "default",
|
||
RouteAddresses = new List<RouteAddress>
|
||
{
|
||
new RouteAddress { Order = 1, AddressId = 7},
|
||
new RouteAddress { Order = 2, AddressId = 2},
|
||
new RouteAddress { Order = 3, AddressId = 3},
|
||
new RouteAddress { Order = 4, AddressId = 8},
|
||
new RouteAddress { Order = 5, AddressId = 9},
|
||
}
|
||
});
|
||
await dbContext.SaveChangesAsync();
|
||
|
||
|
||
await dbContext.Companies.AddAsync(
|
||
new Company
|
||
{
|
||
OwnerId = companyUser.Id,
|
||
Name = "Default Company",
|
||
CompanyDrivers = new []
|
||
{
|
||
new CompanyDriver
|
||
{
|
||
DriverId = driverUser.Id
|
||
}
|
||
}
|
||
});
|
||
await dbContext.SaveChangesAsync();
|
||
|
||
await dbContext.Vehicles.AddRangeAsync(
|
||
new Vehicle
|
||
{
|
||
Number = "AC8376BH",
|
||
Type = "Cruiser",
|
||
Capacity = 50,
|
||
CompanyId = 1,
|
||
HasClimateControl = true,
|
||
HasWiFi = true,
|
||
HasWC = true,
|
||
HasStewardess = true,
|
||
HasTV = false,
|
||
HasOutlet = true,
|
||
HasBelts = true,
|
||
},
|
||
new Vehicle
|
||
{
|
||
Number = "HA8934MK",
|
||
Type = "Lon Rider",
|
||
Capacity = 25,
|
||
CompanyId = 1,
|
||
HasClimateControl = false,
|
||
HasWiFi = false,
|
||
HasWC = false,
|
||
HasStewardess = false,
|
||
HasTV = false,
|
||
HasOutlet = false,
|
||
HasBelts = false,
|
||
});
|
||
await dbContext.SaveChangesAsync();
|
||
|
||
await dbContext.VehicleEnrollments.AddRangeAsync(
|
||
new VehicleEnrollment
|
||
{
|
||
RouteId = 1,
|
||
VehicleId = 1,
|
||
DepartureDateTimeUtc = new DateTime(DateTime.UtcNow.Year, DateTime.UtcNow.Month, DateTime.UtcNow.Day, 07, 45, 00, DateTimeKind.Utc),
|
||
RouteAddressDetails = new List<RouteAddressDetails>
|
||
{
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 1,
|
||
TimeSpanToNextCity = TimeSpan.FromMinutes(20),
|
||
CostToNextCity = 25
|
||
},
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 2,
|
||
TimeSpanToNextCity = TimeSpan.FromMinutes(5),
|
||
WaitTimeSpan = TimeSpan.FromMinutes(15),
|
||
CostToNextCity = 20
|
||
},
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 3,
|
||
TimeSpanToNextCity = TimeSpan.FromMinutes(5),
|
||
WaitTimeSpan = TimeSpan.FromMinutes(25),
|
||
CostToNextCity = 30
|
||
},
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 4,
|
||
TimeSpanToNextCity = TimeSpan.FromMinutes(5),
|
||
WaitTimeSpan = TimeSpan.FromMinutes(20),
|
||
CostToNextCity = 35
|
||
},
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 5,
|
||
TimeSpanToNextCity = TimeSpan.FromMinutes(5),
|
||
WaitTimeSpan = TimeSpan.FromMinutes(30),
|
||
CostToNextCity = 40
|
||
},
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 6,
|
||
}
|
||
}
|
||
},
|
||
new VehicleEnrollment
|
||
{
|
||
RouteId = 2,
|
||
VehicleId = 2,
|
||
DepartureDateTimeUtc = new DateTime(DateTime.UtcNow.Year, DateTime.UtcNow.Month, DateTime.UtcNow.Day, 08, 00, 00, DateTimeKind.Utc),
|
||
RouteAddressDetails = new List<RouteAddressDetails>
|
||
{
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 7,
|
||
TimeSpanToNextCity = TimeSpan.FromMinutes(20),
|
||
CostToNextCity = 25
|
||
},
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 8,
|
||
TimeSpanToNextCity = TimeSpan.FromMinutes(5),
|
||
WaitTimeSpan = TimeSpan.FromMinutes(15),
|
||
CostToNextCity = 20
|
||
},
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 9,
|
||
TimeSpanToNextCity = TimeSpan.FromMinutes(5),
|
||
WaitTimeSpan = TimeSpan.FromMinutes(25),
|
||
CostToNextCity = 30
|
||
},
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 10,
|
||
TimeSpanToNextCity = TimeSpan.FromMinutes(5),
|
||
WaitTimeSpan = TimeSpan.FromMinutes(20),
|
||
CostToNextCity = 35
|
||
},
|
||
new RouteAddressDetails
|
||
{
|
||
RouteAddressId = 11,
|
||
}
|
||
}
|
||
});
|
||
|
||
await dbContext.SaveChangesAsync();
|
||
|
||
await dbContext.TicketGroups.AddAsync(
|
||
new TicketGroup
|
||
{
|
||
UserId = companyUser.Id,
|
||
PurchaseDateTimeUtc = new DateTime(DateTime.UtcNow.Year, DateTime.UtcNow.Month, DateTime.UtcNow.Day, 08, 00, 00, DateTimeKind.Utc).AddDays(-1),
|
||
Tickets = new List<Ticket>
|
||
{
|
||
new Ticket
|
||
{
|
||
VehicleEnrollmentId = 1,
|
||
FirstRouteAddressId = 1,
|
||
LastRouteAddressId = 2
|
||
},
|
||
new Ticket
|
||
{
|
||
VehicleEnrollmentId = 2,
|
||
FirstRouteAddressId = 2,
|
||
LastRouteAddressId = 9
|
||
}
|
||
}
|
||
});
|
||
|
||
dbContext.Reviews.AddRangeAsync(
|
||
new Review
|
||
{
|
||
UserId = companyUser.Id,
|
||
VehicleEnrollmentId = 1,
|
||
Rating = 85,
|
||
Comment = "Good overall",
|
||
PostDateTimeUtc = new DateTime(DateTime.UtcNow.Year, DateTime.UtcNow.Month, DateTime.UtcNow.Day, 08, 00, 00, DateTimeKind.Utc).AddDays(1),
|
||
}, new Review
|
||
{
|
||
UserId = companyUser.Id,
|
||
VehicleEnrollmentId = 2,
|
||
Rating = 100,
|
||
Comment = "Amazing",
|
||
PostDateTimeUtc = new DateTime(DateTime.UtcNow.Year, DateTime.UtcNow.Month, DateTime.UtcNow.Day, 08, 00, 00, DateTimeKind.Utc).AddDays(1),
|
||
});
|
||
|
||
await dbContext.SaveChangesAsync();
|
||
}
|
||
}
|
||
} |