Implement IOcrClient

OcrClient/Models/OcrResponse.cs

@@ -2,4 +2,27 @@ namespace OcrClient.Models;
 
 public class OcrResponse
 {
+	public Result? Result { get; set; }
+}
+public class Result
+{
+	public TextAnnotation? TextAnnotation { get; set; }
+}
+public class TextAnnotation
+{
+	public List<Table>? Tables { get; set; }
+}
+
+public class Table
+{
+	public string? RowCount { get; set; }
+	public string? ColumnCount { get; set; }
+	public List<Cell>? Cells { get; set; }
+}
+
+public class Cell
+{
+	public string? RowIndex { get; set; }
+	public string? ColumnIndex { get; set; }
+	public string? Text { get; set; }
 }

OcrClient/OcrClient.csproj

@@ -8,6 +8,7 @@
   </PropertyGroup>
 
   <ItemGroup>
+    <PackageReference Include="Newtonsoft.Json" Version="13.0.3" />
     <PackageReference Include="System.Net.Http" Version="4.3.4" />
   </ItemGroup>
 

OcrClient/Services/IOcrClient.cs

@@ -4,5 +4,5 @@ namespace OcrClient.Services;
 
 public interface IOcrClient
 {
-	public Task<OcrResponse> ProcessImage(string base64Image);
+	public Task<IEnumerable<object[,]>> ProcessImage(string base64Image, string xFolderId, string apiKey);
 }

OcrClient/Services/OcrClient.cs

@@ -1,19 +0,0 @@
-using OcrClient.Models;
-using System.Net.Http;
-
-namespace OcrClient.Services;
-
-public class YandexOcrClient : IOcrClient
-{
-	private readonly HttpClient _httpClient;
-	
-	public YandexOcrClient(HttpClient httpClient)
-	{
-		_httpClient = httpClient;
-	}
-
-	public Task<OcrResponse> ProcessImage(string base64Image)
-	{
-		throw new NotImplementedException();
-	}
-}

OcrClient/Services/YandexOcrClient.cs

@@ -0,0 +1,70 @@
+using System.Net.Http;
+using System.Net.Http.Headers;
+using System.Text;
+using Newtonsoft.Json;
+using OcrClient.Models;
+
+namespace OcrClient.Services;
+
+public class YandexOcrClient : IOcrClient
+{
+	private readonly HttpClient _httpClient;
+	public YandexOcrClient(HttpClient httpClient)
+	{
+		_httpClient = httpClient;
+		_httpClient.BaseAddress = new Uri("https://ocr.api.cloud.yandex.net/ocr/v1/");
+	}
+
+	public async Task<IEnumerable<object[,]>> ProcessImage(string base64Image, string xFolderId, string apiKey)
+	{
+		using StringContent jsonContent = new(
+			JsonConvert.SerializeObject(new
+			{
+				mimeType = "PNG",
+				languageCodes = new string[] { "ru", "en" },
+				model = "table",
+				content = base64Image
+			}),
+			Encoding.UTF8,
+			"application/json");
+		_httpClient.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Api-Key", apiKey);
+		_httpClient.DefaultRequestHeaders.Add("x-folder-id", xFolderId);
+		_httpClient.DefaultRequestHeaders.Add("x-data-logging-enable", "true");
+
+		using HttpResponseMessage response = await _httpClient.PostAsync("recognizeText", jsonContent);
+		response.EnsureSuccessStatusCode();
+
+		string jsonResponse = await response.Content.ReadAsStringAsync();
+		OcrResponse? deserialized = JsonConvert.DeserializeObject<OcrResponse>(jsonResponse);
+
+		if (deserialized != null)
+		{
+			var tables = deserialized?.Result?.TextAnnotation?.Tables ?? Enumerable.Empty<Table>();
+			if (tables.Any())
+			{
+				List<object[,]> result = new();
+				foreach (var table in tables)
+				{
+					if (table.Cells == null || table.Cells.Count == 0) 
+					{
+						continue;
+					}
+					int columnCount = int.Parse(table.ColumnCount);
+					int rowCount = int.Parse(table.RowCount);
+					object[,] cells = new object[rowCount, columnCount];
+					
+					foreach (Cell cell in table.Cells)
+					{
+						int rowIndex = int.Parse(cell.RowIndex);
+						int columnIndex = int.Parse(cell.ColumnIndex);
+						cells[rowIndex, columnIndex] = double.TryParse(cell.Text, out double v) ?
+							v : cell.Text ?? string.Empty;
+					}
+					result.Add(cells);
+				}
+				return result;
+			}
+		}
+		return null;
+	}
+}

RhSolutions.AddIn/Controllers/RibbonController.cs

