public async Task <IHttpActionResult> PostHero(HeroDataTransferObject hero)
{
if (!ModelState.IsValid)
{
return(BadRequest(ModelState));
}
var modelHero = hero.ToHeroModel(); //create db model from DTO
var roleConverter = new RoleStringToModelConverter(ServiceLocator.GetInstance().GetService <IDataSource>());
var roleModelSet = await roleConverter.CreateRoleList(hero.Roles); //create role models from string list in dto.
foreach (Role curr in roleModelSet)
{
if (curr.EntityState == ModelEntityState.Unchanged)
{
var dbRole = await db.Set <Role>().FindAsync(curr.RoleId);
modelHero.Roles.Add(dbRole);
}
else
{
modelHero.Roles.Add(curr);
}
}
db.Heroes.Add(modelHero);
try
{
await db.SaveChangesAsync();
}
catch (DbUpdateException)
{
if (HeroExists(modelHero.HeroId))
{
return(Conflict());
}
else
{
throw;
}
}
return(CreatedAtRoute("DefaultApi", new { id = hero.HeroId }, hero));
}
public async Task <IHttpActionResult> PutHero(int id, HeroDataTransferObject hero)
{
if (!ModelState.IsValid)
{
return(BadRequest(ModelState));
}
if (id != hero.HeroId)
{
return(BadRequest());
}
var dataSourceHero = await db.Heroes.Include(h => h.Roles).FirstOrDefaultAsync(h => h.HeroId == id);
if (dataSourceHero == null)
{
return(NotFound());
}
//using attach stuff from https://blog.maskalik.com/entity-framework/2013/12/23/entity-framework-updating-database-from-detached-objects/
//entity is already in the context
var attachedEntry = db.Entry(dataSourceHero);
attachedEntry.CurrentValues.SetValues(hero); //copy values from dto to data source object.
var roleConverter = new RoleStringToModelConverter(ServiceLocator.GetInstance().GetService <IDataSource>());
var roleModelSet = await roleConverter.CreateRoleList(hero.Roles); //create role models from string list in dto.
//get roles to add
var rolesToAdd = GetRolesToAdd(dataSourceHero, roleModelSet);
//get roles to remove
var rolesToRemove = GetRolesToRemove(dataSourceHero, roleModelSet);
//remove the roles we don't want
foreach (Role curr in rolesToRemove)
{
dataSourceHero.Roles.Remove(curr);
}
//add the roles to the datasourceHero that are new.
foreach (Role curr in rolesToAdd)
{
if (curr.EntityState == ModelEntityState.Unchanged)
{
var dbRole = await db.Set <Role>().FindAsync(curr.RoleId);
dataSourceHero.Roles.Add(dbRole);
}
else
{
dataSourceHero.Roles.Add(curr);
}
}
try
{
await db.SaveChangesAsync();
}
catch (DbUpdateConcurrencyException)
{
if (!HeroExists(id))
{
return(NotFound());
}
else
{
throw;
}
}
return(StatusCode(HttpStatusCode.NoContent));
}
