ProviderClient.WebResearch.cs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

using Lilith.Core.WebResearch;

namespace Lilith.Core.Providers;

public abstract partial class ProviderClient
{
    private bool? _seleniumHeadless;
    public event Action<ResearchBrowserFrame>? ResearchBrowserFrameCaptured;
    public bool SeleniumHeadless => _seleniumHeadless ?? BrowserResearchOptions.Load().Headless;

    public bool SetSeleniumHeadless(bool? enabled)
    {
        using var operation = EnterOperation();
        _seleniumHeadless = enabled ?? !SeleniumHeadless;
        return _seleniumHeadless.Value;
    }

    internal IResearchBrowser CreateResearchBrowser()
    {
        if (ResearchBrowser is not null) return ResearchBrowser;
        var options = BrowserResearchOptions.Load();
        return new SeleniumResearchBrowser(options with { Headless = _seleniumHeadless ?? options.Headless },
            reportLifecycle: message => Logger.WriteInfo("Selenium", message),
            captureFrame: ResearchBrowserFrameCaptured is null ? null : frame => ResearchBrowserFrameCaptured?.Invoke(frame));
    }
}