cs-codex-dist-tests/ProjectPlugins/CodexPlugin/CodexPlugin.cs

68 lines
2.2 KiB
C#
Raw Normal View History

2023-09-12 11:32:06 +00:00
using Core;
2023-09-11 14:57:57 +00:00
using KubernetesWorkflow;
namespace CodexPlugin
{
2023-09-13 14:06:05 +00:00
public class CodexPlugin : IProjectPlugin, IHasLogPrefix, IHasMetadata
2023-09-11 14:57:57 +00:00
{
2023-09-13 08:03:11 +00:00
private readonly CodexStarter codexStarter;
private readonly IPluginTools tools;
private readonly CodexLogLevel defaultLogLevel = CodexLogLevel.Trace;
2023-09-11 14:57:57 +00:00
2023-09-13 08:03:11 +00:00
public CodexPlugin(IPluginTools tools)
2023-09-11 14:57:57 +00:00
{
2023-09-13 08:03:11 +00:00
codexStarter = new CodexStarter(tools);
this.tools = tools;
2023-09-12 08:31:55 +00:00
}
public string LogPrefix => "(Codex) ";
2023-09-13 08:03:11 +00:00
public void Announce()
2023-09-12 08:31:55 +00:00
{
2023-09-13 14:06:05 +00:00
tools.GetLog().Log($"Loaded with Codex ID: '{codexStarter.GetCodexId()}'");
}
public void AddMetadata(IAddMetadata metadata)
{
metadata.Add("codexid", codexStarter.GetCodexId());
2023-09-11 14:57:57 +00:00
}
2023-09-13 08:03:11 +00:00
public void Decommission()
2023-09-12 08:31:55 +00:00
{
}
2023-09-11 14:57:57 +00:00
public RunningContainers[] StartCodexNodes(int numberOfNodes, Action<ICodexSetup> setup)
{
2023-09-20 06:45:55 +00:00
var codexSetup = GetSetup(numberOfNodes, setup);
2023-09-11 14:57:57 +00:00
return codexStarter.BringOnline(codexSetup);
}
2023-09-19 11:58:45 +00:00
public ICodexNodeGroup WrapCodexContainers(CoreInterface coreInterface, RunningContainers[] containers)
2023-09-11 14:57:57 +00:00
{
2023-09-20 08:13:29 +00:00
containers = containers.Select(c => SerializeGate.Gate(c)).ToArray();
return codexStarter.WrapCodexContainers(coreInterface, containers);
2023-09-11 14:57:57 +00:00
}
2023-09-20 06:45:55 +00:00
public void WireUpMarketplace(ICodexNodeGroup result, Action<ICodexSetup> setup)
{
var codexSetup = GetSetup(1, setup);
if (codexSetup.MarketplaceConfig == null) return;
var mconfig = codexSetup.MarketplaceConfig;
foreach (var node in result)
{
mconfig.GethNode.SendEth(node, mconfig.InitialEth);
mconfig.CodexContracts.MintTestTokens(mconfig.GethNode, node, mconfig.InitialTokens);
}
}
private CodexSetup GetSetup(int numberOfNodes, Action<ICodexSetup> setup)
{
var codexSetup = new CodexSetup(numberOfNodes);
codexSetup.LogLevel = defaultLogLevel;
setup(codexSetup);
return codexSetup;
}
2023-09-11 14:57:57 +00:00
}
}