Storing in DB now

This commit is contained in:
Lani Aung
2021-10-18 22:50:28 -06:00
parent 372d0ad86a
commit 804ccd0eeb
9 changed files with 161 additions and 17 deletions
+6 -5
View File
@@ -3,6 +3,7 @@ using System.Collections.Generic;
using System.Diagnostics; using System.Diagnostics;
using System.IO; using System.IO;
using System.IO.IsolatedStorage; using System.IO.IsolatedStorage;
using System.Threading.Tasks;
using fxl.codes.kisekae.Models; using fxl.codes.kisekae.Models;
using fxl.codes.kisekae.Services; using fxl.codes.kisekae.Services;
using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Http;
@@ -13,16 +14,16 @@ namespace fxl.codes.kisekae.Controllers
{ {
public class HomeController : Controller public class HomeController : Controller
{ {
private readonly FileParserService _fileParserService; private readonly DatabaseService _databaseService;
private readonly ILogger<HomeController> _logger; private readonly ILogger<HomeController> _logger;
private readonly ConfigurationReaderService _readerService; private readonly ConfigurationReaderService _readerService;
private readonly IsolatedStorageFile _storage; private readonly IsolatedStorageFile _storage;
public HomeController(ILogger<HomeController> logger, ConfigurationReaderService readerService, FileParserService fileParserService) public HomeController(ILogger<HomeController> logger, ConfigurationReaderService readerService, DatabaseService databaseService)
{ {
_logger = logger; _logger = logger;
_readerService = readerService; _readerService = readerService;
_fileParserService = fileParserService; _databaseService = databaseService;
_storage = IsolatedStorageFile.GetUserStoreForApplication(); _storage = IsolatedStorageFile.GetUserStoreForApplication();
} }
@@ -48,13 +49,13 @@ namespace fxl.codes.kisekae.Controllers
} }
[HttpPost] [HttpPost]
public IActionResult Upload(IFormFile file) public async Task<IActionResult> Upload(IFormFile file)
{ {
_logger.LogTrace($"File uploaded: {file?.FileName}"); _logger.LogTrace($"File uploaded: {file?.FileName}");
if (!(file?.FileName.EndsWith("lzh", StringComparison.InvariantCultureIgnoreCase) ?? false)) if (!(file?.FileName.EndsWith("lzh", StringComparison.InvariantCultureIgnoreCase) ?? false))
throw new Exception("Please select a *.lzh file"); throw new Exception("Please select a *.lzh file");
_fileParserService.UnzipLzh(file); await _databaseService.StoreToDatabase(file);
return Redirect("/"); return Redirect("/");
} }
+10
View File
@@ -0,0 +1,10 @@
namespace fxl.codes.kisekae.Entities
{
public class CelDto : IKisekaeFile, IKisekaeParseable
{
public int Id { get; set; }
public int KisekaeId { get; set; }
public string Filename { get; set; }
public byte[] Data { get; set; }
}
}
+10
View File
@@ -0,0 +1,10 @@
namespace fxl.codes.kisekae.Entities
{
public class ConfigurationDto : IKisekaeFile
{
public int Id { get; set; }
public string Filename { get; set; }
public string Data { get; set; }
public int KisekaeId { get; set; }
}
}
+14
View File
@@ -0,0 +1,14 @@
namespace fxl.codes.kisekae.Entities
{
public interface IKisekaeFile
{
int Id { get; set; }
int KisekaeId { get; set; }
string Filename { get; set; }
}
public interface IKisekaeParseable
{
byte[] Data { get; set; }
}
}
@@ -1,10 +1,14 @@
using System.Collections.Generic;
namespace fxl.codes.kisekae.Entities namespace fxl.codes.kisekae.Entities
{ {
public class Kisekae public class KisekaeDto
{ {
public int Id { get; set; } public int Id { get; set; }
public string Name { get; set; } public string Name { get; set; }
public string Filename { get; set; } public string Filename { get; set; }
public string Checksum { get; set; } public string Checksum { get; set; }
public IEnumerable<CelDto> Cels { get; set; }
} }
} }
+11
View File
@@ -0,0 +1,11 @@
namespace fxl.codes.kisekae.Entities
{
public class PaletteDto : IKisekaeFile, IKisekaeParseable
{
public int Id { get; set; }
public int KisekaeId { get; set; }
public string Filename { get; set; }
public string Comment { get; set; }
public byte[] Data { get; set; }
}
}
+85 -5
View File
@@ -1,9 +1,14 @@
using System;
using System.Collections.Generic;
using System.IO; using System.IO;
using System.IO.IsolatedStorage; using System.IO.IsolatedStorage;
using System.Security.Cryptography;
using System.Threading.Tasks;
using Dapper; using Dapper;
using fxl.codes.kisekae.Entities; using fxl.codes.kisekae.Entities;
using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.Configuration; using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.Logging;
using Npgsql; using Npgsql;
namespace fxl.codes.kisekae.Services namespace fxl.codes.kisekae.Services
@@ -12,28 +17,103 @@ namespace fxl.codes.kisekae.Services
{ {
private readonly string _connectionString; private readonly string _connectionString;
private readonly FileParserService _fileParserService; private readonly FileParserService _fileParserService;
private readonly ILogger<DatabaseService> _logger;
private readonly IsolatedStorageFile _storage; private readonly IsolatedStorageFile _storage;
public DatabaseService(IConfiguration configuration, FileParserService fileParserService) public DatabaseService(ILogger<DatabaseService> logger, IConfiguration configuration, FileParserService fileParserService)
{ {
_logger = logger;
_fileParserService = fileParserService; _fileParserService = fileParserService;
_connectionString = configuration.GetConnectionString("kisekae"); _connectionString = configuration.GetConnectionString("kisekae");
_storage = IsolatedStorageFile.GetUserStoreForApplication(); _storage = IsolatedStorageFile.GetUserStoreForApplication();
} }
public async void StoreToDatabase(IFormFile file) public async Task<KisekaeDto> StoreToDatabase(IFormFile file)
{ {
await using var connection = new NpgsqlConnection(_connectionString); await using var connection = new NpgsqlConnection(_connectionString);
await connection.OpenAsync(); await connection.OpenAsync();
var existing = connection.QuerySingle<Kisekae>("select * from kisekae where kisekae.filename = @Filename", file); var existing = connection.QuerySingleOrDefault<KisekaeDto>("select * from kisekae where kisekae.filename = @Filename", new { Filename = file.FileName });
if (existing != null) return; if (existing != null)
{
_logger.LogInformation($"Already existing upload {file.FileName} under id {existing.Id}");
return existing;
}
_fileParserService.UnzipLzh(file); await using var memoryStream = new MemoryStream();
await file.CopyToAsync(memoryStream);
var checksum = Convert.ToBase64String(await new SHA256Managed().ComputeHashAsync(memoryStream));
existing = connection.QuerySingleOrDefault<KisekaeDto>("select * from kisekae where kisekae.checksum = @Checksum", new { Checksum = checksum });
if (existing != null)
{
_logger.LogInformation($"Already existing upload (checksum: {checksum}) under id {existing.Id} and filename {existing.Filename}");
return existing;
}
memoryStream.Position = 0;
_fileParserService.UnzipLzh(file, memoryStream);
var directory = Path.GetFileNameWithoutExtension(file.FileName); var directory = Path.GetFileNameWithoutExtension(file.FileName);
var kisekae = new KisekaeDto
{
Filename = file.FileName,
Name = directory,
Checksum = checksum
};
var id = connection.QueryFirst<int>("insert into kisekae(name, filename, checksum) values (@Name, @Filename, @Checksum) returning id", kisekae);
kisekae.Id = id;
var filenames = _storage.GetFileNames($"{Path.Combine(directory, "*")}");
var cels = new List<CelDto>();
var palettes = new List<PaletteDto>();
var configurations = new List<ConfigurationDto>();
foreach (var filename in filenames)
{
await using var reader = _storage.OpenFile(Path.Combine(directory, filename), FileMode.Open);
if (!reader.CanRead) throw new InvalidDataException();
var bytes = new byte[reader.Length];
reader.Read(bytes, 0, bytes.Length);
switch (Path.GetExtension(filename).ToLower())
{
case ".cel":
cels.Add(new CelDto
{
Filename = filename,
Data = bytes,
KisekaeId = id
});
break;
case ".cnf":
configurations.Add(new ConfigurationDto
{
Filename = filename,
Data = BitConverter.ToString(bytes),
KisekaeId = id
});
break;
case ".kcf":
palettes.Add(new PaletteDto
{
Filename = filename,
Data = bytes,
KisekaeId = id
});
break;
}
}
await connection.ExecuteAsync("insert into cel (filename, data, kisekae_id) values (@Filename, @Data, @KisekaeId)", cels);
await connection.ExecuteAsync("insert into configuration (filename, data, kisekae_id) values (@Filename, @Data, @KisekaeId)", configurations);
await connection.ExecuteAsync("insert into palette (filename, data, kisekae_id) values (@Filename, @Data, @KisekaeId)", palettes);
await connection.CloseAsync(); await connection.CloseAsync();
return kisekae;
} }
} }
} }
+17 -3
View File
@@ -28,12 +28,17 @@ namespace fxl.codes.kisekae.Services
_storage = IsolatedStorageFile.GetUserStoreForApplication(); _storage = IsolatedStorageFile.GetUserStoreForApplication();
} }
public void UnzipLzh(IFormFile file) public void UnzipLzh(IFormFile file, MemoryStream memoryStream = null)
{ {
if (_storage.FileExists(file.FileName)) _storage.DeleteFile(file.FileName); if (_storage.FileExists(file.FileName)) _storage.DeleteFile(file.FileName);
using var writer = _storage.CreateFile(file.FileName); using var writer = _storage.CreateFile(file.FileName);
file.CopyTo(writer);
if (memoryStream != null)
memoryStream.CopyTo(writer);
else
file.CopyTo(writer);
writer.Flush(); writer.Flush();
writer.Close(); writer.Close();
@@ -43,7 +48,14 @@ namespace fxl.codes.kisekae.Services
.ToString() ?? ""; .ToString() ?? "";
var directory = Path.GetFileNameWithoutExtension(file.FileName); var directory = Path.GetFileNameWithoutExtension(file.FileName);
if (_storage.DirectoryExists(directory)) _storage.DeleteDirectory(directory); if (_storage.DirectoryExists(directory))
{
var allFiles = _storage.GetFileNames(Path.Combine(directory, "*"));
foreach (var existingFile in allFiles) _storage.DeleteFile(Path.Combine(directory, existingFile));
_storage.DeleteDirectory(directory);
}
_storage.CreateDirectory(directory); _storage.CreateDirectory(directory);
var process = new Process var process = new Process
@@ -62,6 +74,8 @@ namespace fxl.codes.kisekae.Services
while (!process.StandardOutput.EndOfStream) _logger.LogTrace(process.StandardOutput.ReadLine()); while (!process.StandardOutput.EndOfStream) _logger.LogTrace(process.StandardOutput.ReadLine());
process.WaitForExit(); process.WaitForExit();
_storage.DeleteFile(file.FileName);
} }
public void ParsePalette(string directory, PaletteModel palette) public void ParsePalette(string directory, PaletteModel palette)
+1 -1
View File
@@ -35,7 +35,7 @@ namespace fxl.codes.kisekae
if (env.IsDevelopment()) if (env.IsDevelopment())
{ {
app.UseDeveloperExceptionPage(); app.UseDeveloperExceptionPage();
NpgsqlLogManager.Provider = new ConsoleLoggingProvider(NpgsqlLogLevel.Trace); NpgsqlLogManager.Provider = new ConsoleLoggingProvider();
NpgsqlLogManager.IsParameterLoggingEnabled = true; NpgsqlLogManager.IsParameterLoggingEnabled = true;
} }
else else