mirror of
https://github.com/Syping/ragephoto-cli.git
synced 2025-11-09 13:15:17 +01:00
split code to Commands class and Json class
This commit is contained in:
Syping
parent
cfa54e0217
commit
dccd33bcf4
3 changed files with 434 additions and 425 deletions
342
Commands.cs
Normal file
342
Commands.cs
Normal file
|
|
@ -0,0 +1,342 @@
|
|||
using System.CommandLine;
|
||||
using System.Text;
|
||||
namespace RagePhoto.Cli;
|
||||
|
||||
internal static class Commands {
|
||||
|
||||
internal static void Create(String format, String jpegFile, String outputFile, String? description, String? json, String? title) {
|
||||
try {
|
||||
using Photo photo = new();
|
||||
|
||||
photo.Format = format.ToLowerInvariant() switch {
|
||||
"gta5" => PhotoFormat.GTA5,
|
||||
"rdr2" => PhotoFormat.RDR2,
|
||||
_ => throw new ArgumentException("Invalid format", nameof(format))
|
||||
};
|
||||
|
||||
if (photo.Format == PhotoFormat.GTA5) {
|
||||
photo.SetHeader("PHOTO - 10/26/25 02:28:08", 798615001, 0);
|
||||
}
|
||||
else {
|
||||
photo.SetHeader("PHOTO - 10/26/25 02:31:34", 3077307752, 2901366738);
|
||||
}
|
||||
|
||||
if (jpegFile == String.Empty) {
|
||||
photo.Jpeg = Properties.Resources.EmptyJpeg;
|
||||
}
|
||||
else if (jpegFile != null) {
|
||||
using MemoryStream jpegStream = new();
|
||||
using Stream input = jpegFile == "-" ? Console.OpenStandardInput() : File.OpenRead(jpegFile);
|
||||
input.CopyTo(jpegStream);
|
||||
photo.Jpeg = jpegStream.ToArray();
|
||||
}
|
||||
|
||||
if (photo.Format == PhotoFormat.GTA5) {
|
||||
photo.Json = json == null ?
|
||||
Json.Initialize(PhotoFormat.GTA5, photo, Random.Shared.Next(),
|
||||
DateTimeOffset.FromUnixTimeSeconds(Random.Shared.Next(1356998400, 1388534399))) :
|
||||
Json.UpdateSign(photo, photo.Json);
|
||||
}
|
||||
else {
|
||||
photo.Json = json == null ?
|
||||
Json.Initialize(PhotoFormat.RDR2, photo, Random.Shared.Next(),
|
||||
DateTimeOffset.FromUnixTimeSeconds(Random.Shared.NextInt64(-2240524800, -2208988801))) :
|
||||
Json.UpdateSign(photo, photo.Json);
|
||||
}
|
||||
|
||||
photo.Description = description ?? String.Empty;
|
||||
photo.Title = title ?? "Custom Photo";
|
||||
|
||||
if (outputFile == "-" || outputFile == String.Empty) {
|
||||
using MemoryStream photoStream = new(photo.Save());
|
||||
using Stream output = Console.OpenStandardOutput();
|
||||
photoStream.CopyTo(output);
|
||||
}
|
||||
else {
|
||||
String tempFile = Path.GetTempFileName();
|
||||
photo.SaveFile(tempFile);
|
||||
File.Move(tempFile, outputFile, true);
|
||||
}
|
||||
}
|
||||
catch (RagePhotoException exception) {
|
||||
Console.Error.WriteLine(exception.Message);
|
||||
Environment.Exit(exception.Photo != null ? (Int32)exception.Error + 2 : -1);
|
||||
}
|
||||
catch (ArgumentException exception) {
|
||||
Console.Error.WriteLine(exception.Message);
|
||||
Environment.Exit(1);
|
||||
}
|
||||
catch (Exception exception) {
|
||||
Console.Error.WriteLine(exception.Message);
|
||||
Environment.Exit(-1);
|
||||
}
|
||||
}
|
||||
|
||||
internal static void Get(String inputFile, String dataType, String outputFile) {
|
||||
try {
|
||||
using Photo photo = new();
|
||||
|
||||
if (inputFile == "-" || inputFile == String.Empty) {
|
||||
using MemoryStream photoStream = new();
|
||||
using Stream input = Console.OpenStandardInput();
|
||||
input.CopyTo(photoStream);
|
||||
photo.Load(photoStream.ToArray());
|
||||
}
|
||||
else {
|
||||
photo.LoadFile(inputFile);
|
||||
}
|
||||
|
||||
Byte[] content = [];
|
||||
switch (dataType.ToLowerInvariant()) {
|
||||
case "d":
|
||||
case "description":
|
||||
content = Encoding.UTF8.GetBytes(photo.Description);
|
||||
break;
|
||||
case "f":
|
||||
case "format":
|
||||
content = Encoding.UTF8.GetBytes(photo.Format switch {
|
||||
PhotoFormat.GTA5 => "gta5",
|
||||
PhotoFormat.RDR2 => "rdr2",
|
||||
_ => "unknown"
|
||||
});
|
||||
break;
|
||||
case "i":
|
||||
case "image":
|
||||
case "jpeg":
|
||||
content = photo.Jpeg;
|
||||
break;
|
||||
case "j":
|
||||
case "json":
|
||||
content = Encoding.UTF8.GetBytes(photo.Json);
|
||||
break;
|
||||
case "s":
|
||||
case "sign":
|
||||
content = Encoding.UTF8.GetBytes($"{photo.Sign}");
|
||||
break;
|
||||
case "t":
|
||||
case "title":
|
||||
content = Encoding.UTF8.GetBytes(photo.Title);
|
||||
break;
|
||||
default:
|
||||
Console.Error.WriteLine($"Unknown Content Type: {dataType}");
|
||||
Environment.Exit(1);
|
||||
break;
|
||||
}
|
||||
|
||||
if (outputFile == "-" || outputFile == String.Empty) {
|
||||
using MemoryStream contentStream = new(content);
|
||||
using Stream output = Console.OpenStandardOutput();
|
||||
contentStream.CopyTo(output);
|
||||
}
|
||||
else {
|
||||
String tempFile = Path.GetTempFileName();
|
||||
using (MemoryStream contentStream = new(content)) {
|
||||
using FileStream output = File.Create(tempFile);
|
||||
contentStream.CopyTo(output);
|
||||
}
|
||||
File.Move(tempFile, outputFile, true);
|
||||
}
|
||||
}
|
||||
catch (RagePhotoException exception) {
|
||||
Console.Error.WriteLine(exception.Message);
|
||||
Environment.Exit(exception.Photo != null ? (Int32)exception.Error + 2 : -1);
|
||||
}
|
||||
catch (Exception exception) {
|
||||
Console.Error.WriteLine(exception.Message);
|
||||
Environment.Exit(-1);
|
||||
}
|
||||
}
|
||||
|
||||
internal static void Set(String inputFile, String? format, String? jpegFile, String? description, String? json, String? title, bool updateSign, String? outputFile) {
|
||||
if (format == null && jpegFile == null && description == null
|
||||
&& json == null && title == null && !updateSign) {
|
||||
Console.Error.WriteLine("No value has being set");
|
||||
Environment.Exit(1);
|
||||
}
|
||||
else if (inputFile == "-" && jpegFile == "-") {
|
||||
Console.Error.WriteLine("Multiple pipes are not supported");
|
||||
Environment.Exit(1);
|
||||
}
|
||||
|
||||
try {
|
||||
using Photo photo = new();
|
||||
|
||||
if (inputFile == "-" || inputFile == String.Empty) {
|
||||
using MemoryStream photoStream = new();
|
||||
using Stream input = Console.OpenStandardInput();
|
||||
input.CopyTo(photoStream);
|
||||
photo.Load(photoStream.ToArray());
|
||||
}
|
||||
else {
|
||||
photo.LoadFile(inputFile);
|
||||
}
|
||||
|
||||
if (format != null) {
|
||||
photo.Format = format.ToLowerInvariant() switch {
|
||||
"gta5" => PhotoFormat.GTA5,
|
||||
"rdr2" => PhotoFormat.RDR2,
|
||||
_ => throw new ArgumentException("Invalid format", nameof(format))
|
||||
};
|
||||
}
|
||||
|
||||
if (description != null)
|
||||
photo.Description = description;
|
||||
|
||||
if (json != null)
|
||||
photo.Json = json;
|
||||
|
||||
if (title != null)
|
||||
photo.Title = title;
|
||||
|
||||
if (jpegFile == String.Empty) {
|
||||
photo.Jpeg = Properties.Resources.EmptyJpeg;
|
||||
photo.Json = Json.UpdateSign(photo, photo.Json);
|
||||
}
|
||||
else if (jpegFile != null) {
|
||||
using MemoryStream jpegStream = new();
|
||||
using Stream input = jpegFile == "-" ? Console.OpenStandardInput() : File.OpenRead(jpegFile);
|
||||
input.CopyTo(jpegStream);
|
||||
photo.Jpeg = jpegStream.ToArray();
|
||||
photo.Json = Json.UpdateSign(photo, photo.Json);
|
||||
}
|
||||
else if (updateSign) {
|
||||
photo.Json = Json.UpdateSign(photo, photo.Json);
|
||||
}
|
||||
|
||||
if (outputFile == "-") {
|
||||
using MemoryStream photoStream = new(photo.Save());
|
||||
using Stream output = Console.OpenStandardOutput();
|
||||
photoStream.CopyTo(output);
|
||||
}
|
||||
else {
|
||||
String tempFile = Path.GetTempFileName();
|
||||
photo.SaveFile(tempFile);
|
||||
File.Move(tempFile, !String.IsNullOrEmpty(outputFile) ? outputFile : inputFile, true);
|
||||
}
|
||||
}
|
||||
catch (RagePhotoException exception) {
|
||||
Console.Error.WriteLine(exception.Message);
|
||||
Environment.Exit(exception.Photo != null ? (Int32)exception.Error + 2 : -1);
|
||||
}
|
||||
catch (ArgumentException exception) {
|
||||
Console.Error.WriteLine(exception.Message);
|
||||
Environment.Exit(1);
|
||||
}
|
||||
catch (Exception exception) {
|
||||
Console.Error.WriteLine(exception.Message);
|
||||
Environment.Exit(-1);
|
||||
}
|
||||
}
|
||||
|
||||
internal static Command CreateCommand {
|
||||
get {
|
||||
Argument<String> formatArgument = new("format") {
|
||||
Description = "Photo Format"
|
||||
};
|
||||
formatArgument.CompletionSources.Add(_ => [
|
||||
new("gta5"),
|
||||
new("rdr2")]);
|
||||
Argument<String> jpegArgument = new("jpeg") {
|
||||
Description = "JPEG File",
|
||||
DefaultValueFactory = _ => "-"
|
||||
};
|
||||
Argument<String> outputArgument = new("output") {
|
||||
Description = "Output File",
|
||||
DefaultValueFactory = _ => "-"
|
||||
};
|
||||
Option<String?> descriptionOption = new("--description", "-d") {
|
||||
Description = "Photo Description"
|
||||
};
|
||||
Option<String?> jsonOption = new("--json", "-j") {
|
||||
Description = "Photo JSON"
|
||||
};
|
||||
Option<String?> titleOption = new("--title", "-t") {
|
||||
Description = "Photo Title"
|
||||
};
|
||||
Command createCommand = new("create", "Create Photo") {
|
||||
formatArgument, jpegArgument, outputArgument, descriptionOption, jsonOption, titleOption
|
||||
};
|
||||
createCommand.SetAction(result => Create(
|
||||
result.GetRequiredValue(formatArgument),
|
||||
result.GetRequiredValue(jpegArgument),
|
||||
result.GetRequiredValue(outputArgument),
|
||||
result.GetValue(descriptionOption),
|
||||
result.GetValue(jsonOption),
|
||||
result.GetValue(titleOption)));
|
||||
return createCommand;
|
||||
}
|
||||
}
|
||||
|
||||
internal static Command GetCommand {
|
||||
get {
|
||||
Argument<String> inputArgument = new("input") {
|
||||
Description = "Input File"
|
||||
};
|
||||
Argument<String> dataTypeArgument = new("dataType") {
|
||||
Description = "Data Type",
|
||||
DefaultValueFactory = _ => "jpeg"
|
||||
};
|
||||
dataTypeArgument.CompletionSources.Add(_ => [
|
||||
new("description"),
|
||||
new("format"),
|
||||
new("jpeg"),
|
||||
new("json"),
|
||||
new("sign"),
|
||||
new("title")]);
|
||||
Option<String> outputOption = new("--output", "-o") {
|
||||
Description = "Output File",
|
||||
DefaultValueFactory = _ => "-"
|
||||
};
|
||||
Command getCommand = new("get", "Get Photo Data") {
|
||||
inputArgument, dataTypeArgument, outputOption
|
||||
};
|
||||
getCommand.SetAction(result => Get(
|
||||
result.GetRequiredValue(inputArgument),
|
||||
result.GetRequiredValue(dataTypeArgument),
|
||||
result.GetRequiredValue(outputOption)));
|
||||
return getCommand;
|
||||
}
|
||||
}
|
||||
|
||||
internal static Command SetCommand {
|
||||
get {
|
||||
Argument<String> inputArgument = new("input") {
|
||||
Description = "Input File"
|
||||
};
|
||||
Option<String?> formatOption = new("--format", "-f") {
|
||||
Description = "Photo Format"
|
||||
};
|
||||
Option<String?> jpegOption = new("--jpeg", "--image", "-i") {
|
||||
Description = "JPEG File"
|
||||
};
|
||||
Option<String?> descriptionOption = new("--description", "-d") {
|
||||
Description = "Photo Description"
|
||||
};
|
||||
Option<String?> jsonOption = new("--json", "-j") {
|
||||
Description = "Photo JSON"
|
||||
};
|
||||
Option<String?> titleOption = new("--title", "-t") {
|
||||
Description = "Photo Title"
|
||||
};
|
||||
Option<bool> updateSignOption = new("--update-sign", "-u") {
|
||||
Description = "Update Photo Signature"
|
||||
};
|
||||
Option<String?> outputOption = new("--output", "-o") {
|
||||
Description = "Output File"
|
||||
};
|
||||
Command setCommand = new("set", "Set Photo Data") {
|
||||
inputArgument, formatOption, jpegOption, descriptionOption, jsonOption, titleOption, updateSignOption, outputOption
|
||||
};
|
||||
setCommand.SetAction(result => Set(
|
||||
result.GetRequiredValue(inputArgument),
|
||||
result.GetValue(formatOption),
|
||||
result.GetValue(jpegOption),
|
||||
result.GetValue(descriptionOption),
|
||||
result.GetValue(jsonOption),
|
||||
result.GetValue(titleOption),
|
||||
result.GetValue(updateSignOption),
|
||||
result.GetValue(outputOption)));
|
||||
return setCommand;
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue