344 lines
12 KiB
C#
344 lines
12 KiB
C#
using AuthorBuddy.Web.Data;
|
|
using AuthorBuddy.Web.Models;
|
|
using LiteDB;
|
|
|
|
namespace AuthorBuddy.Web.Services;
|
|
|
|
public class SettingsService : ISettingsService
|
|
{
|
|
private const string KeyOllamaUrl = "ollama_url";
|
|
private const string KeyRagModel = "ollama_rag_model";
|
|
private const string KeyStyleModel = "ollama_style_model";
|
|
private const string KeySpellingModel = "ollama_spelling_model";
|
|
private const string KeyTheme = "theme";
|
|
private const string KeyLektorPrompt = "ollama_lektor_prompt";
|
|
private const string KeyFactCheckTemplate = "ollama_fact_check";
|
|
private const string KeySpellingPrompt = "ollama_spelling_prompt";
|
|
private const string KeyAdvocatusDiaboli = "ollama_advocatus_diaboli";
|
|
private const string KeyHookFinder = "ollama_hook_finder";
|
|
private const string KeyEli5 = "ollama_eli5";
|
|
private const string KeyLanguage = "language";
|
|
|
|
private const string KeyLastProjectId = "last_project_id";
|
|
|
|
private readonly ILiteDatabase _db;
|
|
private readonly OllamaSettings _ollamaConfig;
|
|
private int? _currentUserId;
|
|
private int? _currentProjectId;
|
|
|
|
public SettingsService(ILiteDatabase db, OllamaSettings ollamaConfig)
|
|
{
|
|
_db = db;
|
|
_ollamaConfig = ollamaConfig;
|
|
}
|
|
|
|
public int? CurrentProjectId
|
|
{
|
|
get => _currentProjectId;
|
|
set => _currentProjectId = value;
|
|
}
|
|
|
|
public string? CurrentFilePath { get; set; }
|
|
public string? PendingAssistantInput { get; set; }
|
|
public string? PendingAssistantOutput { get; set; }
|
|
|
|
public Task<int?> GetLastProjectIdAsync()
|
|
{
|
|
var col = _db.GetCollection<SettingEntity>("settings");
|
|
var setting = col.FindOne(s => s.Key == KeyLastProjectId && s.UserId == _currentUserId && s.ProjectId == null);
|
|
if (setting != null && int.TryParse(setting.Value, out var id))
|
|
return Task.FromResult<int?>(id);
|
|
return Task.FromResult<int?>(null);
|
|
}
|
|
|
|
public Task SaveLastProjectIdAsync(int projectId)
|
|
{
|
|
var col = _db.GetCollection<SettingEntity>("settings");
|
|
var existing = col.FindOne(s => s.Key == KeyLastProjectId && s.UserId == _currentUserId && s.ProjectId == null);
|
|
if (existing != null)
|
|
{
|
|
existing.Value = projectId.ToString();
|
|
col.Update(existing);
|
|
}
|
|
else
|
|
{
|
|
col.Insert(new SettingEntity
|
|
{
|
|
Key = KeyLastProjectId,
|
|
Value = projectId.ToString(),
|
|
UserId = _currentUserId,
|
|
ProjectId = null
|
|
});
|
|
}
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
private string? GetSetting(string key, int? userId, int? projectId)
|
|
{
|
|
var col = _db.GetCollection<SettingEntity>("settings");
|
|
|
|
var matches = col.Find(s =>
|
|
s.Key == key &&
|
|
(s.UserId == userId || s.UserId == null) &&
|
|
(s.ProjectId == projectId || s.ProjectId == null));
|
|
|
|
return matches
|
|
.OrderByDescending(s => s.UserId != null)
|
|
.ThenByDescending(s => s.ProjectId != null)
|
|
.FirstOrDefault()?.Value;
|
|
}
|
|
|
|
private string? Get(string key)
|
|
{
|
|
return GetSetting(key, _currentUserId, null);
|
|
}
|
|
|
|
private void Set(string key, string value)
|
|
{
|
|
var col = _db.GetCollection<SettingEntity>("settings");
|
|
|
|
var existing = col.FindOne(s =>
|
|
s.Key == key && s.UserId == _currentUserId && s.ProjectId == null);
|
|
|
|
if (existing != null)
|
|
{
|
|
existing.Value = value;
|
|
col.Update(existing);
|
|
}
|
|
else
|
|
{
|
|
col.Insert(new SettingEntity
|
|
{
|
|
Key = key,
|
|
Value = value,
|
|
UserId = _currentUserId,
|
|
ProjectId = null
|
|
});
|
|
}
|
|
}
|
|
|
|
public Task<string?> GetOllamaUrlAsync() => Task.FromResult(Get(KeyOllamaUrl));
|
|
public Task<string?> GetRagModelAsync() => Task.FromResult(Get(KeyRagModel));
|
|
public Task<string?> GetStyleModelAsync() => Task.FromResult(Get(KeyStyleModel));
|
|
public Task<string?> GetLektorSystemPromptAsync() => Task.FromResult(Get(KeyLektorPrompt));
|
|
public Task<string?> GetFactCheckTemplateAsync() => Task.FromResult(Get(KeyFactCheckTemplate));
|
|
public Task<string?> GetSpellingPromptAsync() => Task.FromResult(Get(KeySpellingPrompt));
|
|
public Task<string?> GetSpellingModelAsync() => Task.FromResult(Get(KeySpellingModel));
|
|
public Task<string?> GetAdvocatusDiaboliPromptAsync() => Task.FromResult(Get(KeyAdvocatusDiaboli));
|
|
public Task<string?> GetHookFinderPromptAsync() => Task.FromResult(Get(KeyHookFinder));
|
|
public Task<string?> GetEli5PromptAsync() => Task.FromResult(Get(KeyEli5));
|
|
public Task<string?> GetThemeAsync() => Task.FromResult(Get(KeyTheme));
|
|
public Task<string?> GetLanguageAsync() => Task.FromResult(Get(KeyLanguage));
|
|
public Task SaveLanguageAsync(string language)
|
|
{
|
|
Set(KeyLanguage, language);
|
|
_ollamaConfig.Language = language;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveOllamaUrlAsync(string url)
|
|
{
|
|
Set(KeyOllamaUrl, url);
|
|
_ollamaConfig.OllamaUrl = url;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveRagModelAsync(string model)
|
|
{
|
|
Set(KeyRagModel, model);
|
|
_ollamaConfig.EmbedModel = model;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveStyleModelAsync(string model)
|
|
{
|
|
Set(KeyStyleModel, model);
|
|
_ollamaConfig.StyleModel = model;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveLektorSystemPromptAsync(string prompt)
|
|
{
|
|
Set(KeyLektorPrompt, prompt);
|
|
_ollamaConfig.LektorSystemPrompt = prompt;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveFactCheckTemplateAsync(string prompt)
|
|
{
|
|
Set(KeyFactCheckTemplate, prompt);
|
|
_ollamaConfig.FactCheckTemplate = prompt;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveSpellingPromptAsync(string prompt)
|
|
{
|
|
Set(KeySpellingPrompt, prompt);
|
|
_ollamaConfig.SpellingPrompt = prompt;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveSpellingModelAsync(string model)
|
|
{
|
|
Set(KeySpellingModel, model);
|
|
_ollamaConfig.SpellingModel = model;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveAdvocatusDiaboliPromptAsync(string prompt)
|
|
{
|
|
Set(KeyAdvocatusDiaboli, prompt);
|
|
_ollamaConfig.AdvocatusDiaboliPrompt = prompt;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveHookFinderPromptAsync(string prompt)
|
|
{
|
|
Set(KeyHookFinder, prompt);
|
|
_ollamaConfig.HookFinderPrompt = prompt;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveEli5PromptAsync(string prompt)
|
|
{
|
|
Set(KeyEli5, prompt);
|
|
_ollamaConfig.Eli5Prompt = prompt;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task SaveThemeAsync(string theme)
|
|
{
|
|
Set(KeyTheme, theme);
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task<List<UserEntity>> GetUsersAsync()
|
|
{
|
|
var col = _db.GetCollection<UserEntity>("users");
|
|
return Task.FromResult(col.FindAll().ToList());
|
|
}
|
|
|
|
public Task<UserEntity> GetCurrentUserAsync()
|
|
{
|
|
var col = _db.GetCollection<UserEntity>("users");
|
|
var user = col.FindOne(u => u.IsCurrent);
|
|
return Task.FromResult(user ?? col.FindAll().First());
|
|
}
|
|
|
|
public Task SetCurrentUserAsync(int userId)
|
|
{
|
|
var col = _db.GetCollection<UserEntity>("users");
|
|
var users = col.FindAll().ToList();
|
|
foreach (var u in users)
|
|
{
|
|
u.IsCurrent = u.Id == userId;
|
|
col.Update(u);
|
|
}
|
|
_currentUserId = userId;
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task<UserEntity> CreateUserAsync(string name)
|
|
{
|
|
var col = _db.GetCollection<UserEntity>("users");
|
|
var user = new UserEntity { Name = name };
|
|
col.Insert(user);
|
|
return Task.FromResult(user);
|
|
}
|
|
|
|
public Task LoadProjectSettingsAsync()
|
|
{
|
|
var settings = _db.GetCollection<SettingEntity>("settings");
|
|
var entries = new (string Key, Action<string> Setter)[]
|
|
{
|
|
(KeyOllamaUrl, val => _ollamaConfig.OllamaUrl = val),
|
|
(KeyRagModel, val => _ollamaConfig.EmbedModel = val),
|
|
(KeyStyleModel, val => _ollamaConfig.StyleModel = val),
|
|
(KeySpellingModel, val => _ollamaConfig.SpellingModel = val),
|
|
(KeyTheme, val => _ollamaConfig.Theme = val),
|
|
(KeyLektorPrompt, val => _ollamaConfig.LektorSystemPrompt = val),
|
|
(KeyFactCheckTemplate, val => _ollamaConfig.FactCheckTemplate = val),
|
|
(KeySpellingPrompt, val => _ollamaConfig.SpellingPrompt = val),
|
|
(KeyAdvocatusDiaboli, val => _ollamaConfig.AdvocatusDiaboliPrompt = val),
|
|
(KeyHookFinder, val => _ollamaConfig.HookFinderPrompt = val),
|
|
(KeyEli5, val => _ollamaConfig.Eli5Prompt = val),
|
|
(KeyLanguage, val => _ollamaConfig.Language = val),
|
|
};
|
|
|
|
foreach (var entry in entries)
|
|
{
|
|
var val = Get(entry.Key);
|
|
if (val != null) entry.Setter(val);
|
|
}
|
|
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
public Task InitializeAsync(int projectId = -1)
|
|
{
|
|
var users = _db.GetCollection<UserEntity>("users");
|
|
var settings = _db.GetCollection<SettingEntity>("settings");
|
|
|
|
var userExists = users.FindAll().Any();
|
|
if (userExists)
|
|
{
|
|
var current = users.FindOne(u => u.IsCurrent);
|
|
_currentUserId = current?.Id ?? users.FindAll().First().Id;
|
|
}
|
|
|
|
if (_currentUserId == null) return Task.CompletedTask;
|
|
|
|
var lang = _ollamaConfig.Language;
|
|
|
|
var entries = new (string Key, string Default)[]
|
|
{
|
|
(KeyOllamaUrl, _ollamaConfig.OllamaUrl),
|
|
(KeyRagModel, _ollamaConfig.StyleModel),
|
|
(KeyStyleModel, _ollamaConfig.StyleModel),
|
|
(KeySpellingModel, _ollamaConfig.SpellingModel),
|
|
(KeyTheme, _ollamaConfig.Theme),
|
|
(KeyLanguage, lang),
|
|
(KeyLektorPrompt, lang == "en" ? _ollamaConfig.LektorSystemPromptEn : _ollamaConfig.LektorSystemPrompt),
|
|
(KeyFactCheckTemplate, lang == "en" ? _ollamaConfig.FactCheckTemplateEn : _ollamaConfig.FactCheckTemplate),
|
|
(KeySpellingPrompt, lang == "en" ? _ollamaConfig.SpellingPromptEn : _ollamaConfig.SpellingPrompt),
|
|
(KeyAdvocatusDiaboli, lang == "en" ? _ollamaConfig.AdvocatusDiaboliPromptEn : _ollamaConfig.AdvocatusDiaboliPrompt),
|
|
(KeyHookFinder, lang == "en" ? _ollamaConfig.HookFinderPromptEn : _ollamaConfig.HookFinderPrompt),
|
|
(KeyEli5, lang == "en" ? _ollamaConfig.Eli5PromptEn : _ollamaConfig.Eli5Prompt),
|
|
};
|
|
|
|
foreach (var entry in entries)
|
|
{
|
|
var existing = settings.FindOne(s =>
|
|
s.Key == entry.Key && s.UserId == _currentUserId && s.ProjectId == projectId);
|
|
|
|
if (existing != null)
|
|
{
|
|
var val = existing.Value;
|
|
if (entry.Key == KeyOllamaUrl) _ollamaConfig.OllamaUrl = val;
|
|
else if (entry.Key == KeyRagModel) _ollamaConfig.EmbedModel = val;
|
|
else if (entry.Key == KeyStyleModel) _ollamaConfig.StyleModel = val;
|
|
else if (entry.Key == KeySpellingModel) _ollamaConfig.SpellingModel = val;
|
|
else if (entry.Key == KeyTheme) _ollamaConfig.Theme = val;
|
|
else if (entry.Key == KeyLanguage) _ollamaConfig.Language = val;
|
|
else if (entry.Key == KeyLektorPrompt) _ollamaConfig.LektorSystemPrompt = val;
|
|
else if (entry.Key == KeyFactCheckTemplate) _ollamaConfig.FactCheckTemplate = val;
|
|
else if (entry.Key == KeySpellingPrompt) _ollamaConfig.SpellingPrompt = val;
|
|
else if (entry.Key == KeyAdvocatusDiaboli) _ollamaConfig.AdvocatusDiaboliPrompt = val;
|
|
else if (entry.Key == KeyHookFinder) _ollamaConfig.HookFinderPrompt = val;
|
|
else if (entry.Key == KeyEli5) _ollamaConfig.Eli5Prompt = val;
|
|
}
|
|
else
|
|
{
|
|
settings.Insert(new SettingEntity
|
|
{
|
|
Key = entry.Key,
|
|
Value = entry.Default,
|
|
UserId = _currentUserId,
|
|
ProjectId = null
|
|
});
|
|
}
|
|
}
|
|
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|