[query] Add select keys #427 (#428)

* Create select_keys.rs

* add select_keys

src/agdb/query_builder.rs

@@ -38,6 +38,8 @@ mod select_id_from;
 mod select_id_to;
 mod select_key_count;
 mod select_key_count_from;
+mod select_keys;
+mod select_keys_from;
 mod select_limit;
 mod select_offset;
 

src/agdb/query_builder/select.rs

@@ -3,11 +3,13 @@ use super::select_alias::SelectAlias;
 use super::select_aliases::SelectAliases;
 use super::select_id::SelectId;
 use super::select_key_count::SelectKeyCount;
+use super::select_keys::SelectKeys;
 use crate::query::query_id::QueryId;
 use crate::query::query_ids::QueryIds;
 use crate::query::search_query::SearchQuery;
 use crate::query::select_aliases_query::SelectAliasesQuery;
 use crate::query::select_key_count_query::SelectKeyCountQuery;
+use crate::query::select_keys_query::SelectKeysQuery;
 
 pub struct Select {}
 
@@ -57,6 +59,10 @@ impl Select {
         })
     }
 
+    pub fn keys(self) -> SelectKeys {
+        SelectKeys(SelectKeysQuery(QueryIds::Id(0.into())))
+    }
+
     pub fn key_count(self) -> SelectKeyCount {
         SelectKeyCount(SelectKeyCountQuery(QueryIds::Id(0.into())))
     }

src/agdb/query_builder/select_keys.rs

@@ -0,0 +1,30 @@
+use super::select_keys_from::SelectKeysFrom;
+use crate::query::query_id::QueryId;
+use crate::query::query_ids::QueryIds;
+use crate::query::search_query::SearchQuery;
+use crate::query::select_keys_query::SelectKeysQuery;
+
+pub struct SelectKeys(pub SelectKeysQuery);
+
+impl SelectKeys {
+    #[allow(clippy::wrong_self_convention)]
+    pub fn from(mut self, id: QueryId) -> SelectKeysFrom {
+        self.0 .0 = QueryIds::Id(id);
+
+        SelectKeysFrom(self.0)
+    }
+
+    #[allow(clippy::wrong_self_convention)]
+    pub fn from_ids(mut self, ids: &[QueryId]) -> SelectKeysFrom {
+        self.0 .0 = QueryIds::Ids(ids.to_vec());
+
+        SelectKeysFrom(self.0)
+    }
+
+    #[allow(clippy::wrong_self_convention)]
+    pub fn from_query(mut self, query: SearchQuery) -> SelectKeysFrom {
+        self.0 .0 = QueryIds::Search(query);
+
+        SelectKeysFrom(self.0)
+    }
+}

src/agdb/query_builder/select_keys_from.rs

@@ -0,0 +1,9 @@
+use crate::query::select_keys_query::SelectKeysQuery;
+
+pub struct SelectKeysFrom(pub SelectKeysQuery);
+
+impl SelectKeysFrom {
+    pub fn query(self) -> SelectKeysQuery {
+        self.0
+    }
+}

tests/select_keys.rs

@@ -0,0 +1,22 @@
+use agdb::QueryBuilder;
+
+#[test]
+fn select_keys_from() {
+    let _query = QueryBuilder::select().keys().from("alias".into()).query();
+}
+
+#[test]
+fn select_keys_from_ids() {
+    let _query = QueryBuilder::select()
+        .keys()
+        .from_ids(&["alias".into()])
+        .query();
+}
+
+#[test]
+fn select_keys_from_query() {
+    let _query = QueryBuilder::select()
+        .keys()
+        .from_query(QueryBuilder::select().id().from("alias".into()).query())
+        .query();
+}