add array support to PostgreSQLDataType + add RETURNING * support

Author: tanner0101

Sources/PostgreSQL/Column/PostgreSQLColumnType.swift

@@ -17,7 +17,7 @@ extension UUID: PostgreSQLDataTypeStaticRepresentable {
 
 extension Date: PostgreSQLDataTypeStaticRepresentable {
     /// See `PostgreSQLDataTypeStaticRepresentable`.
-    public static var postgreSQLDataType: PostgreSQLDataType { return .timestamp(nil) }
+    public static var postgreSQLDataType: PostgreSQLDataType { return .timestamp }
 }
 
 extension Int: PostgreSQLDataTypeStaticRepresentable {
@@ -27,7 +27,7 @@ extension Int: PostgreSQLDataTypeStaticRepresentable {
 
 extension Int8: PostgreSQLDataTypeStaticRepresentable  {
     /// See `PostgreSQLDataTypeStaticRepresentable`.
-    public static var postgreSQLDataType: PostgreSQLDataType { return .char(nil) }
+    public static var postgreSQLDataType: PostgreSQLDataType { return .char }
 }
 
 extension Int16: PostgreSQLDataTypeStaticRepresentable {
@@ -52,7 +52,7 @@ extension UInt: PostgreSQLDataTypeStaticRepresentable {
 
 extension UInt8: PostgreSQLDataTypeStaticRepresentable {
     /// See `PostgreSQLDataTypeStaticRepresentable`.
-    public static var postgreSQLDataType: PostgreSQLDataType { return .char(nil) }
+    public static var postgreSQLDataType: PostgreSQLDataType { return .char }
 }
 
 extension UInt16: PostgreSQLDataTypeStaticRepresentable {

Sources/PostgreSQL/SQL/PostgreSQLDataTypeStaticRepresentable.swift

@@ -59,11 +59,6 @@ public typealias PostgreSQLGroupBy = GenericSQLGroupBy<PostgreSQLExpression>
 /// See `SQLQuery`.
 public typealias PostgreSQLIdentifier = GenericSQLIdentifier
 
-/// See `SQLQuery`.
-public typealias PostgreSQLInsert = GenericSQLInsert<
-    PostgreSQLTableIdentifier, PostgreSQLColumnIdentifier, PostgreSQLExpression
->
-
 /// See `SQLQuery`.
 public typealias PostgreSQLJoin = GenericSQLJoin<
     PostgreSQLJoinMethod, PostgreSQLTableIdentifier, PostgreSQLExpression

Sources/PostgreSQL/SQL/PostgreSQLGeneric.swift

@@ -0,0 +1,49 @@
+public struct PostgreSQLInsert: SQLInsert {
+    /// See `SQLInsert`.
+    public static func insert(_ table: PostgreSQLTableIdentifier) -> PostgreSQLInsert {
+        return .init(insert: .insert(table), returning: [])
+    }
+    
+    /// See `SQLInsert`.
+    public typealias TableIdentifier = PostgreSQLTableIdentifier
+    
+    /// See `SQLInsert`.
+    public typealias ColumnIdentifier = PostgreSQLColumnIdentifier
+    
+    /// See `SQLInsert`.
+    public typealias Expression = PostgreSQLExpression
+    
+    /// Root insert statement.
+    private var insert: GenericSQLInsert<TableIdentifier, ColumnIdentifier, Expression>
+
+    /// `RETURNING *`
+    public var returning: [PostgreSQLSelectExpression]
+    
+    /// See `SQLInsert`.
+    public var columns: [PostgreSQLColumnIdentifier] {
+        get { return insert.columns }
+        set { insert.columns = newValue }
+    }
+    
+    /// See `SQLInsert`.
+    public var values: [[PostgreSQLExpression]] {
+        get { return insert.values }
+        set { insert.values = newValue}
+    }
+    
+    /// See `SQLSerializable`.
+    public func serialize(_ binds: inout [Encodable]) -> String {
+        if returning.isEmpty {
+            return insert.serialize(&binds)
+        } else {
+            return insert.serialize(&binds) + " RETURNING (" + returning.serialize(&binds) + ")"
+        }
+    }
+}
+
+extension SQLInsertBuilder where Connection.Query.Insert == PostgreSQLInsert {
+    public func returning(_ exprs: PostgreSQLSelectExpression...) -> Self {
+        insert.returning += exprs
+        return self
+    }
+}