sdk/dotnet-client/PluginFunctionsManager.cs - develop - krasninja/querycat

sdk/dotnet-client/PluginFunctionsManager.cs
94 строки3 KB
Ivan Kozhin
Move messages to resource for Sdk Client project
01 мар 2024, 15:54
01 мар 2024, 15:5487b961d
using System;
using System.Collections.Generic;
using System.Reflection;
using QueryCat.Backend.Core.Execution;
using QueryCat.Backend.Core.Functions;
using QueryCat.Plugins.Sdk;
using VariantValue = QueryCat.Backend.Core.Types.VariantValue;

namespace QueryCat.Plugins.Client;

/// <summary>
/// Plugin functions manager. This is the simplified and limited version of functions manager.
/// It does not support aggregates and does not parse functions signature. No overloading.
/// </summary>
public sealed class PluginFunctionsManager : IFunctionsManager
{
    private readonly Dictionary<string, PluginFunction> _functions = new();

    /// <inheritdoc />
    public IFunction? ResolveUri(string uri)
    {
        throw new NotImplementedException();
    }

    /// <inheritdoc />
    public void RegisterAggregate<TAggregate>(Func<TAggregate> factory)
        where TAggregate : IAggregateFunction
    {
        throw ThrowNotImplementedException();
    }

    /// <inheritdoc />
    public void RegisterFunction(string signature, FunctionDelegate @delegate, string? description = null)
    {
        var firstBracketIndex = signature.IndexOf('(');
        if (firstBracketIndex < 0)
        {
            return;
        }
        var name = signature.Substring(0, firstBracketIndex).ToUpper();
        _functions[name] = new PluginFunction(
            name,
            signature,
            @delegate)
        {
            Description = description ?? string.Empty,
            IsSafe = @delegate.Method.GetCustomAttribute<SafeFunctionAttribute>() != null,
        };
    }

    /// <inheritdoc />
    public void RegisterFactory(Action<IFunctionsManager> registerFunction, bool postpone = true)
    {
        throw ThrowNotImplementedException();
    }

    /// <inheritdoc />
    public bool TryFindByName(string name, FunctionCallArgumentsTypes? functionArgumentsTypes, out IFunction[] functions)
    {
        if (_functions.TryGetValue(name.ToUpper(), out var functionInfo))
        {
            functions = [functionInfo];
            return true;
        }
        functions = Array.Empty<IFunction>();
        return false;
    }

    /// <inheritdoc />
    public bool TryFindAggregateByName(string name, out IAggregateFunction? aggregateFunction)
    {
        throw ThrowNotImplementedException();
    }

    /// <inheritdoc />
    public IEnumerable<IFunction> GetFunctions() => _functions.Values;

    /// <summary>
    /// Get all functions signatures.
    /// </summary>
    /// <returns>Signatures strings.</returns>
    public IEnumerable<PluginFunction> GetPluginFunctions() => _functions.Values;

    /// <inheritdoc />
    public VariantValue CallFunction(IFunction function, IExecutionThread executionThread, FunctionCallArguments callArguments)
    {
        throw ThrowNotImplementedException();
    }

    private Exception ThrowNotImplementedException()
    {
        return new QueryCatPluginException(ErrorType.GENERIC, Resources.Errors.NotSupported_FunctionsManager);
    }
}