hellbound/Assets/Sources/Lean/Transition/Editor/Scripts/LeanBuilder.cs

using UnityEngine;
using UnityEditor;
using System.Collections.Generic;
using Lean.Common;

namespace Lean.Transition.Editor
{
	/// <summary>This allows you to build a transition method from the specified template.</summary>
	[CreateAssetMenu(fileName = "NewBuilder", menuName = "Lean/Transition/Builder")]
	public class LeanBuilder : ScriptableObject
	{
		[System.Serializable]
		public class Entry
		{
			public string       Title;
			public LeanTemplate Template;

			[Multiline(8)]
			public string Contents;
		}

		/// <summary>This allows you to specify which transition methods will be built.</summary>
		public List<Entry> Entries { get { if (entries == null) entries = new List<Entry>(); return entries; } } [SerializeField] private List<Entry> entries;

		public void Build()
		{
			if (entries != null)
			{
				AssetDatabase.ImportAsset(AssetDatabase.GetAssetPath(this));

				var path      = System.IO.Path.GetDirectoryName(AssetDatabase.GetAssetPath(this));
				var longName  = name;
				var shortName = name;
				var index     = shortName.LastIndexOf(".");

				if (index >= 0)
				{
					shortName = shortName.Substring(index + 1);
				}

				foreach (var entry in entries)
				{
					Build(entry, longName, shortName, path + "/Lean" + shortName + entry.Title + ".cs");
				}
			}
		}

		private void Build(Entry entry, string longName, string shortName, string path)
		{
			var body = entry.Template.Body;
			
			body = body.Replace("{COMPONENT_FULL}", longName);
			body = body.Replace("{COMPONENT}", shortName);
			body = body.Replace("{TITLE}", shortName + entry.Title);

			var contents = entry.Contents.Split(new char[] { '\r', '\n' }, System.StringSplitOptions.RemoveEmptyEntries);

			foreach (var content in contents)
			{
				var indices = GetIndex(content);
				var left    = content.Substring(0, indices.x);
				var right   = content.Substring(indices.y);

				body = body.Replace("{" + left + "}", right);
			}

			body = body.Replace("{ATTRIBUTE}", "");
			body = body.Replace("{FSA}", "");
			body = body.Replace("{DEFAULT}", "");

			System.IO.File.WriteAllText(path, body);

			AssetDatabase.ImportAsset(path);
		}

		private static Vector2Int GetIndex(string s)
		{
			var a = s.IndexOf('\t');

			for (var i = a + 1; i < s.Length; i++)
			{
				if (s[i] != '	')
				{
					return new Vector2Int(a, i);
				}
			}

			return new Vector2Int(-1, -1);
		}
	}

	[CanEditMultipleObjects]
	[CustomEditor(typeof(LeanBuilder))]
	public class LeanBuilder_Editor : LeanEditor
	{
		protected override void OnInspector()
		{
			LeanBuilder tgt; LeanBuilder[] tgts; GetTargets(out tgt, out tgts);

			EditorGUILayout.LabelField("Transition Builder", EditorStyles.boldLabel);

			Separator();

			BeginLabelWidth(100);
				Draw("entries", "This allows you to specify which transition methods will be built.");
			EndLabelWidth();

			Separator();

			if (GUILayout.Button("BUILD") == true)
			{
				Each(tgts, t => t.Build());
			}
		}
	}
}