public Column(string name, DataTypeToken dataType, ValueToken defaultValue, bool isPrimaryKey, bool autogeneratePrimaryKey,
bool isNullable,
ForeignKeyDetailsToken foreignKey,
int?maxLength,
bool isUnique)
{
if (foreignKey != null && isPrimaryKey)
{
throw new ArgumentException($"A column cannot be both a primary and foreign key. Column: {Name}.");
}
Name = name?.ToPostgresStyle() ?? throw new ArgumentNullException(nameof(name));
CSharpStyleName = ColumnPlaceholder.GetCSharpName(name);
CSharpStyleNameLower = ColumnPlaceholder.GetCSharpName(name, true);
DataType = dataType ?? throw new ArgumentNullException(nameof(dataType));
DefaultValue = defaultValue;
IsPrimaryKey = isPrimaryKey;
AutogeneratePrimaryKey = autogeneratePrimaryKey;
IsNullable = isNullable;
ForeignKey = foreignKey;
MaxLength = maxLength;
IsUnique = isUnique;
}
public IReadOnlyList <GeneratedRepository> GetRepositories(IReadOnlyList <GeneratedClass> classes, GeneratorOptions options)
{
var builder = new StringBuilder();
var result = new List <GeneratedRepository>();
var connectionUsing = "using (var connection = new NpgsqlConnection(" + (options.UseUnderscore ? "_connectionString" : "connectionString") + "))";
var namespaceRequired = options.ClassNamespace != options.RepositoryNamespace;
foreach (var generatedClass in classes)
{
var name = GetQualifiedClassName(generatedClass);
foreach (var requiredNamespace in RequiredNamespaces)
{
builder.Append("using ").Append(requiredNamespace).AppendLine(";");
}
if (namespaceRequired)
{
builder.Append("using ").Append(options.ClassNamespace).AppendLine(";");
}
builder.AppendLine();
builder.Append("namespace ").AppendLine(options.RepositoryNamespace).AppendLine("{");
// Now inside the namespace
{
builder.AddTab(options).Append("public class ").Append(generatedClass.Name).Append(options.RepositorySuffix).AppendLine();
builder.AddTab(options).AppendLine("{");
// Now inside the class
{
AddConstructorAndField(generatedClass, options, builder);
builder.AppendLine();
// ReSharper disable UnusedVariable
using (var getAll = new GeneratedMethod($"IEnumerable<{name}>", GetAll, string.Empty, builder, options))
{
using (var conn = new GeneratedUsing(connectionUsing, builder, options))
{
builder.AddTab(options, 4)
.Append("var command = new NpgsqlCommand(@\"");
var statement = generatedClass.Sql.GetSelectAll();
AddWithAlignedIndentation(statement, 4, options, builder);
builder.AppendLine("\", connection);").AppendLine();
using (var reader = new GeneratedUsing("using (var reader = await command.ExecuteReaderAsync())", builder, options, 4))
{
builder.AddTab(options, 5).AppendLine("return Enumerate(reader);");
}
}
}
builder.AppendLine();
var pkCol = generatedClass.Table.GetPrimaryKey();
var pkParamName = pkCol.CSharpStyleNameLower;
var argByPk = pkCol.DataType.CSharpType + " " + pkParamName;
using (var get = new GeneratedMethod(name, Get, argByPk, builder, options))
{
using (var conn = new GeneratedUsing(connectionUsing, builder, options))
{
builder.AddTab(options, 4)
.Append("var command = new NpgsqlCommand(@\"");
var statement = generatedClass.Sql.GetSelectByPk();
AddWithAlignedIndentation(statement, 4, options, builder);
builder.AppendLine("\", connection);").AppendLine();
builder.AddTab(options, 4).Append("command.Parameters.AddWithValue(\"")
.Append(pkParamName)
.Append("\", ").Append(pkParamName).AppendLine(");").AppendLine();
using (var reader = new GeneratedUsing("using (var reader = await command.ExecuteReaderAsync())", builder, options, 4))
{
using (var mvNext = new GeneratedUsing("while (await reader.ReadAsync())", builder, options, 5))
{
builder.AddTab(options, 6).AppendLine("return GetCurrent(reader);");
}
}
builder.AppendLine();
builder.AddTab(options, 4).AppendLine("return null;");
}
}
builder.AppendLine();
using (var del = new GeneratedMethod("bool", Delete, argByPk, builder, options))
{
using (var conn = new GeneratedUsing(connectionUsing, builder, options))
{
builder.AddTab(options, 4)
.Append("var command = new NpgsqlCommand(@\"");
var statement = generatedClass.Sql.GetDeleteByPk();
AddWithAlignedIndentation(statement, 4, options, builder);
builder.AppendLine("\", connection);").AppendLine();
builder.AddTab(options, 4).Append("command.Parameters.AddWithValue(\"")
.Append(pkParamName)
.Append("\", ").Append(pkParamName).AppendLine(");");
builder.AppendLine().AddTab(options, 4).AppendLine("var result = await command.ExecuteNonQueryAsync();");
builder.AppendLine().AddTab(options, 4).AppendLine("return result > 0;");
}
}
builder.AppendLine();
var paramName = ColumnPlaceholder.GetCSharpName(generatedClass.Name, true);
using (var create = new GeneratedMethod(name, Create, $"{name} {paramName}", builder, options))
{
using (var conn = new GeneratedUsing(connectionUsing, builder, options))
{
builder.AddTab(options, 4)
.Append("var command = new NpgsqlCommand(@\"");
var statement = generatedClass.Sql.Create(options);
AddWithAlignedIndentation(statement.sql, 4, options, builder);
builder.AppendLine("\", connection);").AppendLine();
foreach (var param in statement.parameters)
{
builder.AddTab(options, 4).Append("command.Parameters.AddWithValue(\"")
.Append(param)
.Append("\", ").Append($"{paramName}.{ColumnPlaceholder.GetCSharpName(param)}").AppendLine(");");
}
builder.AppendLine();
using (var reader = new GeneratedUsing("using (var reader = await command.ExecuteReaderAsync())", builder, options, 4))
{
using (var mvNext = new GeneratedUsing("while (await reader.ReadAsync())", builder, options, 5))
{
builder.AddTab(options, 6).AppendLine("return GetCurrent(reader);");
}
}
builder.AppendLine();
builder.AddTab(options, 4).AppendLine("return null;");
}
}
builder.AppendLine();
using (var update = new GeneratedMethod(name, Update, $"{name} {paramName}", builder, options))
{
using (var conn = new GeneratedUsing(connectionUsing, builder, options))
{
builder.AddTab(options, 4)
.Append("var command = new NpgsqlCommand(@\"");
var statement = generatedClass.Sql.Update(options);
AddWithAlignedIndentation(statement.sql, 4, options, builder);
builder.AppendLine("\", connection);").AppendLine();
foreach (var param in statement.parameters)
{
builder.AddTab(options, 4).Append("command.Parameters.AddWithValue(\"")
.Append(param)
.Append("\", ").Append($"{paramName}.{ColumnPlaceholder.GetCSharpName(param)}").AppendLine(");");
}
builder.AppendLine();
using (var reader = new GeneratedUsing("using (var reader = await command.ExecuteReaderAsync())", builder, options, 4))
{
using (var mvNext = new GeneratedUsing("while (await reader.ReadAsync())", builder, options, 5))
{
builder.AddTab(options, 6).AppendLine("return GetCurrent(reader);");
}
}
builder.AppendLine();
builder.AddTab(options, 4).AppendLine("return null;");
}
}
builder.AppendLine();
using (var enumer = new GeneratedMethod($"IEnumerable<{name}>", "Enumerate", "DbDataReader reader", builder, options, "private", false, true))
{
using (var reader = new GeneratedUsing("while (reader.Read())", builder, options))
{
builder.AddTab(options, 4).AppendLine("yield return GetCurrent(reader);");
}
}
builder.AppendLine();
using (var enumer = new GeneratedMethod(name, "GetCurrent", "DbDataReader reader", builder, options, "private", false, true))
{
builder.AddTab(options, 3).Append("return new ").Append(name).AppendLine()
.AddTab(options, 3).AppendLine("{");
for (var i = 0; i < generatedClass.Table.Columns.Count; i++)
{
var col = generatedClass.Table.Columns[i];
var colName = ColumnPlaceholder.GetCSharpName(col.Name);
builder.AddTab(options, 4).Append(colName).Append(" = ")
.Append(GetReaderQueryForColumn(col, i));
if (i < generatedClass.Table.Columns.Count - 1)
{
builder.AppendLine(",");
}
else
{
builder.AppendLine();
}
}
builder.AddTab(options, 3).AppendLine("};");
}
}
// ReSharper restore UnusedVariable
builder.AddTab(options).AppendLine("}");
}
builder.Append('}');
var repoCode = builder.ToString();
result.Add(new GeneratedRepository(generatedClass.Table, repoCode, options.RepositoryNamespace));
builder.Clear();
}
return(result);
}
