mirror of
https://github.com/sochix/TLSharp.git
synced 2025-12-06 08:02:00 +01:00
486 lines
21 KiB
C#
486 lines
21 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using Newtonsoft.Json;
|
|
|
|
namespace TeleSharp.Generator
|
|
{
|
|
internal class Program
|
|
{
|
|
private static readonly List<string> keywords = new List<string>(new[]
|
|
{
|
|
"abstract", "as", "base", "bool", "break", "byte", "case", "catch", "char", "checked", "class", "const",
|
|
"continue", "decimal", "default", "delegate", "do", "double", "else", "enum", "event", "explicit", "extern",
|
|
"false", "finally", "fixed", "float", "for", "foreach", "goto", "if", "implicit", "in", "in", "int",
|
|
"interface", "internal", "is", "lock", "long", "namespace", "new", "null", "object", "operator", "out",
|
|
"out", "override", "params", "private", "protected", "public", "readonly", "ref", "return", "sbyte",
|
|
"sealed", "short", "sizeof", "stackalloc", "static", "string", "struct", "switch", "this", "throw", "true",
|
|
"try", "typeof", "uint", "ulong", "unchecked", "unsafe", "ushort", "using", "virtual", "void", "volatile",
|
|
"while", "add", "alias", "ascending", "async", "await", "descending", "dynamic", "from", "get", "global",
|
|
"group", "into", "join", "let", "orderby", "partial", "partial", "remove", "select", "set", "value", "var",
|
|
"where", "where", "yield"
|
|
});
|
|
|
|
private static readonly List<string> interfacesList = new List<string>();
|
|
private static readonly List<string> classesList = new List<string>();
|
|
|
|
private static void Main(string[] args)
|
|
{
|
|
var AbsStyle = File.ReadAllText("ConstructorAbs.tmp");
|
|
var NormalStyle = File.ReadAllText("Constructor.tmp");
|
|
var MethodStyle = File.ReadAllText("Method.tmp");
|
|
//string method = File.ReadAllText("constructor.tt");
|
|
var Json = "";
|
|
string url;
|
|
if (args.Count() == 0) url = "tl-schema.json";
|
|
else url = args[0];
|
|
|
|
Json = File.ReadAllText(url);
|
|
var file = File.OpenWrite("Result.cs");
|
|
var sw = new StreamWriter(file);
|
|
Schema schema = JsonConvert.DeserializeObject<Schema>(Json);
|
|
foreach (var c in schema.constructors)
|
|
{
|
|
interfacesList.Add(c.type);
|
|
classesList.Add(c.predicate);
|
|
}
|
|
foreach (var c in schema.constructors)
|
|
{
|
|
var list = schema.constructors.Where(x => x.type == c.type);
|
|
if (list.Count() > 1)
|
|
{
|
|
var path = (GetNameSpace(c.type).Replace("TeleSharp.TL", "TL\\").Replace(".", "") + "\\" +
|
|
GetNameofClass(c.type, true) + ".cs").Replace("\\\\", "\\");
|
|
var classFile = MakeFile(path);
|
|
using (var writer = new StreamWriter(classFile))
|
|
{
|
|
var nspace = GetNameSpace(c.type)
|
|
.Replace("TeleSharp.TL", "TL\\")
|
|
.Replace(".", "")
|
|
.Replace("\\\\", "\\")
|
|
.Replace("\\", ".");
|
|
if (nspace.EndsWith("."))
|
|
nspace = nspace.Remove(nspace.Length - 1, 1);
|
|
var temp = AbsStyle.Replace("/* NAMESPACE */", "TeleSharp." + nspace);
|
|
temp = temp.Replace("/* NAME */", GetNameofClass(c.type, true));
|
|
writer.Write(temp);
|
|
writer.Close();
|
|
classFile.Close();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
interfacesList.Remove(list.First().type);
|
|
list.First().type = "himself";
|
|
}
|
|
}
|
|
foreach (var c in schema.constructors)
|
|
{
|
|
var path = (GetNameSpace(c.predicate).Replace("TeleSharp.TL", "TL\\").Replace(".", "") + "\\" +
|
|
GetNameofClass(c.predicate, false) + ".cs").Replace("\\\\", "\\");
|
|
var classFile = MakeFile(path);
|
|
using (var writer = new StreamWriter(classFile))
|
|
{
|
|
#region About Class
|
|
|
|
var nspace = GetNameSpace(c.predicate)
|
|
.Replace("TeleSharp.TL", "TL\\")
|
|
.Replace(".", "")
|
|
.Replace("\\\\", "\\")
|
|
.Replace("\\", ".");
|
|
if (nspace.EndsWith("."))
|
|
nspace = nspace.Remove(nspace.Length - 1, 1);
|
|
var temp = NormalStyle.Replace("/* NAMESPACE */", "TeleSharp." + nspace);
|
|
temp = c.type == "himself"
|
|
? temp.Replace("/* PARENT */", "TLObject")
|
|
: temp.Replace("/* PARENT */", GetNameofClass(c.type, true));
|
|
temp = temp.Replace("/*Constructor*/", c.id.ToString());
|
|
temp = temp.Replace("/* NAME */", GetNameofClass(c.predicate, false));
|
|
|
|
#endregion
|
|
|
|
#region Fields
|
|
|
|
var fields = "";
|
|
foreach (var tmp in c.Params)
|
|
fields +=
|
|
$" public {CheckForFlagBase(tmp.type, GetTypeName(tmp.type))} {CheckForKeyword(tmp.name)} " +
|
|
"{get;set;}" + Environment.NewLine;
|
|
temp = temp.Replace("/* PARAMS */", fields);
|
|
|
|
#endregion
|
|
|
|
#region ComputeFlagFunc
|
|
|
|
if (!c.Params.Any(x => x.name == "flags"))
|
|
{
|
|
temp = temp.Replace("/* COMPUTE */", "");
|
|
}
|
|
else
|
|
{
|
|
var compute = "flags = 0;" + Environment.NewLine;
|
|
foreach (var param in c.Params.Where(x => IsFlagBase(x.type)))
|
|
if (IsTrueFlag(param.type))
|
|
compute +=
|
|
$"flags = {CheckForKeyword(param.name)} ? (flags | {GetBitMask(param.type)}) : (flags & ~{GetBitMask(param.type)});" +
|
|
Environment.NewLine;
|
|
else
|
|
compute +=
|
|
$"flags = {CheckForKeyword(param.name)} != null ? (flags | {GetBitMask(param.type)}) : (flags & ~{GetBitMask(param.type)});" +
|
|
Environment.NewLine;
|
|
temp = temp.Replace("/* COMPUTE */", compute);
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region SerializeFunc
|
|
|
|
var serialize = "";
|
|
|
|
if (c.Params.Any(x => x.name == "flags"))
|
|
serialize += "ComputeFlags();" + Environment.NewLine + "bw.Write(flags);" + Environment.NewLine;
|
|
foreach (var p in c.Params.Where(x => x.name != "flags"))
|
|
serialize += WriteWriteCode(p) + Environment.NewLine;
|
|
temp = temp.Replace("/* SERIALIZE */", serialize);
|
|
|
|
#endregion
|
|
|
|
#region DeSerializeFunc
|
|
|
|
var deserialize = "";
|
|
|
|
foreach (var p in c.Params)
|
|
deserialize += WriteReadCode(p) + Environment.NewLine;
|
|
temp = temp.Replace("/* DESERIALIZE */", deserialize);
|
|
|
|
#endregion
|
|
|
|
writer.Write(temp);
|
|
writer.Close();
|
|
classFile.Close();
|
|
}
|
|
}
|
|
foreach (var c in schema.methods)
|
|
{
|
|
var path = (GetNameSpace(c.method).Replace("TeleSharp.TL", "TL\\").Replace(".", "") + "\\" +
|
|
GetNameofClass(c.method, false, true) + ".cs").Replace("\\\\", "\\");
|
|
var classFile = MakeFile(path);
|
|
using (var writer = new StreamWriter(classFile))
|
|
{
|
|
#region About Class
|
|
|
|
var nspace = GetNameSpace(c.method)
|
|
.Replace("TeleSharp.TL", "TL\\")
|
|
.Replace(".", "")
|
|
.Replace("\\\\", "\\")
|
|
.Replace("\\", ".");
|
|
if (nspace.EndsWith("."))
|
|
nspace = nspace.Remove(nspace.Length - 1, 1);
|
|
var temp = MethodStyle.Replace("/* NAMESPACE */", "TeleSharp." + nspace);
|
|
temp = temp.Replace("/* PARENT */", "TLMethod");
|
|
temp = temp.Replace("/*Constructor*/", c.id.ToString());
|
|
temp = temp.Replace("/* NAME */", GetNameofClass(c.method, false, true));
|
|
|
|
#endregion
|
|
|
|
#region Fields
|
|
|
|
var fields = "";
|
|
foreach (var tmp in c.Params)
|
|
fields +=
|
|
$" public {CheckForFlagBase(tmp.type, GetTypeName(tmp.type))} {CheckForKeyword(tmp.name)} " +
|
|
"{get;set;}" + Environment.NewLine;
|
|
fields += $" public {CheckForFlagBase(c.type, GetTypeName(c.type))} Response" +
|
|
"{ get; set;}" + Environment.NewLine;
|
|
temp = temp.Replace("/* PARAMS */", fields);
|
|
|
|
#endregion
|
|
|
|
#region ComputeFlagFunc
|
|
|
|
if (!c.Params.Any(x => x.name == "flags"))
|
|
{
|
|
temp = temp.Replace("/* COMPUTE */", "");
|
|
}
|
|
else
|
|
{
|
|
var compute = "flags = 0;" + Environment.NewLine;
|
|
foreach (var param in c.Params.Where(x => IsFlagBase(x.type)))
|
|
if (IsTrueFlag(param.type))
|
|
compute +=
|
|
$"flags = {CheckForKeyword(param.name)} ? (flags | {GetBitMask(param.type)}) : (flags & ~{GetBitMask(param.type)});" +
|
|
Environment.NewLine;
|
|
else
|
|
compute +=
|
|
$"flags = {CheckForKeyword(param.name)} != null ? (flags | {GetBitMask(param.type)}) : (flags & ~{GetBitMask(param.type)});" +
|
|
Environment.NewLine;
|
|
temp = temp.Replace("/* COMPUTE */", compute);
|
|
}
|
|
|
|
#endregion
|
|
|
|
#region SerializeFunc
|
|
|
|
var serialize = "";
|
|
|
|
if (c.Params.Any(x => x.name == "flags"))
|
|
serialize += "ComputeFlags();" + Environment.NewLine + "bw.Write(flags);" + Environment.NewLine;
|
|
foreach (var p in c.Params.Where(x => x.name != "flags"))
|
|
serialize += WriteWriteCode(p) + Environment.NewLine;
|
|
temp = temp.Replace("/* SERIALIZE */", serialize);
|
|
|
|
#endregion
|
|
|
|
#region DeSerializeFunc
|
|
|
|
var deserialize = "";
|
|
|
|
foreach (var p in c.Params)
|
|
deserialize += WriteReadCode(p) + Environment.NewLine;
|
|
temp = temp.Replace("/* DESERIALIZE */", deserialize);
|
|
|
|
#endregion
|
|
|
|
#region DeSerializeRespFunc
|
|
|
|
var deserializeResp = "";
|
|
var p2 = new Param {name = "Response", type = c.type};
|
|
deserializeResp += WriteReadCode(p2) + Environment.NewLine;
|
|
temp = temp.Replace("/* DESERIALIZEResp */", deserializeResp);
|
|
|
|
#endregion
|
|
|
|
writer.Write(temp);
|
|
writer.Close();
|
|
classFile.Close();
|
|
}
|
|
}
|
|
}
|
|
|
|
public static string FormatName(string input)
|
|
{
|
|
if (string.IsNullOrEmpty(input))
|
|
throw new ArgumentException("ARGH!");
|
|
if (input.IndexOf('.') != -1)
|
|
{
|
|
input = input.Replace(".", " ");
|
|
var temp = "";
|
|
foreach (var s in input.Split(' '))
|
|
temp += FormatName(s) + " ";
|
|
input = temp.Trim();
|
|
}
|
|
return input.First().ToString().ToUpper() + input.Substring(1);
|
|
}
|
|
|
|
public static string CheckForKeyword(string name)
|
|
{
|
|
if (keywords.Contains(name)) return "@" + name;
|
|
return name;
|
|
}
|
|
|
|
public static string GetNameofClass(string type, bool isinterface = false, bool ismethod = false)
|
|
{
|
|
if (!ismethod)
|
|
if (type.IndexOf('.') != -1 && type.IndexOf('?') == -1)
|
|
return isinterface
|
|
? "TLAbs" + FormatName(type.Split('.')[1])
|
|
: "TL" + FormatName(type.Split('.')[1]);
|
|
else if (type.IndexOf('.') != -1 && type.IndexOf('?') != -1)
|
|
return isinterface
|
|
? "TLAbs" + FormatName(type.Split('?')[1])
|
|
: "TL" + FormatName(type.Split('?')[1]);
|
|
else
|
|
return isinterface ? "TLAbs" + FormatName(type) : "TL" + FormatName(type);
|
|
if (type.IndexOf('.') != -1 && type.IndexOf('?') == -1)
|
|
return "TLRequest" + FormatName(type.Split('.')[1]);
|
|
if (type.IndexOf('.') != -1 && type.IndexOf('?') != -1)
|
|
return "TLRequest" + FormatName(type.Split('?')[1]);
|
|
return "TLRequest" + FormatName(type);
|
|
}
|
|
|
|
private static bool IsFlagBase(string type)
|
|
{
|
|
return type.IndexOf("?") != -1;
|
|
}
|
|
|
|
private static int GetBitMask(string type)
|
|
{
|
|
return (int) Math.Pow(2, int.Parse(type.Split('?')[0].Split('.')[1]));
|
|
}
|
|
|
|
private static bool IsTrueFlag(string type)
|
|
{
|
|
return type.Split('?')[1] == "true";
|
|
}
|
|
|
|
public static string GetNameSpace(string type)
|
|
{
|
|
if (type.IndexOf('.') != -1)
|
|
return "TeleSharp.TL" + FormatName(type.Split('.')[0]);
|
|
return "TeleSharp.TL";
|
|
}
|
|
|
|
public static string CheckForFlagBase(string type, string result)
|
|
{
|
|
if (type.IndexOf('?') == -1)
|
|
return result;
|
|
var innerType = type.Split('?')[1];
|
|
if (innerType == "true") return result;
|
|
if (new[] {"bool", "int", "uint", "long", "double"}.Contains(result)) return result + "?";
|
|
return result;
|
|
}
|
|
|
|
public static string GetTypeName(string type)
|
|
{
|
|
switch (type.ToLower())
|
|
{
|
|
case "#":
|
|
case "int":
|
|
return "int";
|
|
case "uint":
|
|
return "uint";
|
|
case "long":
|
|
return "long";
|
|
case "double":
|
|
return "double";
|
|
case "string":
|
|
return "string";
|
|
case "bytes":
|
|
return "byte[]";
|
|
case "true":
|
|
case "bool":
|
|
return "bool";
|
|
case "!x":
|
|
return "TLObject";
|
|
case "x":
|
|
return "TLObject";
|
|
}
|
|
|
|
if (type.StartsWith("Vector"))
|
|
return "TLVector<" + GetTypeName(type.Replace("Vector<", "").Replace(">", "")) + ">";
|
|
|
|
if (type.ToLower().Contains("inputcontact"))
|
|
return "TLInputPhoneContact";
|
|
|
|
|
|
if (type.IndexOf('.') != -1 && type.IndexOf('?') == -1)
|
|
if (interfacesList.Any(x => x.ToLower() == type.ToLower()))
|
|
return FormatName(type.Split('.')[0]) + "." + "TLAbs" + type.Split('.')[1];
|
|
else if (classesList.Any(x => x.ToLower() == type.ToLower()))
|
|
return FormatName(type.Split('.')[0]) + "." + "TL" + type.Split('.')[1];
|
|
else
|
|
return FormatName(type.Split('.')[1]);
|
|
if (type.IndexOf('?') == -1)
|
|
if (interfacesList.Any(x => x.ToLower() == type.ToLower()))
|
|
return "TLAbs" + type;
|
|
else if (classesList.Any(x => x.ToLower() == type.ToLower()))
|
|
return "TL" + type;
|
|
else
|
|
return type;
|
|
return GetTypeName(type.Split('?')[1]);
|
|
}
|
|
|
|
public static string LookTypeInLists(string src)
|
|
{
|
|
if (interfacesList.Any(x => x.ToLower() == src.ToLower()))
|
|
return "TLAbs" + FormatName(src);
|
|
if (classesList.Any(x => x.ToLower() == src.ToLower()))
|
|
return "TL" + FormatName(src);
|
|
return src;
|
|
}
|
|
|
|
public static string WriteWriteCode(Param p, bool flag = false)
|
|
{
|
|
switch (p.type.ToLower())
|
|
{
|
|
case "#":
|
|
case "int":
|
|
return flag
|
|
? $"bw.Write({CheckForKeyword(p.name)}.Value);"
|
|
: $"bw.Write({CheckForKeyword(p.name)});";
|
|
case "long":
|
|
return flag
|
|
? $"bw.Write({CheckForKeyword(p.name)}.Value);"
|
|
: $"bw.Write({CheckForKeyword(p.name)});";
|
|
case "string":
|
|
return $"StringUtil.Serialize({CheckForKeyword(p.name)},bw);";
|
|
case "bool":
|
|
return flag
|
|
? $"BoolUtil.Serialize({CheckForKeyword(p.name)}.Value,bw);"
|
|
: $"BoolUtil.Serialize({CheckForKeyword(p.name)},bw);";
|
|
case "true":
|
|
return $"BoolUtil.Serialize({CheckForKeyword(p.name)},bw);";
|
|
case "bytes":
|
|
return $"BytesUtil.Serialize({CheckForKeyword(p.name)},bw);";
|
|
case "double":
|
|
return flag
|
|
? $"bw.Write({CheckForKeyword(p.name)}.Value);"
|
|
: $"bw.Write({CheckForKeyword(p.name)});";
|
|
default:
|
|
if (!IsFlagBase(p.type))
|
|
{
|
|
return $"ObjectUtils.SerializeObject({CheckForKeyword(p.name)},bw);";
|
|
}
|
|
else
|
|
{
|
|
if (IsTrueFlag(p.type))
|
|
return $"";
|
|
var p2 = new Param {name = p.name, type = p.type.Split('?')[1]};
|
|
return $"if ((flags & {GetBitMask(p.type)}) != 0)" + Environment.NewLine +
|
|
WriteWriteCode(p2, true);
|
|
}
|
|
}
|
|
}
|
|
|
|
public static string WriteReadCode(Param p)
|
|
{
|
|
switch (p.type.ToLower())
|
|
{
|
|
case "#":
|
|
case "int":
|
|
return $"{CheckForKeyword(p.name)} = br.ReadInt32();";
|
|
case "long":
|
|
return $"{CheckForKeyword(p.name)} = br.ReadInt64();";
|
|
case "string":
|
|
return $"{CheckForKeyword(p.name)} = StringUtil.Deserialize(br);";
|
|
case "bool":
|
|
case "true":
|
|
return $"{CheckForKeyword(p.name)} = BoolUtil.Deserialize(br);";
|
|
case "bytes":
|
|
return $"{CheckForKeyword(p.name)} = BytesUtil.Deserialize(br);";
|
|
case "double":
|
|
return $"{CheckForKeyword(p.name)} = br.ReadDouble();";
|
|
default:
|
|
if (!IsFlagBase(p.type))
|
|
{
|
|
if (p.type.ToLower().Contains("vector"))
|
|
return
|
|
$"{CheckForKeyword(p.name)} = ({GetTypeName(p.type)})ObjectUtils.DeserializeVector<{GetTypeName(p.type).Replace("TLVector<", "").Replace(">", "")}>(br);";
|
|
return $"{CheckForKeyword(p.name)} = ({GetTypeName(p.type)})ObjectUtils.DeserializeObject(br);";
|
|
}
|
|
else
|
|
{
|
|
if (IsTrueFlag(p.type))
|
|
{
|
|
return $"{CheckForKeyword(p.name)} = (flags & {GetBitMask(p.type)}) != 0;";
|
|
}
|
|
var p2 = new Param {name = p.name, type = p.type.Split('?')[1]};
|
|
return $"if ((flags & {GetBitMask(p.type)}) != 0)" + Environment.NewLine +
|
|
WriteReadCode(p2) + Environment.NewLine +
|
|
"else" + Environment.NewLine +
|
|
$"{CheckForKeyword(p.name)} = null;" + Environment.NewLine;
|
|
}
|
|
}
|
|
}
|
|
|
|
public static FileStream MakeFile(string path)
|
|
{
|
|
if (!Directory.Exists(Path.GetDirectoryName(path)))
|
|
Directory.CreateDirectory(Path.GetDirectoryName(path));
|
|
if (File.Exists(path))
|
|
File.Delete(path);
|
|
return File.OpenWrite(path);
|
|
}
|
|
}
|
|
} |