kapozade · March 12, 2019 12:49 · Mar 12, 2019
diff --git a/ScreenshotController.cs b/ScreenshotController.cs
@@ -0,0 +1,41 @@
+using System;
+using System.Drawing;
+using System.IO;
+using System.Reflection;
+using Microsoft.AspNetCore.Mvc;
+using OpenQA.Selenium;
+using OpenQA.Selenium.Chrome;
+
+namespace ScreenshotApi.Controllers
+{
+    [Route("api/[controller]")]
+    [ApiController]
+    public class ScreenshotController : ControllerBase
+    {
+        [HttpGet("screenshot")]
+        public IActionResult GetScreenshot([FromQuery]string url, [FromQuery]int width = 600, [FromQuery]int height = 600)
+        {
+            if (string.IsNullOrWhiteSpace(url))
+                return BadRequest();
+
+            try
+            {
+                var options = new ChromeOptions();
+                options.AddArgument("headless");
+                using (var driver = new ChromeDriver(Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location),
+                    options))
+                {
+                    driver.Manage().Window.Size = new Size(width, height);
+                    driver.Navigate().GoToUrl(url);
+
+                    var screenshotContent = ((ITakesScreenshot)driver).GetScreenshot().AsByteArray;
+                    return File(screenshotContent, "image/jpeg");
+                }
+            }
+            catch (Exception ex)
+            {
+                return StatusCode(500, ex.Message);
+            }
+        }
+    }
+}