pk3DS/pk3DS.WinForms/Subforms/TextEditor.cs
2024-06-02 18:50:22 -05:00

318 lines
11 KiB
C#

using pk3DS.Core;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Windows.Forms;
namespace pk3DS.WinForms;
public partial class TextEditor : Form
{
public TextEditor(string[][] infiles, string mode)
{
InitializeComponent();
files = infiles;
Mode = mode;
for (int i = 0; i < files.Length; i++)
CB_Entry.Items.Add(i.ToString());
CB_Entry.SelectedIndex = 0;
dgv.EditMode = DataGridViewEditMode.EditOnEnter;
}
private readonly string[][] files;
private readonly string Mode;
private int entry = -1;
// IO
private void B_Export_Click(object sender, EventArgs e)
{
if (files.Length == 0)
return;
var Dump = new SaveFileDialog { Filter = "Text File|*.txt" };
var sdr = Dump.ShowDialog();
if (sdr != DialogResult.OK)
return;
bool newline = WinFormsUtil.Prompt(MessageBoxButtons.YesNo, "Remove newline formatting codes? (\\n,\\r,\\c)", "Removing newline formatting will make it more readable but will prevent any importing of that dump.") == DialogResult.Yes;
string path = Dump.FileName;
ExportTextFile(path, newline, files);
}
private void B_Import_Click(object sender, EventArgs e)
{
if (files.Length == 0)
return;
var Dump = new OpenFileDialog { Filter = "Text File|*.txt" };
var odr = Dump.ShowDialog();
if (odr != DialogResult.OK)
return;
string path = Dump.FileName;
if (!ImportTextFile(path))
return;
// Reload the form with the new data.
ChangeEntry(null, null);
WinFormsUtil.Alert("Imported Text from Input Path:", path);
}
public static void ExportTextFile(string fileName, bool newline, string[][] fileData)
{
using var ms = new MemoryStream();
ms.Write([0xFF, 0xFE], 0, 2); // Write Unicode BOM
using (var tw = new StreamWriter(ms, new UnicodeEncoding()))
{
for (int i = 0; i < fileData.Length; i++)
{
// Get Strings for the File
string[] data = fileData[i];
// Append the File Header
tw.WriteLine("~~~~~~~~~~~~~~~");
tw.WriteLine("Text File : " + i);
tw.WriteLine("~~~~~~~~~~~~~~~");
// Write the String to the File
if (data == null) continue;
foreach (string line in data)
{
tw.WriteLine(newline
? line.Replace("\\n\\n", " ")
.Replace("\\n", " ")
.Replace("\\c", "")
.Replace("\\r", "")
.Replace("\\\\", "\\")
.Replace("\\[", "[")
: line);
}
}
}
File.WriteAllBytes(fileName, ms.ToArray());
}
private bool ImportTextFile(string fileName)
{
string[] fileText = File.ReadAllLines(fileName, Encoding.Unicode);
string[][] textLines = new string[files.Length][];
int ctr = 0;
bool newlineFormatting = false;
// Loop through all files
for (int i = 0; i < fileText.Length; i++)
{
string line = fileText[i];
if (line != "~~~~~~~~~~~~~~~")
continue;
string[] brokenLine = fileText[i++ + 1].Split([" : "], StringSplitOptions.None);
if (brokenLine.Length != 2)
{ WinFormsUtil.Error($"Invalid Line @ {i}, expected Text File : {ctr}"); return false; }
int file = Util.ToInt32(brokenLine[1]);
if (file != ctr)
{ WinFormsUtil.Error($"Invalid Line @ {i}, expected Text File : {ctr}"); return false; }
i += 2; // Skip over the other header line
List<string> Lines = [];
while (i < fileText.Length && fileText[i] != "~~~~~~~~~~~~~~~")
{
Lines.Add(fileText[i]);
newlineFormatting |= fileText[i].Contains("\\n"); // Check if any line wasn't stripped of ingame formatting codes for human readability.
i++;
}
i--;
textLines[ctr++] = [.. Lines];
}
// Error Check
if (ctr != files.Length)
{
WinFormsUtil.Error("The amount of Text Files in the input file does not match the required for the text file.",
$"Received: {ctr}, Expected: {files.Length}"); return false;
}
if (!newlineFormatting)
{
WinFormsUtil.Error("The input Text Files do not have the ingame newline formatting codes (\\n,\\r,\\c).",
"When exporting text, do not remove newline formatting."); return false;
}
// All Text Lines received. Store all back.
for (int i = 0; i < files.Length; i++)
{
try { files[i] = textLines[i]; }
catch (Exception e) { WinFormsUtil.Error($"The input Text File (# {i}) failed to convert:", e.ToString()); return false; }
}
return true;
}
private void ChangeEntry(object sender, EventArgs e)
{
// Save All the old text
if (entry > -1 && sender != null)
{
try
{
files[entry] = GetCurrentDGLines();
}
catch (Exception ex) { WinFormsUtil.Error(ex.ToString()); }
}
// Reset
entry = CB_Entry.SelectedIndex;
SetStringsDataGridView(files[entry]);
}
// Main Handling
private void SetStringsDataGridView(string[] textArray)
{
// Clear the datagrid row content to remove all text lines.
dgv.Rows.Clear();
// Clear the header columns, these are repopulated every time.
dgv.Columns.Clear();
if (textArray == null || textArray.Length == 0)
return;
// Reset settings and columns.
dgv.AllowUserToResizeColumns = false;
DataGridViewColumn dgvLine = new DataGridViewTextBoxColumn
{
HeaderText = "Line",
DisplayIndex = 0,
Width = 32,
ReadOnly = true,
SortMode = DataGridViewColumnSortMode.NotSortable,
};
dgvLine.DefaultCellStyle.Alignment = DataGridViewContentAlignment.MiddleRight;
var dgvText = new DataGridViewTextBoxColumn
{
HeaderText = "Text",
DisplayIndex = 1,
SortMode = DataGridViewColumnSortMode.NotSortable,
AutoSizeMode = DataGridViewAutoSizeColumnMode.Fill,
};
dgv.Columns.Add(dgvLine);
dgv.Columns.Add(dgvText);
dgv.Rows.Add(textArray.Length);
// Add the text lines into their cells.
for (int i = 0; i < textArray.Length; i++)
{
dgv.Rows[i].Cells[0].Value = i;
dgv.Rows[i].Cells[1].Value = textArray[i];
}
}
private string[] GetCurrentDGLines()
{
// Get Line Count
string[] lines = new string[dgv.RowCount];
for (int i = 0; i < dgv.RowCount; i++)
lines[i] = (string)dgv.Rows[i].Cells[1].Value;
return lines;
}
// Meta Usage
private void B_AddLine_Click(object sender, EventArgs e)
{
int currentRow = 0;
try { currentRow = dgv.CurrentRow.Index; }
catch { dgv.Rows.Add(); }
if (dgv.Rows.Count != 1 && (currentRow < dgv.Rows.Count - 1 || currentRow == 0))
{
if (ModifierKeys != Keys.Control && currentRow != 0)
{
if (WinFormsUtil.Prompt(MessageBoxButtons.YesNo, "Inserting in between rows will shift all subsequent lines.", "Continue?") != DialogResult.Yes)
return;
}
// Insert new Row after current row.
dgv.Rows.Insert(currentRow + 1);
}
for (int i = 0; i < dgv.Rows.Count; i++)
dgv.Rows[i].Cells[0].Value = i.ToString();
}
private void B_RemoveLine_Click(object sender, EventArgs e)
{
int currentRow = dgv.CurrentRow.Index;
if (currentRow < dgv.Rows.Count - 1)
{
if (ModifierKeys != Keys.Control && DialogResult.Yes != WinFormsUtil.Prompt(MessageBoxButtons.YesNo, "Deleting a row above other lines will shift all subsequent lines.", "Continue?"))
return;
}
dgv.Rows.RemoveAt(currentRow);
// Resequence the Index Value column
for (int i = 0; i < dgv.Rows.Count; i++)
dgv.Rows[i].Cells[0].Value = i.ToString();
}
private void TextEditor_FormClosing(object sender, FormClosingEventArgs e)
{
// Save any pending edits
dgv.EndEdit();
// Save All the old text
if (entry > -1) files[entry] = GetCurrentDGLines();
}
private void B_Randomize_Click(object sender, EventArgs e)
{
// gametext can be horribly broken if randomized
if (Mode == "gametext" && DialogResult.Yes != WinFormsUtil.Prompt(MessageBoxButtons.YesNo, "Randomizing Game Text is dangerous!", "Continue?"))
return;
// get if the user wants to randomize current text file or all files
var dr = WinFormsUtil.Prompt(MessageBoxButtons.YesNoCancel, $"Yes: Randomize ALL{Environment.NewLine}No: Randomize current textfile{Environment.NewLine}Cancel: Abort");
if (dr == DialogResult.Cancel)
return;
// get if pure shuffle or smart shuffle (no shuffle if variable present)
var drs = WinFormsUtil.Prompt(MessageBoxButtons.YesNo, $"Smart shuffle:{Environment.NewLine}Yes: Shuffle if no Variable present{Environment.NewLine}No: Pure random!");
if (drs == DialogResult.Cancel)
return;
bool all = dr == DialogResult.Yes;
bool smart = drs == DialogResult.Yes;
// save current
if (entry > -1)
files[entry] = GetCurrentDGLines();
// single-entire looping
int start = all ? 0 : entry;
int end = all ? files.Length - 1 : entry;
// Gather strings
List<string> strings = [];
for (int i = start; i <= end; i++)
{
string[] data = files[i];
strings.AddRange(smart
? data.Where(line => !line.Contains('['))
: data);
}
// Shuffle up
string[] pool = [.. strings];
Util.Shuffle(pool);
// Apply Text
int ctr = 0;
for (int i = start; i <= end; i++)
{
string[] data = files[i];
for (int j = 0; j < data.Length; j++) // apply lines
{
if (!smart || !data[j].Contains('['))
data[j] = pool[ctr++];
}
files[i] = data;
}
// Load current text file
SetStringsDataGridView(files[entry]);
WinFormsUtil.Alert("Strings randomized!");
}
}