Implement DatabaseConnection, change Table to Schema

Author: technetos

postgres_resource_derive/src/builder.rs

@@ -41,3 +41,23 @@ impl<'i> Builder<'i> for Schema {
         Ok(quote!(crate::schema::#model))
     }
 }
+
+struct DefaultDatabaseConnection;
+
+impl<'i> Builder<'i> for DefaultDatabaseConnection {
+    fn build(self, _: &'i Input) -> Result<proc_macro2::TokenStream> {
+        Ok(quote!(&self.connection()))
+    }
+}
+        
+pub struct DatabaseConnection;
+
+impl<'i> Builder<'i> for DatabaseConnection {
+    fn build(self, input: &'i Input) -> Result<proc_macro2::TokenStream> {
+        if let Some(ref env_var) = input.parsed_struct.attrs.db_conn {
+            Ok(quote!(&self.connection_string(#env_var)))
+        } else {
+            DefaultDatabaseConnection.build(input)
+        }
+    }
+}

postgres_resource_derive/src/lib.rs

@@ -75,13 +75,12 @@ impl Input {
     }
 
     fn gen_controller(&self) -> Result<proc_macro2::TokenStream> {
-        let model_with_id = self.parsed_struct.model_name_with_id();
-        let model = self.parsed_struct.inner_model_name();
-        let controller = self.parsed_struct.controller_name();
-
         let schema = Schema.build(&self)?;
+        let connection = DatabaseConnection.build(&self)?;
 
-        let connection = quote!(&self.connection());
+        let model = self.parsed_struct.inner_model_name();
+        let model_with_id = self.parsed_struct.model_name_with_id();
+        let controller = self.parsed_struct.controller_name();
 
         Ok(quote! {
             pub struct #controller;