internal static AddPassengerViewModel AddConnectionInPTD(TrinityContext context, AddPassengerViewModel drive, string ssn)
{
var person = context.Persons
.Where(p => p.SSN == ssn)
.FirstOrDefault();
context.PTD.Add(new PTD()
{
PID = person.ID, DID = drive.ID
});
context.SaveChanges();
var update = context.Drives.
Where(s => s.ID == drive.ID).First();
update.Passengers--;
context.SaveChanges();
AddPassengerViewModel updatedDrive = new AddPassengerViewModel()
{
ID = update.ID, Date = update.Date, StartingPoint = update.StartingPoint, Destination = update.Destination, Driver = update.Driver, Passengers = update.Passengers
};
return(updatedDrive);
}
internal static void AddPerson(TrinityContext context, JoinViewModel person)
{
context.Persons.Add(new Persons()
{
FirstName = person.FirstName, LastName = person.LastName, SSN = person.SSN
});
context.SaveChanges();
}
internal static void RemoveTraveller(MyDrivesViewModel drive, TrinityContext context)
{
var person = context.Persons
.Where(p => p.SSN == drive.SSN)
.First();
var travellers = context.PTD
.Where(t => t.DID == drive.ID)
.ToList();
if (travellers[0].PID == person.ID)
{
foreach (var item in travellers)
{
context.Remove(item);
context.SaveChanges();
}
var _drive = context.Drives
.Where(d => d.ID == drive.ID)
.FirstOrDefault();
context.Remove(_drive);
context.SaveChanges();
}
else
{
var updatePassengers = context.Drives
.Where(p => p.ID == drive.ID)
.FirstOrDefault();
updatePassengers.Passengers++;
context.SaveChanges();
var traveller = travellers
.Where(p => p.PID == person.ID)
.FirstOrDefault();
context.Remove(traveller);
context.SaveChanges();
}
}
internal static int AddDrive(TrinityContext context, AddDriveViewModel drive)
{
Drives newDrive = new Drives()
{
StartingPoint = drive.StartingPoint, Destination = drive.Destination, Date = drive.Date, Driver = true, Passengers = drive.Passengers
};
context.Drives.Add(newDrive);
context.SaveChanges();
int did = newDrive.ID;
return(did);
}
internal static bool SSNInDB(TrinityContext context, AddDriveViewModel drive)
{
var driveSSN = context.Persons
.Where(p => p.SSN == drive.SSN)
.FirstOrDefault();
if (driveSSN != null)
{
int did = AddDrive(context, drive);
var person = context.Persons.Where(p => p.SSN == drive.SSN).FirstOrDefault();
context.PTD.Add(new PTD()
{
PID = person.ID, DID = did
});
context.SaveChanges();
return(true);
}
else
{
return(false);
}
}
public void SaveDbChanges()
{
Db.SaveChanges();
}