@@ -33,7 +33,7 @@ public class RibbonController : ExcelRibbon
 				<button id='dxfexport' getEnabled='GetDxfEnabled' label='DXF' size='large' image='DXF' onAction='OnToolPressed'/>
 			</group>
 			<group id='importTab' label='OCR'>
-				<button id='ocr' label='Распознать таблицу' size='large' imageMso='TableInsert' onAction='OnToolPressed'/>
+				<button id='ocr' getEnabled='GetOcrEnabled' label='Распознать таблицу' size='large' imageMso='TableInsert' onAction='OnToolPressed'/>
 			</group>
 			<group id='settings' getLabel='GetVersionLabel'>
 				<button id='setPriceList' getLabel='GetPriceListPathLabel' size='large' image='RhSolutions' onAction='OnSetPricePressed'/>
@@ -103,6 +103,13 @@ public class RibbonController : ExcelRibbon
 		}
 	}
 	
+	public bool GetOcrEnabled(IRibbonControl control) 
+	{
+		return RhSolutionsAddIn.Excel.ActiveWorkbook != null &&
+			!string.IsNullOrEmpty(RhSolutionsAddIn.Configuration["x-folder-id"]) &&
+			!string.IsNullOrEmpty(RhSolutionsAddIn.Configuration["apiKey"]);
+	}
+
 	public string GetVersionLabel(IRibbonControl control)
 	{
 		string version = Assembly.GetExecutingAssembly().GetName().Version.ToString();

RhSolutions.AddIn/Properties/AssemblyInfo.cs

@@ -1,5 +1,6 @@
 using System.Reflection;
 using System.Runtime.InteropServices;
+using Microsoft.Extensions.Configuration.UserSecrets;
 
 // General Information about an assembly is controlled through the following
 // set of attributes. Change these attribute values to modify the information
@@ -33,3 +34,4 @@ using System.Runtime.InteropServices;
 // [assembly: AssemblyVersion("1.0.*")]
 [assembly: AssemblyVersion("1.9.5.1")]
 [assembly: AssemblyFileVersion("1.9.5.1")]
+[assembly: UserSecretsId("d4bb704e-14a5-421f-8f2d-0ffb66d090a2")]

RhSolutions.AddIn/RhSolutions.AddIn.csproj

@@ -2,12 +2,11 @@
 	<PropertyGroup>
 		<TargetFrameworks>net472;net6.0-windows</TargetFrameworks>
 		<LangVersion>10</LangVersion>
-		<OutputType>Library</OutputType>
+		<ProduceReferenceAssembly>false</ProduceReferenceAssembly>
-		<RootNamespace>RhSolutions</RootNamespace>
-		<AssemblyName>RhSolutions.AddIn</AssemblyName>
 		<GenerateAssemblyInfo>false</GenerateAssemblyInfo>
+		<EnableDynamicLoading>true</EnableDynamicLoading>
 		<UseWindowsForms>true</UseWindowsForms>
-		<ImportWindowsDesktopTargets>true</ImportWindowsDesktopTargets>
+		<UserSecretsId>d4bb704e-14a5-421f-8f2d-0ffb66d090a2</UserSecretsId>
 	</PropertyGroup>
 	<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|AnyCPU'">
 		<NoWarn>CA1416</NoWarn>
@@ -21,6 +20,9 @@
 		<PackageReference Include="Microsoft.Bcl.HashCode" Version="1.1.1" />
 		<PackageReference Include="Microsoft.CSharp" Version="4.7.0" />
 		<PackageReference Include="Microsoft.Extensions.Caching.Memory" Version="8.0.1" />
+		<PackageReference Include="Microsoft.Extensions.Configuration" Version="8.0.0" />
+		<PackageReference Include="Microsoft.Extensions.Configuration.UserSecrets" Version="8.0.1" />
+		<PackageReference Include="Microsoft.Extensions.FileProviders.Embedded" Version="8.0.10" />
 		<PackageReference Include="Microsoft.Extensions.Http" Version="8.0.1" />
 		<PackageReference Include="netDxf" Version="2022.11.2" />
 		<PackageReference Include="Newtonsoft.Json" Version="13.0.3" />

RhSolutions.AddIn/Services/AddInConfiguration.cs

@@ -1,21 +1,36 @@
-using Microsoft.Win32;
+using Microsoft.Extensions.Configuration;
+using Microsoft.Win32;
 using System.IO;
 
 namespace RhSolutions.Services;
 
 public class AddInConfiguration : IAddInConfiguration
 {
+	private IConfiguration _configuration;
 	private RegistryKey _rootKey;
 	private string _priceListPath;
-    private Dictionary<string, string> _priceListHeaders;
+	private Dictionary<string, string> settingsValues;
+
+	public string this[string key]
+	{
+		get => _configuration[key];
+	}
 
 	public event IAddInConfiguration.SettingsHandler OnSettingsChange;
 
 	public AddInConfiguration()
 	{
+		// EmbeddedFileProvider embeddedProvider = new (typeof(RhSolutionsAddIn).Assembly);
+		// using Stream stream = embeddedProvider.GetFileInfo("appsettings.json").CreateReadStream();
+
+		_configuration = new ConfigurationBuilder()
+			.AddUserSecrets<RhSolutionsAddIn>()
+			// .AddJsonStream(stream)
+			.Build();
+
 		_rootKey = Registry.CurrentUser.CreateSubKey(@"SOFTWARE\RhSolutions\RhSolutions-AddIn");
 		_priceListPath = (string)_rootKey.GetValue("PriceListPath");
-        _priceListHeaders = new()
+		settingsValues = new()
 		{
 			["Amount"] = "Кол-во",
 			["OldSku"] = "Прежний материал",
@@ -27,7 +42,7 @@ public class AddInConfiguration : IAddInConfiguration
 	}
 
 	public string GetPriceListFileName() => Path.GetFileName(_priceListPath);
-    public Dictionary<string, string> GetPriceListHeaders() => _priceListHeaders;
+	public Dictionary<string, string> GetPriceListHeaders() => settingsValues;
 	public string GetPriceListPath() => _priceListPath;
 
 	public void SaveSettings()

RhSolutions.AddIn/Services/IAddInConfiguration.cs

@@ -1,7 +1,10 @@
-namespace RhSolutions.Services;
+using Microsoft.Extensions.Configuration;
+
+namespace RhSolutions.Services;
 
 public interface IAddInConfiguration
 {
+	string this[string key] { get; }
 	public string GetPriceListPath();
 	public void SetPriceListPath(string value);
 	public string GetPriceListFileName();

RhSolutions.AddIn/Tools/EventsUtil.cs

@@ -29,6 +29,7 @@ internal static class EventsUtil
 		RibbonController.RefreshControl("guess");
 		RibbonController.RefreshControl("fillsleeves");
 		RibbonController.RefreshControl("fillcouplings");
+		RibbonController.RefreshControl("ocr");
 	}
 
 	private static void RefreshExportButton(object sh, Range target)

RhSolutions.AddIn/Tools/OcrTool.cs

@@ -1,14 +1,19 @@
 using System.Threading.Tasks;
 using SnippingTool;
 using OcrClient.Services;
+using System.Windows.Forms;
+using Application = Microsoft.Office.Interop.Excel.Application;
 
 namespace RhSolutions.Tools;
 
 internal class OcrTool : ITool
 {
 	private IOcrClient client = RhSolutionsAddIn.ServiceProvider.GetService<IOcrClient>();
+	private Application app = RhSolutionsAddIn.Excel;
+	private string xFolderId = RhSolutionsAddIn.Configuration["x-folder-id"];
+	private string apiKey = RhSolutionsAddIn.Configuration["apiKey"];
 
-	public void Execute()
+	public async void Execute()
 	{
 		try
 		{
@@ -19,7 +24,46 @@ internal class OcrTool : ITool
 			}).Wait();
 
 			string shot = Snipper.SnipBase64();
-			var result = client.ProcessImage(shot);
+			RhSolutionsAddIn.Excel.Visible = true;
+
+			if (shot != null)
+			{
+				IEnumerable<object[,]> tables = await client.ProcessImage(shot, xFolderId, apiKey);
+				if (tables != null)
+				{
+					foreach (var table in tables)
+					{
+						int rowCount = table.GetLength(0);
+						int columnCount = table.GetLength(1);
+
+						Range currentCell = app.ActiveCell;
+						Range tableRange = app.ActiveSheet.Range(currentCell,
+							app.ActiveSheet.Cells(currentCell.Row + rowCount - 1, currentCell.Column + columnCount - 1));
+
+						if (app.WorksheetFunction.CountA(tableRange) > 0)
+						{
+							MessageBox.Show(@"На листе отсустствует диапазон для вставки распознанной таблицы.Попробуйте в другом месте или на пустом листе.",
+								"Ошибка",
+								MessageBoxButtons.OK,
+								MessageBoxIcon.Error);
+							RhSolutionsAddIn.Excel.Visible = true;
+							return;
+						}
+						tableRange.Borders.LineStyle = XlLineStyle.xlContinuous;
+
+						for (int row = 0; row < rowCount; row++)
+							for (int column = 0; column < columnCount; column++)
+							{
+								Range excelCell = app.ActiveSheet.Cells(currentCell.Row + row,
+								currentCell.Column + column);
+								excelCell.Value2 = table[row,column];
+								excelCell.EntireColumn.AutoFit();
+								excelCell.EntireRow.AutoFit();
+							}
+						app.ActiveSheet.Cells(currentCell.Row + rowCount + 1, currentCell.Column).Activate();
+					}
+				}
+			}
 		}
 		catch (Exception)
 		{