|
1 | 1 | using PuppeteerPDFAPI.Misc; |
2 | 2 | using PuppeteerSharp; |
3 | 3 | using PuppeteerSharp.BrowserData; |
4 | | -using System.IO; |
5 | | -using System.Reflection; |
6 | 4 |
|
7 | 5 | namespace PuppeteerPDFAPI.Services |
8 | 6 | { |
9 | 7 | public interface IPuppeteerService |
10 | 8 | { |
11 | 9 | Task InitializeAsync(); |
| 10 | + Task<(Stream?, string? ErrorMessage)> ConvertAsync(string html); |
12 | 11 | } |
13 | 12 |
|
14 | | - public class PuppeteerService(ILogger<PuppeteerService> logger) : IPuppeteerService |
| 13 | + public class PuppeteerService(ILogger<PuppeteerService> logger, IHostApplicationLifetime appLifetime) : IPuppeteerService |
15 | 14 | { |
16 | | - const SupportedBrowser Browser = SupportedBrowser.Chrome; |
| 15 | + const SupportedBrowser TargetBrowser = SupportedBrowser.Chrome; |
| 16 | + IBrowser? Browser = default; |
17 | 17 |
|
18 | 18 | public async Task InitializeAsync() |
19 | 19 | { |
| 20 | + appLifetime.ApplicationStopping.Register(async () => |
| 21 | + { |
| 22 | + await Shutdown(); |
| 23 | + }); |
| 24 | + |
20 | 25 | string browserPath = Helpers.GetBrowserPath(); |
21 | 26 |
|
22 | | - BrowserFetcherOptions browserFetcherOptions = new() { Browser = Browser, Path = browserPath }; |
| 27 | + BrowserFetcherOptions browserFetcherOptions = new() { Browser = TargetBrowser, Path = browserPath }; |
23 | 28 | BrowserFetcher? browserFetcher = new(browserFetcherOptions); |
24 | 29 |
|
25 | 30 | InstalledBrowser? installedBrowser = browserFetcher.GetInstalledBrowsers() |
26 | | - .Where(_ => _.Browser == Browser).MaxBy(_ => _.BuildId); |
| 31 | + .Where(_ => _.Browser == TargetBrowser).MaxBy(_ => _.BuildId); |
27 | 32 |
|
28 | 33 | if (installedBrowser == null) |
29 | 34 | { |
30 | | - logger.LogWarning($"{DateTime.Now} [{nameof(PuppeteerService)}] | {Browser} is not installed! Trying to download latest version..."); |
| 35 | + logger.LogWarning($"{DateTime.Now} [{nameof(PuppeteerService)}] | {TargetBrowser} is not installed! Trying to download latest version..."); |
31 | 36 |
|
32 | 37 | installedBrowser = await browserFetcher.DownloadAsync(BrowserTag.Latest); |
33 | 38 |
|
34 | | - logger.LogInformation($"{DateTime.Now} [{nameof(PuppeteerService)}] | {Browser} finished downloading..."); |
| 39 | + logger.LogInformation($"{DateTime.Now} [{nameof(PuppeteerService)}] | {TargetBrowser} finished downloading..."); |
35 | 40 | } |
36 | 41 |
|
37 | 42 | try |
38 | 43 | { |
39 | | - logger.LogInformation($"{DateTime.Now} [{nameof(PuppeteerService)}] | Trying to startup {Browser}..."); |
40 | | - using IBrowser? browser = await Puppeteer.LaunchAsync( |
| 44 | + logger.LogInformation($"{DateTime.Now} [{nameof(PuppeteerService)}] | Trying to startup {TargetBrowser}..."); |
| 45 | + |
| 46 | + Browser = await Puppeteer.LaunchAsync( |
41 | 47 | new LaunchOptions() |
42 | 48 | { |
43 | | - ExecutablePath = Helpers.GetBrowserBinPath(Browser.ToString()) |
| 49 | + ExecutablePath = Helpers.GetBrowserBinPath(TargetBrowser.ToString()) |
44 | 50 | }); |
45 | 51 |
|
46 | | - await browser.NewPageAsync(); |
47 | | - await browser.CloseAsync(); |
48 | | - |
49 | | - logger.LogInformation($"{DateTime.Now} [{nameof(PuppeteerService)}] | {Browser} startup finished."); |
| 52 | + logger.LogInformation($"{DateTime.Now} [{nameof(PuppeteerService)}] | {TargetBrowser} startup finished."); |
50 | 53 | } |
51 | 54 | catch (Exception ex) |
52 | 55 | { |
53 | | - logger.LogInformation($"{DateTime.Now} [{nameof(PuppeteerService)}] | {Browser} startup failed with error: {ex}."); |
| 56 | + logger.LogInformation($"{DateTime.Now} [{nameof(PuppeteerService)}] | {TargetBrowser} startup failed with error: {ex}."); |
54 | 57 | throw; |
55 | 58 | } |
56 | 59 | } |
| 60 | + |
| 61 | + public async Task<(Stream?, string? ErrorMessage)> ConvertAsync(string html) |
| 62 | + { |
| 63 | + if (Browser is null) |
| 64 | + return (default, "Browser not initialized!"); |
| 65 | + |
| 66 | + try |
| 67 | + { |
| 68 | + using IPage page = await Browser.NewPageAsync(); |
| 69 | + await page.SetContentAsync(html); |
| 70 | + |
| 71 | + Stream pdfStream = await page.PdfStreamAsync(); |
| 72 | + |
| 73 | + await page.CloseAsync(); |
| 74 | + |
| 75 | + return (pdfStream, default); |
| 76 | + } |
| 77 | + catch (Exception ex) |
| 78 | + { |
| 79 | + logger.LogError($"{DateTime.Now} [{nameof(PuppeteerService)}] | {ex}"); |
| 80 | + return (null, ex.ToString()); |
| 81 | + } |
| 82 | + } |
| 83 | + |
| 84 | + public async Task Shutdown() |
| 85 | + { |
| 86 | + if (Browser is null) |
| 87 | + return; |
| 88 | + |
| 89 | + await Browser.CloseAsync(); |
| 90 | + await Browser.DisposeAsync(); |
| 91 | + |
| 92 | + } |
57 | 93 | } |
58 | 94 | } |
0 commit comments