public async Task <Unit> Handle(DeletePromptCommand request, CancellationToken cancellationToken = default)
{
if (!_currentUserService.TryGetCurrentUser(out GetUserViewModel? user))
{
throw new DeletePromptUserUnauthorizedException();
}
Prompt?prompt = await _dbContext.Prompts.Include(e => e.Children)
.FirstOrDefaultAsync(e => e.Id == request.Id);
if (prompt == null)
{
throw new DeletePromptDoesNotExistException();
}
if (prompt.OwnerId != user !.Id && (user.Role & RoleEnum.Delete) == 0)
{
throw new DeletePromptUserUnauthorizedException();
}
_dbContext.Prompts.Remove(prompt);
await RemoveAllChildren(prompt.Children);
await _dbContext.SaveChangesAsync(cancellationToken);
return(Unit.Value);
}
public async Task <Unit> Handle(CreateReportCommand request, CancellationToken cancellationToken)
{
_dbContext.Reports.Add(new Report
{
DateCreated = DateTime.UtcNow,
ExtraDetails = request.ExtraDetails,
PromptId = request.PromptId,
ReportReason = request.ReportReason
});
await _dbContext.SaveChangesAsync(cancellationToken);
return(Unit.Value);
}
public async Task <Unit> Handle(ClearReportCommand request, CancellationToken cancellationToken = default)
{
Report?report = await _dbContext.Reports.FirstOrDefaultAsync(report => report.Id == request.Id);
if (report == null)
{
throw new ClearReportNotFoundException();
}
report.Cleared = true;
_dbContext.Reports.Update(report);
await _dbContext.SaveChangesAsync(cancellationToken);
return(Unit.Value);
}
public async Task <int> Handle(CreateTransientUserCommand request, CancellationToken cancellationToken)
{
var username = Guid.NewGuid().ToString();
while (!await UsernameIsUnique(username))
{
username = Guid.NewGuid().ToString();
}
var user = new User {
Username = username, DateCreated = DateTime.UtcNow
};
_dbContext.Users.Add(user);
await _dbContext.SaveChangesAsync(cancellationToken);
return(user.Id);
}
public async Task <int> Handle(CreateUserCommand request, CancellationToken cancellationToken)
{
if (await _dbContext.Users.FirstOrDefaultAsync(e => e.Username == request.Username) != null)
{
throw new UsernameNotUniqueException();
}
var user = new User
{
DateCreated = DateTime.UtcNow,
Password = BCrypt.Net.BCrypt.EnhancedHashPassword(request.Password),
Username = request.Username
};
_dbContext.Users.Add(user);
await _dbContext.SaveChangesAsync(cancellationToken);
return(user.Id);
}
public async Task <Unit> Handle(UpdateUserCommand request, CancellationToken cancellationToken)
{
User?user = await _dbContext.Users.FindAsync(request.Id);
if (user == null)
{
throw new UpdateUserNotFoundException();
}
if (!string.IsNullOrWhiteSpace(request.Username))
{
if (await _dbContext
.Users
.FirstOrDefaultAsync(e =>
EF.Functions.ILike(e.Username, NpgsqlHelper.SafeIlike(request.Username),
NpgsqlHelper.EscapeChar) &&
e.Id != request.Id) != null)
{
throw new UsernameNotUniqueException();
}
user.Username = request.Username;
}
if (!string.IsNullOrWhiteSpace(request.Password))
{
user.Password = BCrypt.Net.BCrypt.EnhancedHashPassword(request.Password);
}
user.DateEdited = DateTime.UtcNow;
_dbContext.Users.Update(user);
await _dbContext.SaveChangesAsync(cancellationToken);
return(Unit.Value);
}
public async Task <int> Handle(CreatePromptCommand request, CancellationToken cancellationToken = default)
{
if (request.ParentId.HasValue)
{
if (!_currentUserService.TryGetCurrentUser(out GetUserViewModel? user))
{
throw new CreatePromptUnauthorizedParentException();
}
Prompt?parent = await _dbContext.Prompts.FindAsync(request.ParentId);
if (parent.OwnerId != user !.Id)
{
throw new CreatePromptUnauthorizedParentException();
}
}
var isDraft = !request.ParentId.HasValue && request.SaveDraft;
var prompt = new Prompt
{
AuthorsNote = request.AuthorsNote?.Replace("\r\n", "\n"),
DateCreated = DateTime.UtcNow,
DateEdited = null,
Memory = request.Memory?.Replace("\r\n", "\n"),
Nsfw = request.Nsfw,
PromptContent = request.PromptContent.Replace("\r\n", "\n"),
Quests = request.Quests?.Replace("\r\n", "\n"),
Title = request.Title.Replace("\r\n", "\n"),
Description = request.Description?.Replace("\r\n", "\n"),
OwnerId = request.OwnerId,
Upvote = 0,
Views = 0,
IsDraft = isDraft,
PublishDate = isDraft ? null : (DateTime?)DateTime.UtcNow,
ParentId = request.ParentId,
ScriptZip = request.ScriptZip,
NovelAiScenario = string.IsNullOrWhiteSpace(request.NovelAiScenario) ? null : request.NovelAiScenario,
HoloAiScenario = string.IsNullOrWhiteSpace(request.HoloAiScenario) ? null : request.HoloAiScenario
};
foreach (var worldInfo in request.WorldInfos)
{
if (string.IsNullOrWhiteSpace(worldInfo.Entry) || string.IsNullOrWhiteSpace(worldInfo.Keys))
{
continue;
}
prompt.WorldInfos.Add(new WorldInfo
{
DateCreated = DateTime.UtcNow,
Entry = worldInfo.Entry.Replace("\r\n", "\n"),
Keys = worldInfo.Keys.Replace("\r\n", "\n"),
Prompt = prompt
});
}
var promptTags = request.PromptTags.Split(',').Select(p => p.Trim().ToLower()).Distinct();
foreach (var promptTag in promptTags)
{
if (string.IsNullOrWhiteSpace(promptTag))
{
continue;
}
if (string.Equals(promptTag, "nsfw", StringComparison.OrdinalIgnoreCase))
{
prompt.Nsfw = true;
continue;
}
Tag?tag = await _dbContext.Tags.FirstOrDefaultAsync(e =>
EF.Functions.ILike(e.Name, NpgsqlHelper.SafeIlike(promptTag), NpgsqlHelper.EscapeChar), cancellationToken : cancellationToken);
if (tag == null)
{
prompt.PromptTags.Add(new PromptTag {
Prompt = prompt, Tag = new Tag {
Name = promptTag
}
});
}
else
{
prompt.PromptTags.Add(new PromptTag {
Prompt = prompt, Tag = tag
});
}
}
_dbContext.Prompts.Add(prompt);
await _dbContext.SaveChangesAsync(cancellationToken);
return(prompt.Id);
}
public async Task <Unit> Handle(UpdatePromptCommand request, CancellationToken cancellationToken = default)
{
if (!_currentUserService.TryGetCurrentUser(out GetUserViewModel? user))
{
return(Unit.Value);
}
Prompt?prompt = await _dbContext.Prompts
.Include(e => e.PromptTags)
.Include(e => e.WorldInfos)
.FirstOrDefaultAsync(e => e.Id == request.Id, cancellationToken);
var isOwner = user !.Id == prompt.OwnerId;
var canEditField = isOwner ? isOwner : (user.Role & RoleEnum.FieldEdit) != 0;
var canEditTags = isOwner ? isOwner : (user.Role & RoleEnum.TagEdit) != 0;
if (canEditField)
{
var isDraft = !prompt.ParentId.HasValue &&
(isOwner
? request.SaveDraft
: prompt.IsDraft);
prompt.AuthorsNote = request.AuthorsNote?.Replace("\r\n", "\n");
prompt.DateEdited = DateTime.UtcNow;
prompt.Memory = request.Memory?.Replace("\r\n", "\n");
prompt.Nsfw = request.Nsfw;
prompt.PromptContent = request.PromptContent.Replace("\r\n", "\n");
prompt.Quests = request.Quests?.Replace("\r\n", "\n");
prompt.Title = request.Title.Replace("\r\n", "\n");
prompt.Description = request.Description?.Replace("\r\n", "\n");
prompt.WorldInfos = new List <WorldInfo>();
prompt.IsDraft = isDraft;
prompt.PublishDate ??= isDraft ? null : (DateTime?)DateTime.UtcNow;
prompt.ScriptZip = isOwner
? request.ScriptZip ?? prompt.ScriptZip
: prompt.ScriptZip;
prompt.NovelAiScenario = string.IsNullOrWhiteSpace(request.NovelAiScenario) ? null : request.NovelAiScenario;
prompt.HoloAiScenario = string.IsNullOrWhiteSpace(request.HoloAiScenario) ? null : request.HoloAiScenario;
foreach (var worldInfo in request.WorldInfos)
{
if (string.IsNullOrWhiteSpace(worldInfo.Entry) || string.IsNullOrWhiteSpace(worldInfo.Keys))
{
continue;
}
prompt.WorldInfos.Add(new WorldInfo
{
DateCreated = DateTime.UtcNow,
Entry = worldInfo.Entry.Replace("\r\n", "\n"),
Keys = worldInfo.Keys.Replace("\r\n", "\n"),
Prompt = prompt
});
}
}
if (canEditTags)
{
prompt.PromptTags = new List <PromptTag>();
var promptTags =
request.PromptTags.Split(',').Select(p => p.Trim().ToLower()).Distinct();
foreach (var promptTag in promptTags)
{
if (string.IsNullOrWhiteSpace(promptTag))
{
continue;
}
if (string.Equals(promptTag, "nsfw", StringComparison.OrdinalIgnoreCase))
{
prompt.Nsfw = true;
continue;
}
Tag?tag = await _dbContext.Tags.FirstOrDefaultAsync(e =>
EF.Functions.ILike(e.Name, NpgsqlHelper.SafeIlike(promptTag), NpgsqlHelper.EscapeChar));
if (tag == null)
{
prompt.PromptTags.Add(new PromptTag {
Prompt = prompt, Tag = new Tag {
Name = promptTag
}
});
}
else
{
prompt.PromptTags.Add(new PromptTag {
Prompt = prompt, Tag = tag
});
}
}
}
_dbContext.Prompts.Update(prompt);
await _dbContext.SaveChangesAsync(cancellationToken);
return(Unit.Value);
}
