static void Main(string[] args)
{
var root = Environment.GetEnvironmentVariable("INSIDE_CONTAINER") == "1" ? FindRoot("app")
: Path.GetFullPath(Path.Combine(FindRoot("docker-compose-generator"), ".."));
var composition = DockerComposition.FromEnvironmentVariables();
Console.WriteLine("Crypto: " + string.Join(", ", composition.SelectedCryptos.ToArray()));
Console.WriteLine("Lightning: " + composition.SelectedLN);
Console.WriteLine("ReverseProxy: " + composition.SelectedProxy);
var generatedLocation = Path.GetFullPath(Path.Combine(root, "Generated"));
var name = Environment.GetEnvironmentVariable("BTCPAYGEN_SUBNAME");
name = string.IsNullOrEmpty(name) ? "generated" : name;
try
{
new Program().Run(composition, name, generatedLocation);
}
catch (YamlBuildException ex)
{
ConsoleUtils.WriteLine(ex.Message, ConsoleColor.Red);
Environment.ExitCode = 1;
}
}
public void Build()
{
Console.WriteLine($"Generating {GetFilePath()}");
var deserializer = new DeserializerBuilder().Build();
var serializer = new SerializerBuilder().Build();
var fragmentsNotFound = new HashSet <FragmentName>();
var requiredFragments = new HashSet <FragmentName>();
var recommendedFragments = new HashSet <FragmentName>();
var processedFragments = new HashSet <FragmentName>();
var unprocessedFragments = new HashSet <FragmentName>();
var services = new List <KeyValuePair <YamlNode, YamlNode> >();
var volumes = new List <KeyValuePair <YamlNode, YamlNode> >();
var networks = new List <KeyValuePair <YamlNode, YamlNode> >();
var exclusives = new List <(FragmentName FragmentName, string Exclusivity)>();
var incompatibles = new List <(FragmentName FragmentName, string Exclusivity)>();
foreach (var fragment in Fragments.Where(NotExcluded))
{
unprocessedFragments.Add(fragment);
}
reprocessFragments:
foreach (var fragment in unprocessedFragments.ToList())
{
var fragmentPath = GetFragmentLocation(fragment);
if (!File.Exists(fragmentPath))
{
fragmentsNotFound.Add(fragment);
unprocessedFragments.Remove(fragment);
}
}
foreach (var o in unprocessedFragments.Select(f => (f, ParseDocument(f))).ToList())
{
var doc = o.Item2;
var fragment = o.f;
if (doc.Children.ContainsKey("services") && doc.Children["services"] is YamlMappingNode fragmentServicesRoot)
{
services.AddRange(fragmentServicesRoot.Children);
}
if (doc.Children.ContainsKey("volumes") && doc.Children["volumes"] is YamlMappingNode fragmentVolumesRoot)
{
volumes.AddRange(fragmentVolumesRoot.Children);
}
if (doc.Children.ContainsKey("networks") && doc.Children["networks"] is YamlMappingNode fragmentNetworksRoot)
{
networks.AddRange(fragmentNetworksRoot.Children);
}
if (doc.Children.ContainsKey("exclusive") && doc.Children["exclusive"] is YamlSequenceNode fragmentExclusiveRoot)
{
foreach (var node in fragmentExclusiveRoot)
{
exclusives.Add((fragment, node.ToString()));
}
}
if (doc.Children.ContainsKey("incompatible") && doc.Children["incompatible"] is YamlSequenceNode fragmentIncompatibleRoot)
{
foreach (var node in fragmentIncompatibleRoot)
{
incompatibles.Add((fragment, node.ToString()));
}
}
if (doc.Children.ContainsKey("required") && doc.Children["required"] is YamlSequenceNode fragmentRequireRoot)
{
foreach (var node in fragmentRequireRoot)
{
if (ExcludeFragments.Contains(new FragmentName(node.ToString())))
{
throw new YamlBuildException($"You excluded fragment {new FragmentName(node.ToString())} but it is required by {fragment}");
}
requiredFragments.Add(new FragmentName(node.ToString()));
}
}
if (doc.Children.ContainsKey("recommended") && doc.Children["recommended"] is YamlSequenceNode fragmentRecommendedRoot)
{
foreach (var node in fragmentRecommendedRoot)
{
if (!ExcludeFragments.Contains(new FragmentName(node.ToString())))
{
recommendedFragments.Add(new FragmentName(node.ToString()));
}
}
}
processedFragments.Add(fragment);
unprocessedFragments.Remove(fragment);
}
foreach (var fragment in requiredFragments.Concat(recommendedFragments).Where(f => !processedFragments.Contains(f)))
{
unprocessedFragments.Add(fragment);
}
if (unprocessedFragments.Count != 0)
{
goto reprocessFragments;
}
var exclusiveConflict = exclusives.GroupBy(e => e.Exclusivity)
.Where(e => e.Count() != 1)
.FirstOrDefault();
if (exclusiveConflict != null)
{
throw new YamlBuildException($"The fragments {String.Join(", ", exclusiveConflict.Select(e => e.FragmentName))} can't be used simultaneously (group '{exclusiveConflict.Key}')");
}
var groups = exclusives.ToDictionary(e => e.Exclusivity, e => e.FragmentName);
var incompatible = incompatibles
.Select(i => groups.TryGetValue(i.Exclusivity, out _) ? (i.FragmentName, i.Exclusivity) : (null, null))
.Where(i => i.Exclusivity != null)
.FirstOrDefault();
if (incompatible.Exclusivity != null)
{
throw new YamlBuildException($"The fragment {incompatible.FragmentName} is incompatible with '{incompatible.Exclusivity}'");
}
Console.WriteLine($"Selected fragments:");
foreach (var fragment in processedFragments)
{
Console.WriteLine($"\t{fragment}");
}
foreach (var fragment in fragmentsNotFound)
{
var fragmentPath = GetFragmentLocation(fragment);
ConsoleUtils.WriteLine($"\t{fragment} not found in {fragmentPath}, ignoring...", ConsoleColor.Yellow);
}
YamlMappingNode output = new YamlMappingNode();
output.Add("version", new YamlScalarNode("3")
{
Style = YamlDotNet.Core.ScalarStyle.DoubleQuoted
});
output.Add("services", new YamlMappingNode(Merge(services)));
output.Add("volumes", new YamlMappingNode(volumes));
output.Add("networks", new YamlMappingNode(networks));
PostProcess(output);
var dockerImages = ((YamlMappingNode)output["services"]).Children.Select(kv => kv.Value["image"].ToString()).ToList();
dockerImages.Add("btcpayserver/docker-compose-builder:1.24.1");
dockerImages.Add("btcpayserver/docker-compose-generator:latest");
StringBuilder pullImageSh = new StringBuilder();
pullImageSh.Append($"#!/bin/bash\n\n");
pullImageSh.Append($"# This script is automatically generated via the docker-compose generator and can be use to pull all required docker images \n");
foreach (var image in dockerImages)
{
pullImageSh.Append($"docker pull $BTCPAY_DOCKER_PULL_FLAGS \"{image}\"\n");
}
var outputFile = GetFilePath("pull-images.sh");
File.WriteAllText(outputFile, pullImageSh.ToString());
Console.WriteLine($"Generated {outputFile}");
StringBuilder saveImages = new StringBuilder();
saveImages.Append($"#!/bin/bash\n\n");
saveImages.Append($"# This script is automatically generated via the docker-compose generator and can be use to save the docker images in an archive \n");
saveImages.Append($"# ./save-images.sh output.tar \n");
saveImages.Append($"docker save -o \"$1\" \\\n {string.Join(" \\\n", dockerImages.Select(o => $"\"{o}\""))}");
outputFile = GetFilePath("save-images.sh");
File.WriteAllText(outputFile, saveImages.ToString());
Console.WriteLine($"Generated {outputFile}");
var result = serializer.Serialize(output);
outputFile = GetFilePath();
File.WriteAllText(outputFile, result.Replace("''", ""));
Console.WriteLine($"Generated {outputFile}");
Console.WriteLine();
}
