infrahub-sdk 0.0.2 → 0.0.5

package/src/index.ts

@@ -3,9 +3,18 @@ export type SafeResult<T> =
   | { data: T; error: undefined }
   | { data: undefined; error: Error };
 
+import type { TypedDocumentNode } from '@graphql-typed-document-node/core';
+
 import createClient from 'openapi-fetch';
-import { GraphQLClient, Variables } from 'graphql-request';
+import { GraphQLClient, Variables, gql } from 'graphql-request';
 import type { paths } from './types';
+import { InfrahubBranchManager } from './branch';
+import {
+  InfrahubInfoQuery,
+  InfrahubInfoResponse,
+  InfrahubUserResponse,
+  InfrahubUserQuery
+} from './graphql/client';
 
 // Re-export gql and all exports from graphql-request
 export * from 'graphql-request';
@@ -26,11 +35,13 @@ export class InfrahubClient {
   private baseUrl: string;
   private token?: string;
   public defaultBranch: string;
+  public branch: InfrahubBranchManager;
 
   constructor(options: InfrahubClientOptions) {
     this.baseUrl = options.address;
     this.token = options.token;
     this.defaultBranch = options.branch || DEFAULT_BRANCH_NAME;
+    this.branch = new InfrahubBranchManager(this);
 
     // Initialize the openapi-fetch client
 
@@ -122,7 +133,7 @@ export class InfrahubClient {
     TData = any,
     TVariables extends Variables = Variables
   >(
-    query: string,
+    query: string | TypedDocumentNode<TData, TVariables>,
     variables?: TVariables,
     branchName?: string
   ): Promise<TData> {
@@ -138,4 +149,27 @@ export class InfrahubClient {
       throw error;
     }
   }
+
+  public async getVersion(): Promise<string> {
+    const data = await this.executeGraphQL<
+      InfrahubInfoResponse,
+      Record<any, never>
+    >(InfrahubInfoQuery);
+    return data.InfrahubInfo.version;
+  }
+
+  public async getUser(): Promise<InfrahubUserResponse> {
+    const data = await this.executeGraphQL<
+      InfrahubUserResponse,
+      Record<any, never>
+    >(InfrahubUserQuery);
+    return data;
+  }
+
+  public async getUserPermissions(): Promise<
+    InfrahubUserResponse['AccountProfile']['member_of_groups']['edges']
+  > {
+    const user = await this.getUser();
+    return user.AccountProfile.member_of_groups.edges;
+  }
 }

package/test/main.js

@@ -0,0 +1,11 @@
+import { InfrahubClient } from 'infrahub-sdk';
+
+const options = {
+    address: "https://demo.infrahub.app",
+    token: "your-api-token"
+}
+
+console.log('Executing GraphQL query...');
+
+const client = new InfrahubClient(options);
+await client.getVersion();