@neuralinnovations/dataisland-sdk 0.0.1-dev9 → 0.1.0

package/src/commands/startCommandHandler.ts

@@ -1,14 +1,31 @@
 import { CommandHandler, Command } from "../services/commandService"
 import { UserProfileService } from "../services/userProfileService"
+import { CookieService } from "../services/cookieService"
+import { AnonymousCredential } from "../credentials"
+import { AcquiringService } from "../services/acquiringService"
+import {LibrariesService} from "../services/librariesService"
 
-export class StartCommand extends Command {}
+export class StartCommand extends Command {
+}
 
 export class StartCommandHandler extends CommandHandler<StartCommand> {
   // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async execute(message: StartCommand): Promise<void> {
-    const service = this.context.resolve(
-      UserProfileService
-    ) as UserProfileService
-    await service.fetch()
+    const userService = this.resolve(UserProfileService)!
+    const acquiringService = this.resolve(AcquiringService)!
+    const librariesService = this.resolve(LibrariesService)!
+
+    // Merge anonymous user if needed
+    if (!(this.context.app.credential instanceof AnonymousCredential)) {
+      const cookie = this.resolve(CookieService)!
+      if (cookie.anonymousTokenIsValid) {
+        await userService.merge(cookie.anonymousToken!)
+        cookie.anonymousToken = undefined
+      }
+    }
+
+    await librariesService.initialize()
+    await userService.fetch()
+    await acquiringService.initialize()
   }
 }

package/src/context.ts

@@ -1,6 +1,7 @@
 import { type Constructor, type Registry } from "./internal/registry"
 import { type Lifetime } from "./disposable"
 import { Command, CommandService } from "./services/commandService"
+import { DataIslandApp } from "./dataIslandApp"
 
 /**
  * DataIsland App context.
@@ -9,8 +10,9 @@ export class Context {
   constructor(
     private readonly registry: Registry,
     public readonly lifetime: Lifetime,
-    public readonly appName: string
-  ) {}
+    public readonly app: DataIslandApp
+  ) {
+  }
 
   /**
    * Resolve a service from the context.

package/src/credentials.ts

@@ -83,3 +83,49 @@ export class BearerCredential extends CredentialBase {
     )
   }
 }
+
+export class AnonymousCredential extends CredentialBase {
+  readonly token: string
+
+  constructor(token: string) {
+    super()
+    this.token = token
+  }
+
+  onRegister(lifetime: Lifetime, context: Context): void {
+    const service = context.resolve(MiddlewareService)
+    if (service === undefined) {
+      throw new Error("MiddlewareService is not registered.")
+    }
+    lifetime.add(
+      service.useMiddleware(async (req, next) => {
+        req.headers.set("Authorization", `InternalJWT ${this.token}`)
+        return await next(req)
+      })
+    )
+  }
+}
+
+export class CustomCredential extends CredentialBase {
+  readonly schema: string
+  readonly token: string
+
+  constructor(schema: string, token: string) {
+    super()
+    this.token = token
+    this.schema = schema
+  }
+
+  onRegister(lifetime: Lifetime, context: Context): void {
+    const service = context.resolve(MiddlewareService)
+    if (service === undefined) {
+      throw new Error("MiddlewareService is not registered.")
+    }
+    lifetime.add(
+      service.useMiddleware(async (req, next) => {
+        req.headers.set("Authorization", `${this.schema} ${this.token}`)
+        return await next(req)
+      })
+    )
+  }
+}

package/src/dataIslandApp.ts

@@ -4,6 +4,8 @@ import { Context } from "./context"
 import type { Constructor } from "./internal/registry"
 import { Organizations } from "./storages/organizations/organizations"
 import { UserProfile } from "./storages/user/userProfile"
+import { Acquiring } from "./storages/acquirings/acquiring"
+import {Libraries} from "./storages/library/libraries"
 
 /**
  * DataIsland App instance.
@@ -51,6 +53,16 @@ export abstract class DataIslandApp {
    */
   abstract get userProfile(): UserProfile
 
