github/Ombi
1
0
Fork 0
mirror of https://github.com/Ombi-app/Ombi.git synced 2025-07-14 09:12:57 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added four new endpoints to change the availability for TV shows and Movies #1601

Browse source
This commit is contained in:
Jamie.Rees 2017-10-23 12:09:06 +01:00
parent b4d01386a6
commit f5ffa78304
14 changed files with 207 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

59
src/Ombi.Core/Engine/TvRequestEngine.cs
View file

@ -111,7 +111,7 @@ namespace Ombi.Core.Engine
// Looks like we have removed them all! They were all duplicates...
return new RequestEngineResult
{
RequestAdded = false,
Result = false,
ErrorMessage = "This has already been requested"
};
}
@ -201,7 +201,7 @@ namespace Ombi.Core.Engine
}
return new RequestEngineResult
{
RequestAdded = true
Result = true
};
}
@ -220,19 +220,10 @@ namespace Ombi.Core.Engine
NotificationHelper.Notify(request, NotificationType.RequestDeclined);
return new RequestEngineResult
{
RequestAdded = true
Result = true
};
}
public async Task<ChildRequests> ChangeAvailability(ChildRequests request)
{
if (request.Available)
{
NotificationHelper.Notify(request, NotificationType.RequestAvailable);
}
return await UpdateChildRequest(request);
}
public async Task<ChildRequests> UpdateChildRequest(ChildRequests request)
{
await Audit.Record(AuditType.Updated, AuditArea.TvRequest, $"Updated Request {request.Title}", Username);
@ -272,6 +263,46 @@ namespace Ombi.Core.Engine
return await TvRepository.GetChild().AnyAsync(x => x.RequestedUserId == userId);
}
public async Task<RequestEngineResult> MarkUnavailable(int modelId)
{
var request = await TvRepository.GetChild().FirstOrDefaultAsync(x => x.Id == modelId);
if (request == null)
{
return new RequestEngineResult
{
ErrorMessage = "Child Request does not exist"
};
}
request.Available = false;
await TvRepository.UpdateChild(request);
NotificationHelper.Notify(request, NotificationType.RequestAvailable);
return new RequestEngineResult
{
Result = true,
Message = "Request is now unavailable",
};
}
public async Task<RequestEngineResult> MarkAvailable(int modelId)
{
var request = await TvRepository.GetChild().FirstOrDefaultAsync(x => x.Id == modelId);
if (request == null)
{
return new RequestEngineResult
{
ErrorMessage = "Child Request does not exist"
};
}
request.Available = true;
await TvRepository.UpdateChild(request);
NotificationHelper.Notify(request, NotificationType.RequestAvailable);
return new RequestEngineResult
{
Result = true,
Message = "Request is now available",
};
}
private async Task<RequestEngineResult> AddExistingRequest(ChildRequests newRequest, TvRequests existingRequest)
{
// Add the child
@ -339,7 +370,7 @@ namespace Ombi.Core.Engine
var result = await TvSender.Send(model);
if (result.Success)
{
return new RequestEngineResult {RequestAdded = true};
return new RequestEngineResult {Result = true};
}
return new RequestEngineResult
{
@ -347,7 +378,7 @@ namespace Ombi.Core.Engine
};
}
return new RequestEngineResult {RequestAdded = true};
return new RequestEngineResult {Result = true};
}
}
}