mirror of
https://github.com/MindWorkAI/AI-Studio.git
synced 2026-03-29 11:51:37 +00:00
centralized conversion methods to reduce repetition
This commit is contained in:
nilsk
parent
30d6b64c5b
commit
79c92bb8f1
@ -0,0 +1,271 @@
|
||||
using System.Collections;
|
||||
using Lua;
|
||||
|
||||
namespace AIStudio.Tools.PluginSystem.Assistants.DataModel;
|
||||
|
||||
internal static class AssistantLuaConversion
|
||||
{
|
||||
/// <summary>
|
||||
/// Converts a sequence of scalar .NET values into the array-like Lua table shape used by assistant state.
|
||||
/// </summary>
|
||||
public static LuaTable CreateLuaArray(IEnumerable values) => CreateLuaArrayCore(values);
|
||||
|
||||
/// <summary>
|
||||
/// Reads a Lua value into either a scalar .NET value or one of the structured assistant data model types.
|
||||
/// Lua itself only exposes scalars and tables, so structured assistant types such as dropdown/list items
|
||||
/// must be detected from well-known table shapes.
|
||||
/// </summary>
|
||||
public static bool TryReadScalarOrStructuredValue(LuaValue value, out object result)
|
||||
{
|
||||
if (value.TryRead<string>(out var stringValue))
|
||||
{
|
||||
result = stringValue;
|
||||
return true;
|
||||
}
|
||||
|
||||
if (value.TryRead<bool>(out var boolValue))
|
||||
{
|
||||
result = boolValue;
|
||||
return true;
|
||||
}
|
||||
|
||||
if (value.TryRead<double>(out var doubleValue))
|
||||
{
|
||||
result = doubleValue;
|
||||
return true;
|
||||
}
|
||||
|
||||
if (value.TryRead<LuaTable>(out var table) && TryParseDropdownItem(table, out var dropdownItem))
|
||||
{
|
||||
result = dropdownItem;
|
||||
return true;
|
||||
}
|
||||
|
||||
if (value.TryRead<LuaTable>(out var dropdownListTable) && TryParseDropdownItemList(dropdownListTable, out var dropdownItems))
|
||||
{
|
||||
result = dropdownItems;
|
||||
return true;
|
||||
}
|
||||
|
||||
if (value.TryRead<LuaTable>(out var listItemListTable) && TryParseListItemList(listItemListTable, out var listItems))
|
||||
{
|
||||
result = listItems;
|
||||
return true;
|
||||
}
|
||||
|
||||
result = null!;
|
||||
return false;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Writes an assistant value into a Lua table.
|
||||
/// This supports a broader set of .NET types than <see cref="TryReadScalarOrStructuredValue"/>,
|
||||
/// because assistant props and state already exist as rich C# objects before being serialized back to Lua.
|
||||
/// </summary>
|
||||
public static bool TryWriteAssistantValue(LuaTable table, string key, object? value)
|
||||
{
|
||||
if (value is null or LuaFunction)
|
||||
return false;
|
||||
|
||||
switch (value)
|
||||
{
|
||||
case LuaValue { Type: not LuaValueType.Nil } luaValue:
|
||||
table[key] = luaValue;
|
||||
return true;
|
||||
case LuaTable luaTable:
|
||||
table[key] = luaTable;
|
||||
return true;
|
||||
case string stringValue:
|
||||
table[key] = (LuaValue)stringValue;
|
||||
return true;
|
||||
case bool boolValue:
|
||||
table[key] = boolValue;
|
||||
return true;
|
||||
case byte byteValue:
|
||||
table[key] = byteValue;
|
||||
return true;
|
||||
case sbyte sbyteValue:
|
||||
table[key] = sbyteValue;
|
||||
return true;
|
||||
case short shortValue:
|
||||
table[key] = shortValue;
|
||||
return true;
|
||||
case ushort ushortValue:
|
||||
table[key] = ushortValue;
|
||||
return true;
|
||||
case int intValue:
|
||||
table[key] = intValue;
|
||||
return true;
|
||||
case uint uintValue:
|
||||
table[key] = uintValue;
|
||||
return true;
|
||||
case long longValue:
|
||||
table[key] = longValue;
|
||||
return true;
|
||||
case ulong ulongValue:
|
||||
table[key] = ulongValue;
|
||||
return true;
|
||||
case float floatValue:
|
||||
table[key] = floatValue;
|
||||
return true;
|
||||
case double doubleValue:
|
||||
table[key] = doubleValue;
|
||||
return true;
|
||||
case decimal decimalValue:
|
||||
table[key] = (double)decimalValue;
|
||||
return true;
|
||||
case Enum enumValue:
|
||||
table[key] = enumValue.ToString();
|
||||
return true;
|
||||
case AssistantDropdownItem dropdownItem:
|
||||
table[key] = CreateDropdownItemTable(dropdownItem);
|
||||
return true;
|
||||
case IEnumerable<AssistantDropdownItem> dropdownItems:
|
||||
table[key] = CreateLuaArrayCore(dropdownItems.Select(CreateDropdownItemTable));
|
||||
return true;
|
||||
case IEnumerable<AssistantListItem> listItems:
|
||||
table[key] = CreateLuaArrayCore(listItems.Select(CreateListItemTable));
|
||||
return true;
|
||||
case IEnumerable<string> strings:
|
||||
table[key] = CreateLuaArrayCore(strings);
|
||||
return true;
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
private static bool TryParseDropdownItem(LuaTable table, out AssistantDropdownItem item)
|
||||
{
|
||||
item = new AssistantDropdownItem();
|
||||
|
||||
if (!table.TryGetValue("Value", out var valueValue) || !valueValue.TryRead<string>(out var value))
|
||||
return false;
|
||||
|
||||
if (!table.TryGetValue("Display", out var displayValue) || !displayValue.TryRead<string>(out var display))
|
||||
return false;
|
||||
|
||||
item.Value = value;
|
||||
item.Display = display;
|
||||
return true;
|
||||
}
|
||||
|
||||
private static bool TryParseDropdownItemList(LuaTable table, out List<AssistantDropdownItem> items)
|
||||
{
|
||||
items = new List<AssistantDropdownItem>();
|
||||
|
||||
for (var index = 1; index <= table.ArrayLength; index++)
|
||||
{
|
||||
var value = table[index];
|
||||
if (!value.TryRead<LuaTable>(out var itemTable) || !TryParseDropdownItem(itemTable, out var item))
|
||||
{
|
||||
items = null!;
|
||||
return false;
|
||||
}
|
||||
|
||||
items.Add(item);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
private static bool TryParseListItem(LuaTable table, out AssistantListItem item)
|
||||
{
|
||||
item = new AssistantListItem();
|
||||
|
||||
if (!table.TryGetValue("Text", out var textValue) || !textValue.TryRead<string>(out var text))
|
||||
return false;
|
||||
|
||||
if (!table.TryGetValue("Type", out var typeValue) || !typeValue.TryRead<string>(out var type))
|
||||
return false;
|
||||
|
||||
table.TryGetValue("Icon", out var iconValue);
|
||||
iconValue.TryRead<string>(out var icon);
|
||||
|
||||
table.TryGetValue("IconColor", out var iconColorValue);
|
||||
iconColorValue.TryRead<string>(out var iconColor);
|
||||
|
||||
item.Text = text;
|
||||
item.Type = type;
|
||||
item.Icon = icon;
|
||||
item.IconColor = iconColor;
|
||||
|
||||
if (table.TryGetValue("Href", out var hrefValue) && hrefValue.TryRead<string>(out var href))
|
||||
item.Href = href;
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
private static bool TryParseListItemList(LuaTable table, out List<AssistantListItem> items)
|
||||
{
|
||||
items = new List<AssistantListItem>();
|
||||
|
||||
for (var index = 1; index <= table.ArrayLength; index++)
|
||||
{
|
||||
var value = table[index];
|
||||
if (!value.TryRead<LuaTable>(out var itemTable) || !TryParseListItem(itemTable, out var item))
|
||||
{
|
||||
items = null!;
|
||||
return false;
|
||||
}
|
||||
|
||||
items.Add(item);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
private static LuaTable CreateDropdownItemTable(AssistantDropdownItem item) =>
|
||||
new()
|
||||
{
|
||||
["Value"] = item.Value,
|
||||
["Display"] = item.Display,
|
||||
};
|
||||
|
||||
private static LuaTable CreateListItemTable(AssistantListItem item)
|
||||
{
|
||||
var table = new LuaTable
|
||||
{
|
||||
["Type"] = item.Type,
|
||||
["Text"] = item.Text,
|
||||
["Icon"] = item.Icon,
|
||||
["IconColor"] = item.IconColor,
|
||||
};
|
||||
|
||||
if (!string.IsNullOrWhiteSpace(item.Href))
|
||||
table["Href"] = item.Href;
|
||||
|
||||
return table;
|
||||
}
|
||||
|
||||
private static LuaTable CreateLuaArrayCore(IEnumerable values)
|
||||
{
|
||||
var luaArray = new LuaTable();
|
||||
var index = 1;
|
||||
|
||||
foreach (var value in values)
|
||||
{
|
||||
luaArray[index++] = value switch
|
||||
{
|
||||
null => LuaValue.Nil,
|
||||
LuaValue luaValue => luaValue,
|
||||
LuaTable luaTable => luaTable,
|
||||
string stringValue => (LuaValue)stringValue,
|
||||
bool boolValue => boolValue,
|
||||
byte byteValue => byteValue,
|
||||
sbyte sbyteValue => sbyteValue,
|
||||
short shortValue => shortValue,
|
||||
ushort ushortValue => ushortValue,
|
||||
int intValue => intValue,
|
||||
uint uintValue => uintValue,
|
||||
long longValue => longValue,
|
||||
ulong ulongValue => ulongValue,
|
||||
float floatValue => floatValue,
|
||||
double doubleValue => doubleValue,
|
||||
decimal decimalValue => (double)decimalValue,
|
||||
_ => LuaValue.Nil,
|
||||
};
|
||||
}
|
||||
|
||||
return luaArray;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user