+  /**
+   * Acquiring.
+   */
+  abstract get acquiring(): Acquiring
+
+  /**
+   * Libraries.
+   */
+  abstract get libraries(): Libraries
+
   /**
    * Resolve a service from the app.
    * @param type

package/src/disposable.ts

@@ -77,7 +77,8 @@ export class DisposableContainer implements Disposable {
    * @returns The disposable container.
    */
   public add(disposable: Disposable): Disposable {
-    this._throwIfDisposed()
+    this._throwIfDisposed("Cannot add disposable to disposed container.")
+
     this._disposables.push(disposable)
     return disposable
   }
@@ -89,7 +90,8 @@ export class DisposableContainer implements Disposable {
    * @returns The disposable container.
    */
   public addCallback(callback: () => void, target?: unknown): Disposable {
-    this._throwIfDisposed()
+    this._throwIfDisposed("Cannot add callback to disposed container.")
+
     return this.add({
       dispose() {
         callback.call(target)
@@ -101,6 +103,8 @@ export class DisposableContainer implements Disposable {
    * Defines a nested disposable container.
    */
   defineNested(): DisposableContainer {
+    this._throwIfDisposed("Cannot define nested disposable for disposed container.")
+
     const nested = new DisposableContainer()
     this._disposables.push(nested)
     nested.addCallback(() => {
@@ -116,23 +120,32 @@ export class DisposableContainer implements Disposable {
    * Disposes all disposables in this container. Last added, first disposed.
    */
   public dispose(): void {
-    this._throwIfDisposed()
+    // return if already disposed
+    if (this._isDisposed) {
+      return
+    }
+
+    // mark as disposed
     this._isDisposed = true
+
+    // dispose all disposables
     this._disposables
       .slice()
       .reverse()
       .forEach(it => {
         it.dispose()
       })
+
+    // clear disposables
     this._disposables = []
   }
 
   /**
    * Throws an error if this container is disposed.
    */
-  private _throwIfDisposed(): void {
+  private _throwIfDisposed(message: string): void {
     if (this._isDisposed) {
-      throw new Error("Object disposed")
+      throw new Error("Object disposed, but you try to use it: " + message)
     }
   }
 }

package/src/dto/accessGroupResponse.ts

@@ -1,35 +1,36 @@
 import { UserDto } from "./userInfoResponse"
 
-
 export interface PermitsDto {
-    isAdmin: boolean;
+  isAdmin: boolean;
 }
 
 export interface RegulationDto {
-    isRegulateOrganization: boolean;
-    regulateWorkspaceIds: string[];
+  isRegulateOrganization: boolean;
+  regulateWorkspaceIds: string[];
 }
 
 export interface AccessGroupDto {
-    id: string;
-    name: string;
-    type: number;
-    createdAt: number;
-    modifiedAt: number;
-    organizationId: string;
-    permits: PermitsDto;
-    regulation: RegulationDto;
-    membersCount: number;
+  id: string;
+  name: string;
+  type: number;
+  createdAt: number;
+  modifiedAt: number;
+  organizationId: string;
+  permits: PermitsDto;
+  regulation: RegulationDto;
+  membersCount: number;
 }
 
 export interface AccessGroupResponse {
-    group: AccessGroupDto;
-    members: UserDto[];
+  group: AccessGroupDto;
+  members: UserDto[];
 }
 
 export interface AccessGroupsResponse {
-    groups: AccessGroupDto[];
+  groups: AccessGroupDto[];
 }
 
 
 
+
+

package/src/dto/acquiringResponse.ts

@@ -0,0 +1,54 @@
+import { SegmentItemData } from "./limitsResponse"
+
+export interface OrderResponse {
+  data: OrderData
+}
+
+export interface OrderData {
+  data: string,
+  signature: string
+}
+
+export interface UnsubscribeResponse {
+  isSuccessful: boolean
+  orderId: string
+}
+
+export interface GetOrderStateResponse {
+  isValid: boolean,
+  state: OrderState,
+  plan: AcquiringPlan
+}
+
+export interface AcquiringPlansResponse {
+  plans: AcquiringPlan[]
+  limitSegments: AcquiringSegmentData[]
+}
+
+export interface UserAcquiringPlan {
+  isValid: boolean,
+  state: OrderState,
+  plan: AcquiringPlan
+}
+
+export interface AcquiringPlan {
+  key: string,
+  segmentKey: string,
+  price: number,
+  currency: string,
+  periodicity: number,
+  description: string,
+  isEnabled: boolean
+}
+
+export interface AcquiringSegmentData {
+  key: string
+  dayItems: SegmentItemData[]
+}
+
+export enum OrderState {
+  Created = 0,
+  WaitingForPayment = 10,
+  Paid = 20,
+  Cancelled = 30
+}

package/src/dto/apiKeyResponse.ts

@@ -0,0 +1,29 @@
+
+
+export interface TokenResponse {
+  userJwtToken: string
+  authSchemaName: string
+}
+
+export interface ApiKeyResponse {
+  apiKey: OrganizationApiKey
+}
+
+export interface OrganizationKeysResponse {
+  keys: OrganizationApiKey[]
+}
+
+export interface OrganizationApiKey {
+  keyName: string
+  apiKey: string
+  tokensCreated: number
+  createdAt: number
+  lastTokenCreatedAt: number
+  accessGroupInfo: AccessGroupInfo[]
+}
+
+export interface AccessGroupInfo {
+  id: string
+  name: string
+  type: number
+}

package/src/dto/badRequestResponse.ts

@@ -0,0 +1,6 @@
+
+export interface BadRequest {
+    code: number
+    name: string
+    desc?: string
+}

package/src/dto/chatResponse.ts

@@ -1,3 +1,7 @@
+import { AnswerId, StepId } from "../storages/chats/answer"
+import { ChatId } from "../storages/chats/chat"
+import { UserId } from "../storages/user/userProfile"
+
 export interface SourceDto {
   id: string;
   name: string;
@@ -7,8 +11,8 @@ export interface SourceDto {
 }
 
 export interface AnswerDto {
-  id: string;
-  chatId: string;
+  id: AnswerId;
+  chatId: ChatId;
   question: string;
   context: string;
   sources: SourceDto[];
@@ -16,16 +20,37 @@ export interface AnswerDto {
 }
 
 export interface ChatDto {
-  id: string;
+  id: ChatId;
   name: string;
   createdAt: number;
   modifiedAt: number;
-  userId: string;
-  organizationId: string;
-  workspaceId: string;
+  resources: ChatResourceDto;
+  userId: UserId;
+  model: string;
+  clientContext: string;
   answers: AnswerDto[];
 }
 
+export interface ChatResourceDto {
+  chatType: ChatType
+  libraryId: string
+  organizationId: string
+  workspaceIds: string[]
+  fileId: string
+}
+
+export enum ChatType {
+  None = 0,
+  Organization = 1,
+  Workspaces = 2,
+  File = 3,
+  Library = 100,
+  LibraryOrganization = 101,
+  LibraryWorkspaces = 102,
+  LibraryFile = 103
+
+}
+
 export interface ChatListResponse {
   chats: ChatDto[]
 }
@@ -38,7 +63,7 @@ export enum AnswerStatus {
 }
 
 export interface AnswerStepDto {
-  id: string;
+  id: StepId;
   type: StepType;
   status: StepStatus;
   start_at: string;
@@ -48,7 +73,7 @@ export interface AnswerStepDto {
 }
 
 export interface FetchAnswerResponse {
-  id: string;
+  id: AnswerId;
   status: AnswerStatus;
   steps: AnswerStepDto[];
 }

package/src/dto/invitesResponse.ts

@@ -0,0 +1,25 @@
+
+
+export interface InviteCodeResponse {
+  code: string
+}
+
+export interface InviteResponse {
+  inviteLinks: InviteLink[]
+  users: UserInviteData[]
+}
+
+export interface InviteLink {
+  organizationId: string
+  createdBy: string
+  accessGroupIds: string[]
+  code: string
+  expireAt: 0
+  usages: 0
+}
+
+export interface UserInviteData {
+  id: string
+  name: string
+  email: string
+}

package/src/dto/libraryResponse.ts

@@ -0,0 +1,51 @@
+import { FileDto } from "./workspacesResponse"
+import { OrganizationId } from "../storages/organizations/organizations"
+
+import { LibraryId } from "../storages/library/libraryId"
+
+export interface LibrariesResponse {
+  libraries: LibraryFolderDto[]
+}
+
+export interface LibraryFolderDto {
+  id: string
+  name: string
+  description: string
+  iconId: string
+  createdAt: number
+  modifiedAt: number
+}
+
+export interface LibraryPageResponse {
+  parentIds: string[]
+  totalItemsCount: number
+  itemsPerPage: number
+  folders: LibraryFolderDto[]
+  files: FileDto[]
+}
+
+export interface LibraryFolderResponse {
+  folder: LibraryFolderDto
+}
+
+export interface LibraryOrganizationDto {
+  id: OrganizationId
+  name: string
+}
+
+export interface LibraryDto {
+  id: LibraryId
+  name: string
+  region: number
+  isPublic: boolean
+  organizations: LibraryOrganizationDto[]
+}
+
+export interface LibraryResponse {
+  libraries: LibraryDto[]
+}
+
+export interface CreateLibraryResponse {
+
+  libraryId: LibraryId
+}

package/src/dto/limitsResponse.ts

@@ -0,0 +1,32 @@
+export interface SegmentsData {
+  segments: SegmentData[]
+}
+
+export interface SegmentData {
+  key: string
+  isRegisteredDefault: boolean
+  isAnonymousDefault: boolean
+  isAnonymous: boolean
+  isRegistered: boolean
+  dayItems: SegmentItemData[]
+}
+
+export interface SegmentItemData {
+  daysCount: number
+  actionItems: SegmentActionData[]
+}
+
+export interface SegmentActionData {
+  type: LimitActionType
+  countLimit: number
+  tokenLimit?: number
+}
+
+export enum LimitActionType {
+  UploadFile = 0,
+  CreateChat = 1,
+  AskQuestion = 2,
+  CreateWorkspace = 3,
+  CreateOrganization = 4,
+  FileSizeKb = 5
+}

package/src/dto/queryFlowResponse.ts

@@ -0,0 +1,42 @@
+export enum QueryFlowState {
+  IN_PROGRESS = 0,
+  ERROR = 1,
+  DONE = 100
+}
+
+export enum QueryFlowStatus {
+  STARTED = 0,
+  UPLOADING_FILE = 1,
+  PROCESSING_FILE = 2,
+  PREPARING_RESULTS = 3,
+  DONE = 4,
+  FAILED = 5
+}
+
+export interface QueryFileUrlDto {
+  name: string
+  extension: string
+  url: string
+}
+
+export interface QueryFlowResult {
+  files: QueryFileUrlDto[]
+}
+
+// GET query flow by ID
+export interface QueryFlowDto {
+  name: string
+  state: QueryFlowState,
+  status: QueryFlowStatus,
+  result: QueryFlowResult
+}
+
+// GET org query flows
+export interface QueryFlowListResponse {
+  flowIds: string[]
+}
+
+// POST Create query flow
+export interface QueryFlowResponse {
+  flowId: string
+}

package/src/dto/quizResponse.ts

@@ -0,0 +1,12 @@
+import {SourceDto} from "./chatResponse"
+
+export interface QuizData{
+  quiz: QuestionData[]
+  sources: SourceDto[]
+}
+
+export interface QuestionData {
+  question: string,
+  options: string[],
+  correct_option: number
+}

package/src/dto/statisticsResponse.ts

@@ -0,0 +1,12 @@
+export interface StatisticsResponse {
+    dateFrom: number
+    dateTo: number
+    data: StatisticsData[]
+}
+
+export interface StatisticsData {
+    date: number
+    promptUsage: number
+    completionUsage: number
+}
+