public void DropColumn_ValidArguments()
{
Table.DropColumn("StringColumn");
Assert.AreEqual(3, Table.GetColumns().Count);
}
public override void PATCH()
{
// Get all required values
if (!Params.ContainsKey("table"))
{
Response.Send("Missing params", HttpStatusCode.BadRequest);
return;
}
bool FoundAdd = JSON.TryGetValue <JObject>("Add", out JToken AddValue);
bool FoundDelete = JSON.TryGetValue <JArray>("Delete", out JToken DeleteValue);
bool FoundRename = JSON.TryGetValue <JObject>("Rename", out JToken RenameValue);
if (!FoundAdd && !FoundDelete && !FoundRename)
{
Response.Send("Missing fields", HttpStatusCode.BadRequest);
return;
}
//Check if table exists
GenericDataTable Table = GenericDataTable.GetTableByName(Connection, Params["table"][0]);
if (Table == null)
{
Response.Send("No such table", HttpStatusCode.NotFound);
return;
}
Dictionary <string, DataType> ToAdd = new Dictionary <string, DataType>();
List <string> ToDelete = new List <string>();
Dictionary <string, string> ToRename = new Dictionary <string, string>();
Dictionary <string, DataType> Columns = Table.GetColumns();
//Add all specified columns, if any;
if (FoundAdd)
{
JObject Add = (JObject)AddValue;
foreach (KeyValuePair <string, JToken> Entry in Add)
{
if (Entry.Key.ToLower() == "validated")
{
Table.AddValidatedColumn();
continue;
}
if (Columns.ContainsKey(Entry.Key) ||
!Regex.IsMatch(Entry.Key, GenericDataTable.RX) ||
((string)Entry.Key).ToLower() == "rowid"
)
{
Response.Send("Invalid entry in Add (" + Entry.Key + ")", HttpStatusCode.BadRequest);
return;
}
if (!Enum.TryParse((string)Entry.Value, out DataType DT))
{
Response.Send("Invalid type", HttpStatusCode.BadRequest);
return;
}
ToAdd.Add(Entry.Key, DT);
}
}
//Delete all specified columns, if any;
if (FoundDelete)
{
JArray Delete = (JArray)DeleteValue;
foreach (JToken Entry in Delete)
{
if (Entry.Type != JTokenType.String ||
!Columns.ContainsKey((string)Entry) ||
((string)Entry).ToLower() == "rowid"
)
{
Response.Send("Invalid entry in Delete (" + Entry + ")", HttpStatusCode.BadRequest);
return;
}
ToDelete.Add((string)Entry);
}
}
//Rename all specified columns, if any;
if (FoundRename)
{
JObject Rename = (JObject)RenameValue;
foreach (KeyValuePair <string, JToken> Entry in Rename)
{
if (
Entry.Value.Type != JTokenType.String ||
((string)Entry.Value).ToLower() == "validated" ||
((string)Entry.Value).ToLower() == "rowid" ||
((string)Entry.Key).ToLower() == "validated" ||
((string)Entry.Key).ToLower() == "rowid" ||
Columns.ContainsKey((string)Entry.Value) ||
!Columns.ContainsKey(Entry.Key) ||
!Regex.IsMatch((string)Entry.Value, GenericDataTable.RX)
)
{
Response.Send("Invalid entry in Rename (" + Entry.Key + ")", HttpStatusCode.BadRequest);
return;
}
ToRename.Add(Entry.Key, (string)Entry.Value);
}
}
if (ToAdd.Count > 0)
{
Table.AddColumn(ToAdd);
}
if (ToDelete.Count > 0)
{
Table.DropColumn(ToDelete);
}
if (ToRename.Count > 0)
{
Table.RenameColumn(ToRename);
}
Response.Send(HttpStatusCode.OK);
}