Partsource/PartSource.Automation/Services/WhiSeoService.cs

#pragma warning disable CA2100 // Review SQL queries for security vulnerabilities

using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.Logging;
using PartSource.Automation.Models.Configuration;
using PartSource.Automation.Models.Enums;
using System;
using System.Collections.Generic;
using System.Data;
using System.Data.SqlClient;
using System.Text;

namespace PartSource.Automation.Services
{
	public class WhiSeoService
	{
		private readonly FtpService _ftpService;
		private readonly string _connectionString;
		private readonly ILogger<WhiSeoService> _logger;

		public WhiSeoService(IConfiguration configuration, ILogger<WhiSeoService> logger)
		{
			FtpConfiguration ftpConfiguration = configuration.GetSection("FtpServers:WhiConfiguration").Get<FtpConfiguration>();
			_ftpService = new FtpService(ftpConfiguration);

			_connectionString = configuration.GetConnectionString("FitmentDatabase");

			_logger = logger;
		}

		public void GetFiles(SeoDataType seoDataType)
		{
			string seoDataTypeString = seoDataType.ToString().ToLowerInvariant();
			string[] files = _ftpService.ListFiles(seoDataTypeString);

			foreach (string file in files)
			{
				if (file.Contains(".csv"))
				{
					try
					{
						_ftpService.Download($"{seoDataTypeString}/{file}");
						_logger.LogInformation($"Finished downloading {file}.");
					}


					catch (Exception ex)
					{
						_logger.LogWarning($"Failed to download {file}, quitting", ex);
						throw;
					}
				}
			}
		}

		public void TruncateVehicleTable()
		{
			using SqlConnection connection = new SqlConnection(_connectionString);
			connection.Open();

			using SqlCommand command = new SqlCommand($"truncate table dbo.Vehicle", connection);
			command.ExecuteNonQuery();
		}

		public void TruncateFitmentTables()
		{
			using SqlConnection connection = new SqlConnection(_connectionString);
			connection.Open();

			using SqlCommand command = new SqlCommand($"exec DropFitmentTables", connection);
			command.ExecuteNonQuery();
		}

		public void SaveNotes(IDictionary<string, string> notes)
		{
			using DataTable dataTable = new DataTable();
			dataTable.Columns.Add("NoteText", typeof(string));
			dataTable.Columns.Add("Hash", typeof(string));

			foreach (KeyValuePair<string, string> note in notes)
			{

				dataTable.Rows.Add(new string[] { note.Value, note.Key });
			}

			using SqlConnection connection = new SqlConnection(_connectionString);
			connection.Open();

			using SqlBulkCopy bulk = new SqlBulkCopy(connection)
			{
				DestinationTableName = $"FitmentNote",
				BulkCopyTimeout = 14400
			};

			bulk.WriteToServer(dataTable);
		}

		public void BulkCopyFitment(DataTable dataTable, string tableName)
		{
			using SqlConnection connection = new SqlConnection(_connectionString);
			connection.Open();

			string sql = string.Empty;

			using SqlCommand command = new SqlCommand($"EXEC CreateFitmentTempTable @tableName = '{tableName}'", connection);
			command.ExecuteNonQuery();

			using SqlBulkCopy bulk = new SqlBulkCopy(connection)
			{
				DestinationTableName = $"FitmentTemp.{tableName}",
				BulkCopyTimeout = 1
			};

			bulk.WriteToServer(dataTable);
		}

		public void BulkCopyVehicle(DataTable dataTable, string tableName)
		{
			using SqlConnection connection = new SqlConnection(_connectionString);
			connection.Open();

			string sql = string.Empty;

			using SqlCommand command = new SqlCommand($"EXEC CreateVehicleTempTable @tableName = '{tableName}'", connection);
			command.ExecuteNonQuery();

			using SqlBulkCopy bulk = new SqlBulkCopy(connection)
			{
				DestinationTableName = $"VehicleTemp.{tableName}",
				BulkCopyTimeout = 14400
			};

			bulk.WriteToServer(dataTable);
		}

		public void CreateFitmentTable(string tableName)
		{
			using SqlConnection connection = new SqlConnection(_connectionString);
			connection.Open();

			using SqlCommand command = new SqlCommand($"exec CreateFitmentTable @tableName = '{tableName}'", connection);
			command.CommandTimeout = 1800;
			command.ExecuteNonQuery();
		}

		public void CreateFitmentView()
		{
			using SqlConnection connection = new SqlConnection(_connectionString);
			connection.Open();

			using SqlCommand command = new SqlCommand($"exec CreateFitmentView", connection);
			command.ExecuteNonQuery();

			using SqlCommand command2 = new SqlCommand($"exec CreateFitmentIndexes", connection);
			command2.ExecuteNonQuery();
		}

		public void CreateVehicleTable()
		{
			return;

			using SqlConnection connection = new SqlConnection(_connectionString);
			connection.Open();

			using SqlCommand command = new SqlCommand($"exec CreateVehicleTable", connection);
			command.CommandTimeout = 1800;
			command.ExecuteNonQuery();
		}
	}
}
#pragma warning restore CA2100 // Review SQL queries for security vulnerabilities