222 lines
8.4 KiB
C#
222 lines
8.4 KiB
C#
using System;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Threading.Tasks;
|
|
using Avalonia.Platform.Storage;
|
|
using ImageMagick;
|
|
using MigraDoc.DocumentObjectModel;
|
|
using MigraDoc.Rendering;
|
|
using PdfSharp.Fonts;
|
|
using PdfSharp.Pdf.IO;
|
|
using PdfSharp.Snippets.Font;
|
|
using ReceiptPDFBuilder.Interfaces;
|
|
|
|
namespace ReceiptPDFBuilder.ViewModels;
|
|
|
|
class MainViewModel : BaseViewModel, IFontResolver
|
|
{
|
|
private string _baseDir;
|
|
private bool _isCreatingPDF;
|
|
private string _createPDFLog;
|
|
|
|
public MainViewModel(IChangeViewModel viewModelChanger) : base(viewModelChanger)
|
|
{
|
|
_baseDir = Path.GetDirectoryName(Environment.ProcessPath) ?? "";
|
|
_isCreatingPDF = false;
|
|
_createPDFLog = "Ready to create PDF!";
|
|
}
|
|
|
|
public bool IsCreatingPDF
|
|
{
|
|
get => _isCreatingPDF;
|
|
set { _isCreatingPDF = value; NotifyPropertyChanged(); }
|
|
}
|
|
|
|
public string CreatePDFLog
|
|
{
|
|
get => _createPDFLog;
|
|
set { _createPDFLog = value; NotifyPropertyChanged(); }
|
|
}
|
|
|
|
private void LogInfo(string log)
|
|
{
|
|
Console.WriteLine(log);
|
|
CreatePDFLog += Environment.NewLine + log;
|
|
}
|
|
|
|
public async void ChooseFolder()
|
|
{
|
|
var topLevel = TopLevelGrabber?.GetTopLevel();
|
|
if (topLevel is not null)
|
|
{
|
|
var folders = await topLevel.StorageProvider.OpenFolderPickerAsync(new FolderPickerOpenOptions()
|
|
{
|
|
Title = "Pick a folder of files...",
|
|
AllowMultiple = false,
|
|
});
|
|
if (folders.Count == 1)
|
|
{
|
|
var folder = folders[0];
|
|
LogInfo("Chosen folder: " + folder.Path.LocalPath);
|
|
if (Directory.Exists(folder.Path.LocalPath))
|
|
{
|
|
await Task.Run(() => CreatePDF(folder.Path.LocalPath));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public byte[]? GetFont(string faceName)
|
|
{
|
|
LogInfo(string.Format("Loading font {0}", faceName));
|
|
if (faceName == "Noto Sans JP")
|
|
{
|
|
return File.ReadAllBytes(Path.Combine(_baseDir, "Assets/Fonts/Noto_Sans_JP/static/NotoSansJP-Regular.ttf"));
|
|
}
|
|
if (faceName == "Noto Sans JP Bold")
|
|
{
|
|
return File.ReadAllBytes(Path.Combine(_baseDir, "Assets/Fonts/Noto_Sans_JP/static/NotoSansJP-SemiBold.ttf"));
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public FontResolverInfo? ResolveTypeface(string familyName, bool bold, bool italic)
|
|
{
|
|
// LogInfo(string.Format("Resolving font name {0}", familyName));
|
|
if (familyName == "Noto Sans JP")
|
|
{
|
|
if (bold)
|
|
{
|
|
return new FontResolverInfo(familyName + " Bold");
|
|
}
|
|
return new FontResolverInfo(familyName);
|
|
}
|
|
return null;
|
|
}
|
|
|
|
// https://forum.pdfsharp.net/viewtopic.php?f=2&t=1025
|
|
private async Task CreatePDF(string folderPath)
|
|
{
|
|
// TODO: calculate needed width for images based on page width and margins and all that?
|
|
// TODO: resize (non-HEIC) images for smaller size...?
|
|
IsCreatingPDF = true;
|
|
var pdfDoc = new Document();
|
|
var outputFileName = "MyReceipts.pdf";
|
|
var folderName = new DirectoryInfo(folderPath).Name;
|
|
LogInfo("Folder name is " + folderName);
|
|
if (folderName.Contains('-'))
|
|
{
|
|
// see if year/month format
|
|
var parts = folderName.Split('-');
|
|
LogInfo(string.Format("{0}, {1}", parts[0], parts[1]));
|
|
if (parts[0].Length == 4 &&
|
|
parts[1].Length <= 2 &&
|
|
int.TryParse(parts[0], out int year) && int.TryParse(parts[1], out int month))
|
|
{
|
|
outputFileName = string.Format("{0} {1} Receipts.pdf",
|
|
CultureInfo.CurrentCulture.DateTimeFormat.GetMonthName(month),
|
|
year);
|
|
LogInfo("Auto-changed output file name to " + outputFileName);
|
|
}
|
|
}
|
|
var section = pdfDoc.AddSection();
|
|
section.PageSetup.PageFormat = PageFormat.Letter;
|
|
section.PageSetup.PageWidth = "8.5in";
|
|
section.PageSetup.PageHeight = "11in";
|
|
section.PageSetup.TopMargin = "0.5in";
|
|
section.PageSetup.RightMargin = "0.5in";
|
|
section.PageSetup.BottomMargin = "0.5in";
|
|
section.PageSetup.LeftMargin = "0.5in";
|
|
// setup footer for page number
|
|
var footerPar = new Paragraph();
|
|
footerPar.Format.Alignment = ParagraphAlignment.Center;
|
|
footerPar.AddText("--Page ");
|
|
footerPar.AddPageField();
|
|
footerPar.AddText(" of ");
|
|
footerPar.AddNumPagesField();
|
|
footerPar.AddText("--");
|
|
section.Footers.Primary.Add(footerPar);
|
|
//
|
|
var files = Directory.GetFiles(folderPath);
|
|
files.Sort();
|
|
GlobalFontSettings.FontResolver = this;
|
|
GlobalFontSettings.FallbackFontResolver = new FailsafeFontResolver();
|
|
for (var i = 0; i < files.Length; i++)
|
|
{
|
|
var file = files[i];
|
|
var fileName = Path.GetFileName(file);
|
|
if (fileName == ".DS_Store" || fileName == outputFileName)
|
|
{
|
|
continue;
|
|
}
|
|
var imageTitlePar = section.AddParagraph();
|
|
imageTitlePar.Format.Alignment = ParagraphAlignment.Center;
|
|
imageTitlePar.Format.Font.Size = 12;
|
|
imageTitlePar.Format.Font.Bold = true;
|
|
imageTitlePar.Format.Font.Name = "Noto Sans JP"; // has english letters in it, too
|
|
imageTitlePar.AddText(fileName);
|
|
section.AddParagraph(); // add empty line for spacing
|
|
// now add the image
|
|
var isPDF = fileName.EndsWith(".pdf");
|
|
var isHEIC = fileName.EndsWith(".HEIC") || fileName.EndsWith(".heic");
|
|
if (isHEIC)
|
|
{
|
|
var convertedDir = Path.Combine(folderPath, "converted");
|
|
if (!Directory.Exists(convertedDir))
|
|
{
|
|
Directory.CreateDirectory(convertedDir);
|
|
}
|
|
var info = new FileInfo(file);
|
|
using var mImage = new MagickImage(info.FullName);
|
|
// Save frame as jpg
|
|
var outputPath = Path.Combine(convertedDir, info.Name + ".jpg");
|
|
mImage.Quality = 80;
|
|
mImage.Scale((uint)Math.Floor(mImage.Width * 0.5), (uint)Math.Floor(mImage.Height * 0.5));
|
|
await mImage.WriteAsync(outputPath);
|
|
fileName = Path.Combine("Converted", info.Name + ".jpg");
|
|
LogInfo(string.Format("Converted HEIC image to JPEG; fileName is now {0}", fileName));
|
|
}
|
|
var paragraph = section.AddParagraph();
|
|
paragraph.Format.Alignment = ParagraphAlignment.Center;
|
|
var image = paragraph.AddImage(fileName);
|
|
image.LockAspectRatio = true;
|
|
image.Width = 400; // can't be too wide now...not sure why...maybe due to margins...
|
|
LogInfo(string.Format("Added image: {0}", fileName));
|
|
if (isPDF)
|
|
{
|
|
// add other PDF pages
|
|
// see: https://stackoverflow.com/a/65091204/3938401
|
|
var pdfFileToAdd = PdfReader.Open(file);
|
|
imageTitlePar.AddText(string.Format(" (PDF with {0} page{1}) ",
|
|
pdfFileToAdd.PageCount,
|
|
pdfFileToAdd.PageCount == 1 ? "" : "s"));
|
|
for (var j = 2; j <= pdfFileToAdd.PageCount; j++)
|
|
{
|
|
section.AddPageBreak();
|
|
paragraph = section.AddParagraph();
|
|
paragraph.Format.Alignment = ParagraphAlignment.Center;
|
|
image = paragraph.AddImage(file + "#" + j);
|
|
image.LockAspectRatio = true;
|
|
image.Width = 400;
|
|
}
|
|
}
|
|
if (i < files.Length - 1)
|
|
{
|
|
section.AddPageBreak();
|
|
}
|
|
}
|
|
var pdfRenderer = new PdfDocumentRenderer
|
|
{
|
|
Document = pdfDoc,
|
|
WorkingDirectory = folderPath
|
|
};
|
|
LogInfo("Rendering document...");
|
|
pdfRenderer.RenderDocument();
|
|
string filename = Path.Join(folderPath, outputFileName);
|
|
LogInfo("Saving document to disk...");
|
|
pdfRenderer.PdfDocument.Save(filename);
|
|
LogInfo("Saved PDF output to: " + filename);
|
|
IsCreatingPDF = false;
|
|
return;
|
|
}
|
|
} |