public async Task <IActionResult> ToggleRanjivost([FromBody] ToggleRanjivostViewModel toggleRanjivostViewModel)
{
if (!ModelState.IsValid)
{
return(Fail(ModelState));
}
var pretnja = await _unitOfWork.Pretnje.Get(toggleRanjivostViewModel.PretnjaId);
if (pretnja == null)
{
return(Fail("Ne postoji pretnja"));
}
var ranjivost = await _unitOfWork.Ranjivosti.Get(toggleRanjivostViewModel.RanjivostId);
if (ranjivost == null)
{
return(Fail("Ne postoji ranjivost"));
}
// Ako dobavljac ucestvuje u Procesu / Servisu brisem ga
if (pretnja.PretnjeRanjivosti.Exists(p => p.PretnjaId == toggleRanjivostViewModel.PretnjaId &&
p.RanjivostId == toggleRanjivostViewModel.RanjivostId))
{
int i = 0;
foreach (var rajnivost in pretnja.PretnjeRanjivosti)
{
if (rajnivost.PretnjaId == toggleRanjivostViewModel.PretnjaId &&
rajnivost.RanjivostId == toggleRanjivostViewModel.RanjivostId)
{
pretnja.PretnjeRanjivosti.RemoveAt(i);
break;
}
i++;
}
}
else // dodajem doabvljaca u Proces / Servis
{
var pretnjaRanjivost = new PretnjeRanjivosti
{
PretnjaId = toggleRanjivostViewModel.PretnjaId,
RanjivostId = toggleRanjivostViewModel.RanjivostId
};
pretnja.PretnjeRanjivosti.Add(pretnjaRanjivost);
}
await _unitOfWork.Complete();
return(Success());
}
public async Task <IActionResult> DodajRanjivostPretnji([FromBody] ToggleRanjivostViewModel toggleRanjivostViewModel)
{
var pretnja = await _unitOfWork.Pretnje.Get(toggleRanjivostViewModel.PretnjaId);
if (pretnja == null)
{
return(Fail("Ne postoji pretnja"));
}
var pretnjaRanjivost = new PretnjeRanjivosti
{
PretnjaId = toggleRanjivostViewModel.PretnjaId,
RanjivostId = toggleRanjivostViewModel.RanjivostId
};
pretnja.PretnjeRanjivosti.Add(pretnjaRanjivost);
await _unitOfWork.Complete();
var ranjivosti = await _unitOfWork.PretnjeRanjivosti.RanjivostiZaPretnju(toggleRanjivostViewModel.PretnjaId);
return(Success(ranjivosti));
}