Localisation

using System;
using System.IO;
using Montebearo.Discourse;
using Montebearo.Discourse.DataExport;
using UnityEditor;
using UnityEngine;

namespace SampleCode.Editor
{
	public static class BatchImportExport
	{
		// In this example, we assume all DiscourseEvents have unique names. If they don't, you will want to
		// edit this to ensure you can export/import unique files per-event.
		//
		// One way of doing that would be to export the CSVs using the event's project GUID as the file name,
		// and match events to their CSV files based on their guid in the import step.
		// Or you could export to sub-folders matching your project structure.
		//
		// We leave that decision up to you.
		
		private static string FormatEventFileName(DiscourseEvent eventAsset) => $"{eventAsset.name}.csv";

		private static bool TryGetFileDialogDirectory(string title, out string directory)
		{
			// Open the dialog to the Desktop by default.
			string defaultDirectory = Environment.GetFolderPath(Environment.SpecialFolder.Desktop);
			
			directory = EditorUtility.OpenFolderPanel(title, defaultDirectory, "ExportDirectory");
			return directory.Length > 0;
		}
		
		[MenuItem("Tools/Discourse/Batch Export Event Strings")]
		public static void BatchExport()
		{
			// Select a directory to export to
			if(!TryGetFileDialogDirectory("Select Export Directory", out string exportDirectory)) return;

			DiscourseEvent[] eventAssets = GetProjectEvents();

			int failed = 0;
			foreach (DiscourseEvent eventAsset in eventAssets)
			{
				// Export the event's strings to a CSV file with the same name.
				
				string path = Path.Combine(exportDirectory, FormatEventFileName(eventAsset));
				
				if (!TryExportEventStrings(eventAsset, path)) failed++;
			}
			
			Debug.Log($"<color=cyan>Exported {eventAssets.Length} event strings to {exportDirectory} - {failed} failures.</color>");
		}
		
		[MenuItem("Tools/Discourse/Batch Import Event Strings")]
		public static void BatchImport()
		{
			// Select a directory to import from
			if(!TryGetFileDialogDirectory("Select Import Directory", out string importDirectory)) return;

			DiscourseEvent[] eventAssets = GetProjectEvents();

			int failed = 0;
			int attempts = 0;
			foreach (DiscourseEvent eventAsset in eventAssets)
			{
				//Check if a CSV file exists in the import directory with this event's name, if so, attempt to import it.
				
				string filePath = Path.Combine(importDirectory, FormatEventFileName(eventAsset));
				if(!File.Exists(filePath)) continue;

				attempts++;
				
				if (!TryImportEventStrings(eventAsset, filePath)) failed++;
			}
			
			Debug.Log($"<color=cyan>Attempted to import {attempts} event strings to {importDirectory} - {failed} failures.</color>");
			
			// Save the project to write the modifications to disk.
			AssetDatabase.SaveAssets();
		}
		
		private static bool TryImportEventStrings(DiscourseEvent discourseEvent, string csvFilePath)
		{
			if (discourseEvent == null) return false;

			// Import the CSV text and then set the asset dirty, to ensure changes are serialized.
			ReferenceIO.ReadLocalisedTextFromCsv(discourseEvent.StringLocaliser, csvFilePath);
			EditorUtility.SetDirty(discourseEvent);
			return true;
		}
		
		private static bool TryExportEventStrings(DiscourseEvent discourseEvent, string exportPath)
		{
			if (discourseEvent == null) return false;
			
			//Export the event's strings to the CSV file path.
			ReferenceIO.WriteLocalisedTextToCsv(discourseEvent.StringLocaliser, exportPath);
			return true;
		}

		private static DiscourseEvent[] GetProjectEvents()
		{
			// Find all DiscourseEvent asset guids. If you want to limit this to specific folders, FindAssets()
			// optionally takes an array of folder paths to limit the search to.
			
			string[] guids = AssetDatabase.FindAssets("t:DiscourseEvent");

			DiscourseEvent[] eventAssets = new DiscourseEvent[guids.Length];

			for (int i = 0; i < guids.Length; i++)
			{
				string path = AssetDatabase.GUIDToAssetPath(guids[i]);

				eventAssets[i] = AssetDatabase.LoadAssetAtPath<DiscourseEvent>(path);
			}

			return eventAssets;
		}
	}
}