feat(graphQL): add ldap related apis (#3625)

* feat(graphQL): add ldap related apis Signed-off-by: Wei Zhang <[email protected]> # Conflicts: # Cargo.toml * feat(ui): add LDAP integration (#3650) * feat(ui): add LDAP integration * update * update * update * update * update Signed-off-by: Wei Zhang <[email protected]> # Conflicts: # ee/tabby-ui/lib/tabby/query.ts * chore(ui): drop unused import Signed-off-by: Wei Zhang <[email protected]> * chore(ui): fix style Signed-off-by: Wei Zhang <[email protected]> --------- Signed-off-by: Wei Zhang <[email protected]> Co-authored-by: aliang <[email protected]>
TabbyML · Jan 9, 2025 · 61b15b7 · 61b15b7
1 parent a9f4ffa
commit 61b15b7
Show file tree

Hide file tree

Showing 36 changed files with 1,786 additions and 162 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -68,6 +68,7 @@ mime_guess = "2.0.4"
 assert_matches = "1.5"
 insta = "1.34.0"
 logkit = "0.3"
+ldap3 = "0.11.0"
 async-openai-alt = "0.26.1"
 tracing-test = "0.2"
 clap = "4.3.0"

diff --git a/ee/tabby-db/src/lib.rs b/ee/tabby-db/src/lib.rs
@@ -8,6 +8,7 @@ pub use email_setting::EmailSettingDAO;
 pub use integrations::IntegrationDAO;
 pub use invitations::InvitationDAO;
 pub use job_runs::JobRunDAO;
+pub use ldap_credential::LdapCredentialDAO;
 pub use notifications::NotificationDAO;
 pub use oauth_credential::OAuthCredentialDAO;
 pub use provided_repositories::ProvidedRepositoryDAO;

diff --git a/ee/tabby-schema/Cargo.toml b/ee/tabby-schema/Cargo.toml
@@ -30,6 +30,7 @@ validator = { version = "0.18.1", features = ["derive"] }
 regex.workspace = true
 hash-ids.workspace = true
 url.workspace = true
+ldap3.workspace = true
 
 [dev-dependencies]
 tabby-db = { path = "../../ee/tabby-db", features = ["testutils"]}

diff --git a/ee/tabby-schema/graphql/schema.graphql b/ee/tabby-schema/graphql/schema.graphql
@@ -10,6 +10,13 @@ enum AuthMethod {
   LOGIN
 }
 
+enum AuthProviderKind {
+  OAUTH_GITHUB
+  OAUTH_GOOGLE
+  OAUTH_GITLAB
+  LDAP
+}
+
 "Represents the kind of context source."
 enum ContextSourceKind {
   GIT
@@ -61,6 +68,12 @@ enum Language {
   OTHER
 }
 
+enum LdapEncryptionKind {
+  NONE
+  START_TLS
+  LDAPS
+}
+
 enum LicenseStatus {
   OK
   EXPIRED
@@ -231,6 +244,19 @@ input UpdateIntegrationInput {
   kind: IntegrationKind!
 }
 
+input UpdateLdapCredentialInput {
+  host: String!
+  port: Int!
+  bindDn: String!
+  bindPassword: String
+  baseDn: String!
+  userFilter: String!
+  encryption: LdapEncryptionKind!
+  skipTlsVerify: Boolean!
+  emailAttribute: String!
+  nameAttribute: String
+}
+
 input UpdateMessageInput {
   id: ID!
   threadId: ID!
@@ -288,6 +314,10 @@ interface User {
 """
 scalar DateTime
 
+type AuthProvider {
+  kind: AuthProviderKind!
+}
+
 type CompletionStats {
   start: DateTime!
   end: DateTime!
@@ -464,6 +494,20 @@ type JobStats {
   pending: Int!
 }
 
+type LdapCredential {
+  host: String!
+  port: Int!
+  bindDn: String!
+  baseDn: String!
+  userFilter: String!
+  encryption: LdapEncryptionKind!
+  skipTlsVerify: Boolean!
+  emailAttribute: String!
+  nameAttribute: String
+  createdAt: DateTime!
+  updatedAt: DateTime!
+}
+
 type LicenseInfo {
   type: LicenseType!
   status: LicenseStatus!
@@ -575,6 +619,7 @@ type Mutation {
   updateUserName(id: ID!, name: String!): Boolean!
   register(email: String!, password1: String!, password2: String!, invitationCode: String, name: String!): RegisterResponse!
   tokenAuth(email: String!, password: String!): TokenAuthResponse!
+  tokenAuthLdap(userId: String!, password: String!): TokenAuthResponse!
   verifyToken(token: String!): Boolean!
   refreshToken(refreshToken: String!): RefreshTokenResponse!
   createInvitation(email: String!): ID!
@@ -586,6 +631,9 @@ type Mutation {
   deleteInvitation(id: ID!): ID!
   updateOauthCredential(input: UpdateOAuthCredentialInput!): Boolean!
   deleteOauthCredential(provider: OAuthProvider!): Boolean!
+  testLdapConnection(input: UpdateLdapCredentialInput!): Boolean!
+  updateLdapCredential(input: UpdateLdapCredentialInput!): Boolean!
+  deleteLdapCredential: Boolean!
   updateEmailSetting(input: EmailSettingInput!): Boolean!
   updateSecuritySetting(input: SecuritySettingInput!): Boolean!
   updateNetworkSetting(input: NetworkSettingInput!): Boolean!
@@ -717,8 +765,10 @@ type Query {
     * `func_name lang:go`
   """
   repositoryGrep(kind: RepositoryKind!, id: ID!, rev: String, query: String!): RepositoryGrepOutput!
+  authProviders: [AuthProvider!]!
   oauthCredential(provider: OAuthProvider!): OAuthCredential
   oauthCallbackUrl(provider: OAuthProvider!): String!
+  ldapCredential: LdapCredential
   serverInfo: ServerInfo!
   license: LicenseInfo!
   jobs: [String!]!

diff --git a/ee/tabby-schema/src/dao.rs b/ee/tabby-schema/src/dao.rs
@@ -2,19 +2,20 @@ use anyhow::bail;
 use hash_ids::HashIds;
 use lazy_static::lazy_static;
 use tabby_db::{
-    EmailSettingDAO, IntegrationDAO, InvitationDAO, JobRunDAO, NotificationDAO, OAuthCredentialDAO,
-    ServerSettingDAO, ThreadDAO, ThreadMessageAttachmentClientCode, ThreadMessageAttachmentCode,
-    ThreadMessageAttachmentDoc, ThreadMessageAttachmentIssueDoc, ThreadMessageAttachmentPullDoc,
-    ThreadMessageAttachmentWebDoc, UserEventDAO,
+    EmailSettingDAO, IntegrationDAO, InvitationDAO, JobRunDAO, LdapCredentialDAO, NotificationDAO,
+    OAuthCredentialDAO, ServerSettingDAO, ThreadDAO, ThreadMessageAttachmentClientCode,
+    ThreadMessageAttachmentCode, ThreadMessageAttachmentDoc, ThreadMessageAttachmentIssueDoc,
+    ThreadMessageAttachmentPullDoc, ThreadMessageAttachmentWebDoc, UserEventDAO,
 };
 
 use crate::{
+    auth::LdapEncryptionKind,
     integration::{Integration, IntegrationKind, IntegrationStatus},
     interface::UserValue,
     notification::{Notification, NotificationRecipient},
     repository::RepositoryKind,
     schema::{
-        auth::{self, OAuthCredential, OAuthProvider},
+        auth::{self, LdapCredential, OAuthCredential, OAuthProvider},
         email::{AuthMethod, EmailSetting, Encryption},
         job,
         repository::{
@@ -67,6 +68,26 @@ impl TryFrom<OAuthCredentialDAO> for OAuthCredential {
     }
 }
 
+impl TryFrom<LdapCredentialDAO> for LdapCredential {
+    type Error = anyhow::Error;
+
+    fn try_from(val: LdapCredentialDAO) -> Result<Self, Self::Error> {
+        Ok(LdapCredential {
+            host: val.host,
+            port: val.port as i32,
+            bind_dn: val.bind_dn,
+            base_dn: val.base_dn,
+            user_filter: val.user_filter,
+            encryption: LdapEncryptionKind::from_enum_str(&val.encryption)?,
+            skip_tls_verify: val.skip_tls_verify,
+            email_attribute: val.email_attribute,
+            name_attribute: val.name_attribute,
+            created_at: val.created_at,
+            updated_at: val.updated_at,
+        })
+    }
+}
+
 impl TryFrom<EmailSettingDAO> for EmailSetting {
     type Error = anyhow::Error;
 
@@ -447,6 +468,25 @@ impl DbEnum for OAuthProvider {
     }
 }
 
+impl DbEnum for LdapEncryptionKind {
+    fn as_enum_str(&self) -> &'static str {
+        match self {
+            LdapEncryptionKind::None => "none",
+            LdapEncryptionKind::StartTLS => "starttls",
+            LdapEncryptionKind::LDAPS => "ldaps",
+        }
+    }
+
+    fn from_enum_str(s: &str) -> anyhow::Result<Self> {
+        match s {
+            "none" => Ok(LdapEncryptionKind::None),
+            "starttls" => Ok(LdapEncryptionKind::StartTLS),
+            "ldaps" => Ok(LdapEncryptionKind::LDAPS),
+            _ => bail!("Invalid Ldap encryption kind"),
+        }
+    }
+}
+
 impl DbEnum for AuthMethod {
     fn as_enum_str(&self) -> &'static str {
         match self {