internal InlineKeyboardMarkup setMark(UserVulnerabilityDB vulnerabilityDB)
{
var numberToInput = (vulnerabilityDB.threats == null) ? "threats" : (vulnerabilityDB.techDamage == null) ? "techDamage" :
(vulnerabilityDB.bizDamage == null) ? "bizDamage" : "";
numberToInput += "," + vulnerabilityDB.vulnerability.id;
return(new InlineKeyboardMarkup(new[]
{
new []
{
InlineKeyboardButton.WithCallbackData("1", "1," + numberToInput),
InlineKeyboardButton.WithCallbackData("2", "2," + numberToInput),
InlineKeyboardButton.WithCallbackData("3", "3," + numberToInput),
InlineKeyboardButton.WithCallbackData("4", "4," + numberToInput),
InlineKeyboardButton.WithCallbackData("5", "5," + numberToInput),
},
new []
{
InlineKeyboardButton.WithCallbackData("6", "6," + numberToInput),
InlineKeyboardButton.WithCallbackData("7", "7," + numberToInput),
InlineKeyboardButton.WithCallbackData("8", "8," + numberToInput),
InlineKeyboardButton.WithCallbackData("9", "9," + numberToInput),
InlineKeyboardButton.WithCallbackData("10", "10," + numberToInput),
},
}));;
}
internal UserVulnerabilityDB FindTask(long userId)
{
var emptyVuln = GetAllEmptyVulnerabilities();
var userVulnerabilityDBsID = crudUserVulnerabilitesDB.GetObjects().Select(x => x.vulnerability_id).ToList();
UserVulnerabilityDB newUserVulnerabilityDB = null;
foreach (var proccedVuln in emptyVuln)//emptyVuln.Where(inv => !userVulnerabilityDBsID.Contains(inv.id)).ToList())
{
newUserVulnerabilityDB = crudUserVulnerabilitesDB.ReadByUserVuln(userId, proccedVuln.id);
if (newUserVulnerabilityDB == null)
{
newUserVulnerabilityDB = new UserVulnerabilityDB(Convert.ToInt32(userId), proccedVuln.id);
crudUserVulnerabilitesDB.Create(newUserVulnerabilityDB);
break;
}
else if (newUserVulnerabilityDB.techDamage == null || newUserVulnerabilityDB.bizDamage == null ||
newUserVulnerabilityDB.threats == null)
{
break;
}
else
{
newUserVulnerabilityDB = null;
}
}
if (newUserVulnerabilityDB == null)
{
return(null);
}
newUserVulnerabilityDB = crudUserVulnerabilitesDB.ReadByUserVuln(newUserVulnerabilityDB.userid, newUserVulnerabilityDB.vulnerability_id);
return(newUserVulnerabilityDB);
}
internal string textAboutVuln(UserVulnerabilityDB vulnerabilityDB)
{
var markFor = (vulnerabilityDB.threats == null) ? "threats" : (vulnerabilityDB.techDamage == null) ? "tech damage" :
(vulnerabilityDB.bizDamage == null) ? "biznes damage" : "";
return("Vulnerability name: " + vulnerabilityDB.vulnerability.name + "\n" +
"Description: " + vulnerabilityDB.vulnerability.description + "\n" +
"Make an assessment of " + markFor);
}
public void Delet(UserVulnerabilityDB obj)
{
context = new Context();
obj = context.userVulnerabilityDBs.Find(obj.id);
if (obj == null)
{
return;
}
context.userVulnerabilityDBs.Remove(obj);
context.SaveChanges();
}
internal UserVulnerabilityDB UserCreateMark(int mark, string v, long vulnId, long userId)
{
var userVulnDB = crudUserVulnerabilitesDB.ReadByUserVuln(userId, vulnId);
bool ifNew = false;
if (userVulnDB == null)
{
ifNew = true;
userVulnDB = new UserVulnerabilityDB(userId, vulnId);
}
switch (v)
{
case "threats":
if (userVulnDB.threats != null)
{
return(null);
}
userVulnDB.threats = mark;
break;
case "techDamage":
if (userVulnDB.techDamage != null)
{
return(null);
}
userVulnDB.techDamage = mark;
break;
case "bizDamage":
if (userVulnDB.bizDamage != null)
{
return(null);
}
userVulnDB.bizDamage = mark;
break;
default:
return(null);
}
if (ifNew)
{
crudUserVulnerabilitesDB.Create(userVulnDB);
}
else
{
crudUserVulnerabilitesDB.Update(userVulnDB);
}
if (userVulnDB != null)
{
userVulnDB.vulnerability = crudVulnerabilityDB.Read(userVulnDB.vulnerability_id);
}
return(userVulnDB);
}
public void Update(UserVulnerabilityDB obj)
{
context = new Context();
var entity = context.userVulnerabilityDBs.Find(obj.id);
if (entity == null)
{
return;
}
context.Entry(entity).CurrentValues.SetValues(obj);
context.SaveChanges();
return;
}
private bool FindNewWork(long userId)
{
if (!modelVulnerabilityDB.FindEmptyVuln())
{
SendOneMessage(userId, view.nothingToDo, null);
}
else
{
UserVulnerabilityDB vuln = modelVulnerabilityDB.FindTask(userId);
if (vuln == null)
{
SendOneMessage(userId, view.nothingToDo, null);
}
else
{
SendOneInlineMessage(userId, view.textAboutVuln(vuln), view.setMark(vuln));
return(false);
}
}
return(true);
}
public void Create(UserVulnerabilityDB userVulnerabilityDB)
{
context = new Context();
context.userVulnerabilityDBs.Add(userVulnerabilityDB);
context.SaveChanges();
}
