@neuralinnovations/dataisland-sdk 0.0.1-dev8 → 0.0.1-dev81

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

@@ -2,8 +2,10 @@ import type { Lifetime } from "./disposable"
 import type { CredentialBase } from "./credentials"
 import { Context } from "./context"
 import type { Constructor } from "./internal/registry"
-import { Organizations } from "./storages/organizations"
-import { UserProfile } from "./storages/userProfile"
+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,88 +1,112 @@
+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;
-    url: string;
-    content: string;
-    page: number;
+  id: string;
+  name: string;
+  url: string;
+  content: string;
+  page: number;
 }
 
 export interface AnswerDto {
-    id: string;
-    chatId: string;
-    question: string;
-    context: string;
-    sources: SourceDto[];
-    timestamp: number;
+  id: AnswerId;
+  chatId: ChatId;
+  question: string;
+  context: string;
+  sources: SourceDto[];
+  timestamp: number;
 }
 
 export interface ChatDto {
-    id: string;
-    name: string;
-    createdAt: number;
-    modifiedAt: number;
-    userId: string;
-    organizationId: string;
-    workspaceId: string;
-    answers: AnswerDto[];
+  id: ChatId;
+  name: string;
+  createdAt: number;
+  modifiedAt: number;
+  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[]
+  chats: ChatDto[]
 }
 
 export enum AnswerStatus {
-    RUNNING = 0,
-    SUCCESS = 1,
-    CANCELED = 2,
-    FAIL = 3,
+  RUNNING = 0,
+  SUCCESS = 1,
+  CANCELED = 2,
+  FAIL = 3,
 }
 
-export interface AnswerStepDto{
-    id: string;
-    type: StepType;
-    status: StepStatus;
-    start_at: string;
-    end_at: string;
-    tokens: string[];
-    sources: SourceDto[];
+export interface AnswerStepDto {
+  id: StepId;
+  type: StepType;
+  status: StepStatus;
+  start_at: string;
+  end_at: string;
+  tokens: string[];
+  sources: SourceDto[];
 }
 
 export interface FetchAnswerResponse {
-    id: string;
-    status: AnswerStatus;
-    steps: AnswerStepDto[];
+  id: AnswerId;
+  status: AnswerStatus;
+  steps: AnswerStepDto[];
 }
 
 export interface FetchTokensResponse {
-    id: string;
-    step_id: string;
-    step_status: number;
-    step_tokens: string[];
+  id: string;
+  step_id: string;
+  step_status: number;
+  step_tokens: string[];
 }
 
 export interface AnswerSourcesResponse {
-    chat_uid: string;
-    uid: string;
-    step_id: string;
-    sources: SourceDto[];
+  chat_uid: string;
+  uid: string;
+  step_id: string;
+  sources: SourceDto[];
 }
 
 export enum StepStatus {
-    RUNNING = 0,
-    SUCCESS = 1,
-    FAIL = 2,
-    CANCELED = 3,
+  RUNNING = 0,
+  SUCCESS = 1,
+  FAIL = 2,
+  CANCELED = 3,
 }
 
 export enum StepType {
-    PREPARE = 0,
-    SOURCES = 1,
-    GENERATE_ANSWER = 6,
-    FINALIZE_RESULT = 9,
-    DONE = 10,
+  PREPARE = 0,
+  SOURCES = 1,
+  GENERATE_ANSWER = 6,
+  FINALIZE_RESULT = 9,
+  DONE = 10,
 }
-  
+
 export class StepTypeInfo {
   public static hasTokens(type: StepType): boolean {
     switch (type) {
@@ -101,4 +125,4 @@ export class StepTypeInfo {
     }
     return false
   }
-}
+}

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,28 @@
+
+
+export enum QueryFlowStatus {
+  IN_PROGRESS = 0,
+  ERROR = 1,
+  DONE = 100
+}
+
+export interface QueryFlowResult {
+  fileUrl: string
+}
+
+// GET query flow by ID
+export interface QueryFlowDto {
+  name: string
+  state: 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
+}
+