using System;
using System.Collections.Generic;
using System.IO;
using System.Runtime.InteropServices;
using System.Text;
using System.Text.Json;
using System.Threading;
using System.Threading.Tasks;

namespace AxCopilot.Services.Agent;

public class Base64Tool : IAgentTool
{
	public string Name => "base64_tool";

	public string Description => "Encode or decode Base64 and URL strings. Actions: 'b64encode' — encode text to Base64; 'b64decode' — decode Base64 to text; 'urlencode' — URL-encode text; 'urldecode' — URL-decode text; 'b64file' — encode a file to Base64 (max 5MB).";

	public ToolParameterSchema Parameters
	{
		get
		{
			ToolParameterSchema toolParameterSchema = new ToolParameterSchema();
			Dictionary<string, ToolProperty> dictionary = new Dictionary<string, ToolProperty>();
			ToolProperty obj = new ToolProperty
			{
				Type = "string",
				Description = "Action to perform"
			};
			int num = 5;
			List<string> list = new List<string>(num);
			CollectionsMarshal.SetCount(list, num);
			Span<string> span = CollectionsMarshal.AsSpan(list);
			span[0] = "b64encode";
			span[1] = "b64decode";
			span[2] = "urlencode";
			span[3] = "urldecode";
			span[4] = "b64file";
			obj.Enum = list;
			dictionary["action"] = obj;
			dictionary["text"] = new ToolProperty
			{
				Type = "string",
				Description = "Text to encode/decode"
			};
			dictionary["file_path"] = new ToolProperty
			{
				Type = "string",
				Description = "File path for b64file action"
			};
			toolParameterSchema.Properties = dictionary;
			num = 1;
			List<string> list2 = new List<string>(num);
			CollectionsMarshal.SetCount(list2, num);
			CollectionsMarshal.AsSpan(list2)[0] = "action";
			toolParameterSchema.Required = list2;
			return toolParameterSchema;
		}
	}

	public Task<ToolResult> ExecuteAsync(JsonElement args, AgentContext context, CancellationToken ct = default(CancellationToken))
	{
		string text = args.GetProperty("action").GetString() ?? "";
		JsonElement value;
		string text2 = (args.TryGetProperty("text", out value) ? (value.GetString() ?? "") : "");
		try
		{
			if (1 == 0)
			{
			}
			ToolResult result = text switch
			{
				"b64encode" => ToolResult.Ok(Convert.ToBase64String(Encoding.UTF8.GetBytes(text2))), 
				"b64decode" => ToolResult.Ok(Encoding.UTF8.GetString(Convert.FromBase64String(text2))), 
				"urlencode" => ToolResult.Ok(Uri.EscapeDataString(text2)), 
				"urldecode" => ToolResult.Ok(Uri.UnescapeDataString(text2)), 
				"b64file" => EncodeFile(args, context), 
				_ => ToolResult.Fail("Unknown action: " + text), 
			};
			if (1 == 0)
			{
			}
			return Task.FromResult(result);
		}
		catch (FormatException)
		{
			return Task.FromResult(ToolResult.Fail("Invalid Base64 string"));
		}
		catch (Exception ex2)
		{
			return Task.FromResult(ToolResult.Fail("오류: " + ex2.Message));
		}
	}

	private static ToolResult EncodeFile(JsonElement args, AgentContext context)
	{
		if (!args.TryGetProperty("file_path", out var value))
		{
			return ToolResult.Fail("'file_path' is required for b64file action");
		}
		string text = value.GetString() ?? "";
		if (!Path.IsPathRooted(text))
		{
			text = Path.Combine(context.WorkFolder, text);
		}
		if (!File.Exists(text))
		{
			return ToolResult.Fail("File not found: " + text);
		}
		FileInfo fileInfo = new FileInfo(text);
		if (fileInfo.Length > 5242880)
		{
			return ToolResult.Fail($"File too large ({fileInfo.Length / 1024}KB). Max 5MB.");
		}
		byte[] inArray = File.ReadAllBytes(text);
		string text2 = Convert.ToBase64String(inArray);
		if (text2.Length > 10000)
		{
			return ToolResult.Ok($"Base64 ({text2.Length} chars, first 500):\n{text2.Substring(0, 500)}...");
		}
		return ToolResult.Ok(text2);
	}
}
