~ruther/Nostale-Anonymizer

ref: e12e5db0075652fac50d16997d9902284ef535fb Nostale-Anonymizer/src/Anonymizer.CLI/Program.cs -rw-r--r-- 3.5 KiB
e12e5db0 — František Boháček feat: add support for output directory, matching the name of input 2 years ago
                                                                                
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
//
//  Program.cs
//
//  Copyright (c) František Boháček. All rights reserved.
//  Licensed under the MIT license. See LICENSE file in the project root for full license information.

using System;
using Anonymizer.Extensions;
using Anonymizer.Filters;
using Anonymizer.Sinks;
using CommandLine;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Options;
using NosSmooth.PacketSerializer.Extensions;
using NosSmooth.PacketSerializer.Packets;

namespace Anonymizer.CLI;

/// <summary>
/// A main class.
/// </summary>
public static class Program
{
    /// <summary>
    /// A main method.
    /// </summary>
    /// <param name="args">The cli arguments.</param>
    /// <returns>The exit code.</returns>
    public static int Main(string[] args)
    {
        return Parser.Default.ParseArguments<Options>(args)
            .MapResult
            (
                options =>
                {
                    var directoryInfo = new DirectoryInfo(options.Output);
                    var fileInfo = new FileInfo(options.Output);
                    if (Directory.Exists(options.Output))
                    {
                        fileInfo = null;
                    }
                    else
                    {
                        directoryInfo = null;
                    }

                    return MakeAsync
                        (
                            new FileInfo(options.Input),
                            directoryInfo,
                            fileInfo,
                            new FileInfo(options.Config)
                        )
                        .GetAwaiter().GetResult();
                },
                _ => 1
            );
    }

    private static async Task<int> MakeAsync
    (
        FileInfo input,
        DirectoryInfo? outputDirectory,
        FileInfo? outputFile,
        FileInfo config
    )
    {
        if (!input.Exists)
        {
            Console.Error.WriteLine("The input file does not exist.");
            return 1;
        }

        if (outputFile is null)
        {
            outputFile = new FileInfo(Path.Join(outputDirectory!.FullName, input.Name));
        }

        if (outputFile.Exists)
        {
            Console.Error.WriteLine("The output file already exists.");
            return 1;
        }

        if (!config.Exists)
        {
            Console.Error.WriteLine($"The config file {config.FullName} does not exist.");
            return 1;
        }

        IConfigurationRoot configuration = new ConfigurationBuilder()
            .AddJsonFile(config.FullName)
            .Build();

        var services = new ServiceCollection()
            .Configure<HeaderFilterOptions>(configuration.GetSection("HeaderFilter"))
            .Configure<FileSinkOptions>(configuration.GetSection("PacketFile"))
            .AddPacketSerialization()
            .AddAnonymizer()
            .BuildServiceProvider();

        services.GetRequiredService<IPacketTypesRepository>().AddDefaultPackets();
        var processor = services.GetRequiredService<PacketProcessor>();
        using var fileSink = new FileSink
        (
            input.FullName,
            outputFile.FullName,
            services.GetRequiredService<IOptions<FileSinkOptions>>().Value
        );

        var result = await processor.ProcessSourceDestination(fileSink, fileSink);
        if (!result.IsSuccess)
        {
            Console.WriteLine(result.Error);
        }

        return 0;
    }
}
Do not follow this link