@neuralinnovations/dataisland-sdk 0.0.1-dev1 → 0.0.1-dev10

package/src/services/organizationService.ts

@@ -0,0 +1,28 @@
+import { Service } from "./service"
+import { Organizations } from "../storages/organizations/organizations"
+import { OrganizationDto, UserSettings } from "../dto/userInfoResponse"
+import { OrganizationsImpl } from "../storages/organizations/organizations.impl"
+
+export class OrganizationService extends Service {
+  private _impl?: OrganizationsImpl
+
+  private get impl(): OrganizationsImpl {
+    return this._impl ?? (this._impl = new OrganizationsImpl(this.context))
+  }
+
+  get organizations(): Organizations {
+    return this.impl
+  }
+
+  async initFrom(
+    adminInOrganization: string[],
+    organizations: OrganizationDto[],
+    settings?: UserSettings | null
+  ): Promise<void> {
+    await this.impl.internalInitFrom(
+      adminInOrganization,
+      organizations,
+      settings
+    )
+  }
+}

package/src/services/requestBuilder.ts

@@ -0,0 +1,127 @@
+export class RequestBuilder {
+  private readonly _headers: Headers
+  private readonly _searchParams: URLSearchParams
+
+  constructor(
+    private readonly _url: URL,
+    private readonly _request: (req: Request) => Promise<Response>
+  ) {
+    this._headers = new Headers()
+    this._searchParams = new URLSearchParams()
+  }
+
+  public header(name: string, value: string): RequestBuilder {
+    this._headers.set(name, value)
+    return this
+  }
+
+  public headers(
+    headers?: [string, string][] | Record<string, string> | Headers
+  ): RequestBuilder {
+    if (headers === undefined) {
+      return this
+    }
+    if (headers instanceof Headers) {
+      headers.forEach((value, name) => {
+        this._headers.set(name, value)
+      })
+    } else {
+      Object.entries(headers).forEach(([name, value]) => {
+        this._headers.set(name, value)
+      })
+    }
+    return this
+  }
+
+  public searchParam(name: string, value: string): RequestBuilder {
+    this._searchParams.set(name, value)
+    return this
+  }
+
+  public searchParams(searchParams?: Map<string, string>): RequestBuilder {
+    if (searchParams === undefined) {
+      return this
+    }
+    searchParams.forEach((value, name) => {
+      this._searchParams.set(name, value)
+    })
+    return this
+  }
+
+  public async sendPostFormData(body: FormData): Promise<Response> {
+    const url = this._url
+
+    // set search params
+    url.search = this._searchParams.toString()
+
+    // create request
+    const req = new Request(url, {
+      method: "POST",
+      headers: this._headers,
+      body
+    })
+
+    // discard content type
+    const reqAny = req as any
+    reqAny.discardContentType = true
+
+    return await this._request(
+      req
+    )
+  }
+
+  public async sendPostJson(body: object | null | undefined): Promise<Response> {
+    const url = this._url
+    url.search = this._searchParams.toString()
+    let json: string | null | undefined = null
+    if (body !== undefined && body !== null && typeof body === "object") {
+      json = JSON.stringify(body)
+    }
+    const request = new Request(url, {
+      method: "POST",
+      headers: this._headers,
+      body: json
+    })
+    return await this._request(
+      request
+    )
+  }
+
+  public async sendGet(): Promise<Response> {
+    const url = this._url
+    url.search = this._searchParams.toString()
+    return await this._request(
+      new Request(url, {
+        method: "GET",
+        headers: this._headers
+      })
+    )
+  }
+
+  public async sendDelete(): Promise<Response> {
+    const url = this._url
+    url.search = this._searchParams.toString()
+    return await this._request(
+      new Request(url, {
+        method: "DELETE",
+        headers: this._headers
+      })
+    )
+  }
+
+  public async sendPutJson(body: object | null | undefined): Promise<Response> {
+    const url = this._url
+    url.search = this._searchParams.toString()
+    let json: string | null | undefined = null
+    if (body !== undefined && body !== null && typeof body === "object") {
+      json = JSON.stringify(body)
+    }
+    return await this._request(
+      new Request(url, {
+        method: "PUT",
+        headers: this._headers,
+        body: json
+      })
+    )
+  }
+}

package/src/services/responseUtils.ts

@@ -0,0 +1,32 @@
+export class ResponseUtils {
+  public static isOk(response?: Response | null): boolean {
+    return response !== undefined && response !== null && response.ok
+  }
+
+  public static isFail(response?: Response | null): boolean {
+    return !ResponseUtils.isOk(response)
+  }
+
+  public static async throwError(
+    message: string,
+    response: Response | undefined | null
+  ): Promise<void> {
+    if (response === undefined) {
+      throw new Error(`${message}. Response is undefined`)
+    }
+    if (response === null) {
+      throw new Error(`${message}. Response is null`)
+    }
+    let errorBody: string = ""
+    if (response) {
+      try {
+        errorBody = (await response.text()) ?? ""
+      } catch (e) {
+        console.error(e)
+      }
+    }
+    throw new Error(
+      `${message}. Response fail. Status: ${response?.status},${response?.statusText}, body: ${errorBody}`
+    )
+  }
+}

package/src/services/rpcService.ts

@@ -1,72 +1,149 @@
-import { Service, type ServiceContext } from './service'
-import { MiddlewareService } from './middlewareService'
+import { Service, type ServiceContext } from "./service"
+import { MiddlewareService } from "./middlewareService"
+import { RequestBuilder } from "./requestBuilder"
 
+/**
+ * Options for the RpcService.
+ */
+export interface RequestOptions {
+  searchParams?: Map<string, string>
+  headers?: [string, string][] | Record<string, string> | Headers
+}
+
+/**
+ * RPC service.
+ */
 export class RpcService extends Service {
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  async request(req: Request): Promise<Response> {
-    throw new Error('Not implemented')
-  }
 
-  // eslint-disable-next-line @typescript-eslint/no-unused-vars
-  buildUrl(path: string): string {
-    throw new Error('Not implemented')
-  }
+  constructor(
+    serviceContext: ServiceContext,
+    /**
+     * Host of the RPC service.
+     * It is not used if you use the `urlBuilder` option.
+     */
+    public readonly host: string,
+    /**
+     * Options for the RpcService.
+     */
+    private readonly options?: {
+      // make it possible to override the url builder
+      urlBuilder?: (path: string) => URL
+      // make it possible to override the fetch method
+      fetchMethod?: (uri: Request) => Promise<Response>
+    }
+  ) {
+    super(serviceContext)
+
+    serviceContext.onRegister = async () => {
+      serviceContext.resolve(MiddlewareService)?.useMiddleware((req, next) => {
+        if (!req.headers.has("accept")) {
+          req.headers.set("accept", "text/plain")
+        }
 
-  async get(path: string): Promise<Response> {
-    return await this.request(
-      new Request(this.buildUrl(path), {
-        method: 'GET'
+        if ((req as any).discardContentType) {
+          delete (req as any).discardContentType
+        } else {
+          req.headers.set("content-type", "application/json")
+        }
+
+        return next(req)
       })
-    )
+    }
   }
 
-  async post(path: string, body?: BodyInit | null): Promise<Response> {
-    return await this.request(
-      new Request(this.buildUrl(path), {
-        method: 'POST',
-        body
+  /**
+   * Request method.
+   */
+  async request(req: Request): Promise<Response> {
+    const middlewareService = this.resolve(MiddlewareService)
+    if (middlewareService !== undefined) {
+      return await middlewareService.process(req, async req => {
+        return (await this.options?.fetchMethod?.(req)) ?? (await fetch(req))
       })
-    )
+    }
+    return (await this.options?.fetchMethod?.(req)) ?? (await fetch(req))
   }
 
-  async put(path: string, body?: BodyInit | null): Promise<Response> {
-    return await this.request(
-      new Request(this.buildUrl(path), {
-        method: 'PUT',
-        body
-      })
-    )
+  /**
+   * Build URL.
+   * @param path
+   */
+  buildUrl(path: string): URL {
+    if (this.options !== undefined && this.options.urlBuilder !== undefined) {
+      return this.options.urlBuilder(path)
+    }
+    if (this.host.endsWith("/") && path.startsWith("/")) {
+      return new URL(`${this.host}${path.slice(1)}`)
+    }
+    if (!this.host.endsWith("/") && !path.startsWith("/")) {
+      return new URL(`${this.host}/${path}`)
+    }
+    return new URL(`${this.host}${path}`)
   }
 
-  async delete(path: string): Promise<Response> {
-    return await this.request(
-      new Request(this.buildUrl(path), {
-        method: 'DELETE'
-      })
-    )
+  /**
+   * Create a request builder.
+   * @param path
+   */
+  requestBuilder(path: string): RequestBuilder {
+    return new RequestBuilder(this.buildUrl(path), this.request.bind(this))
   }
-}
 
-export class RpcServiceImpl extends RpcService {
-  constructor(serviceContext: ServiceContext, public readonly host: string) {
-    super(serviceContext)
+  /**
+   * Send a GET request.
+   * @param path
+   * @param options
+   */
+  async get(path: string, options?: RequestOptions): Promise<Response> {
+    return this.requestBuilder(path)
+      .searchParams(options?.searchParams)
+      .headers(options?.headers)
+      .sendGet()
   }
 
-  override async request(req: Request): Promise<Response> {
-    const middlewareService = this.resolve(MiddlewareService)
-    if (middlewareService !== undefined) {
-      return await middlewareService.process(req, async req => {
-        return await fetch(req)
-      })
-    } else {
-      return await fetch(req)
-    }
+  /**
+   * Send a POST request.
+   * @param path
+   * @param body JSON object
+   * @param options
+   */
+  async post(
+    path: string,
+    body: object | null | undefined,
+    options?: RequestOptions
+  ): Promise<Response> {
+    return this.requestBuilder(path)
+      .searchParams(options?.searchParams)
+      .headers(options?.headers)
+      .sendPostJson(body)
   }
 
-  override buildUrl(path: string): string {
-    if (this.host.endsWith('/') && path.startsWith('/')) {
-      return `${this.host}${path.slice(1)}`
-    }
-    return `${this.host}${path}`
+  /**
+   * Send a PUT request.
+   * @param path
+   * @param body JSON object
+   * @param options
+   */
+  async put(
+    path: string,
+    body: object | null | undefined,
+    options?: RequestOptions
+  ): Promise<Response> {
+    return this.requestBuilder(path)
+      .searchParams(options?.searchParams)
+      .headers(options?.headers)
+      .sendPutJson(body)
+  }
+
+  /**
+   * Send a DELETE request.
+   * @param path
+   * @param options
+   */
+  async delete(path: string, options?: RequestOptions): Promise<Response> {
+    return this.requestBuilder(path)
+      .searchParams(options?.searchParams)
+      .headers(options?.headers)
+      .sendDelete()
   }
 }

package/src/services/service.ts

@@ -1,12 +1,13 @@
-import { type Context } from '../internal/context'
-import { type Constructor } from '../internal/registry'
-import { type DisposableContainer, type Lifetime } from '../disposable'
+import { type Context } from "../context"
+import { type Constructor } from "../internal/registry"
+import { type DisposableContainer, type Lifetime } from "../disposable"
 
 export class ServiceContext {
   constructor(
     public readonly context: Context,
     private readonly disposableContainer: DisposableContainer
-  ) {}
+  ) {
+  }
 
   public get lifetime(): Lifetime {
     return this.disposableContainer.lifetime
@@ -16,15 +17,15 @@ export class ServiceContext {
     return this.context.resolve(type)
   }
 
-  public async onRegister(): Promise<void> {
+  public onRegister: () => Promise<void> = async (): Promise<void> => {
     await Promise.resolve()
   }
 
-  public async onStart(): Promise<void> {
+  public onStart: () => Promise<void> = async (): Promise<void> => {
     await Promise.resolve()
   }
 
-  public onUnregister(): void {
+  public onUnregister: () => void = (): void => {
     // do nothing
   }
 }
@@ -42,5 +43,6 @@ export abstract class Service {
     return this.serviceContext.context
   }
 
-  public constructor(private readonly serviceContext: ServiceContext) {}
+  public constructor(private readonly serviceContext: ServiceContext) {
+  }
 }

package/src/services/userProfileService.ts

@@ -0,0 +1,38 @@
+import { Service } from "./service"
+import { RpcService } from "./rpcService"
+import { UserProfile } from "../storages/user/userProfile"
+import { UserInfoResponse } from "../dto/userInfoResponse"
+import { OrganizationService } from "./organizationService"
+import { UserProfileImpl } from "../storages/user/userProfile.impl"
+import { ResponseUtils } from "./responseUtils"
+
+export class UserProfileService extends Service {
+  private readonly impl: UserProfileImpl = new UserProfileImpl()
+
+  get userProfile(): UserProfile {
+    return this.impl
+  }
+
+  async fetch() {
+    const rpc = this.resolve(RpcService) as RpcService
+    const response = await rpc.requestBuilder("api/v1/Users/self2").sendGet()
+    if (ResponseUtils.isFail(response)) {
+      await ResponseUtils.throwError("Failed to fetch user profile", response)
+    }
+    const content = (await response.json()) as UserInfoResponse
+
+    // init user profile from the server's response
+    this.impl.initFrom(content)
+
+    const organizationService = this.resolve(
+      OrganizationService
+    ) as OrganizationService
+
+    // init organization service from user profile
+    await organizationService.initFrom(
+      content.adminInOrganization,
+      content.organizations,
+      content.user.settings
+    )
+  }
+}

package/src/storages/chats/answer.impl.ts

@@ -0,0 +1,163 @@
+import { Context } from "../../context"
+import {
+  AnswerDto,
+  AnswerStatus,
+  AnswerStepDto,
+  FetchAnswerResponse,
+  FetchTokensResponse,
+  SourceDto,
+  StepType
+} from "../../dto/chatResponse"
+import { ResponseUtils } from "../../services/responseUtils"
+import { RpcService } from "../../services/rpcService"
+import { Answer, AnswerId } from "./answer"
+import { Chat } from "./chat"
+
+export class AnswerImpl extends Answer {
+  private _content?: AnswerDto
+
+  private _steps?: AnswerStepDto[]
+  private _status?: AnswerStatus
+  private _id?: AnswerId
+
+  constructor(
+    private readonly chat: Chat,
+    private readonly context: Context) {
+    super()
+  }
+
+  async initFromData(answer: AnswerDto): Promise<AnswerImpl> {
+    this._content = answer
+    this._id = answer.id
+
+    // fetch answer
+    await this.fetch()
+
+    return this
+  }
+
+  async initFromId(id: AnswerId): Promise<AnswerImpl> {
+    this._id = id
+
+    // fetch answer
+    await this.fetch()
+
+    return this
+  }
+
+  get id(): string {
+    return <string>this._id
+  }
+
+  get status(): AnswerStatus {
+    return <AnswerStatus>this._status
+  }
+
+  private getStep(type: StepType): AnswerStepDto | undefined {
+    return this._steps?.find(step => step.type === type)
+  }
+
+  async sources(type: StepType): Promise<SourceDto[]> {
+    // fetch answer
+    await this.fetch()
+    // get step
+    const step = this.getStep(type)
+
+    // check step
+    if (!step) {
+      throw new Error(`Step with type ${type.toString()} is not found, answer: ${this.id}, organization: ${this.chat.organization.id}`)
+    }
+
+    // get sources
+    const response = await this.context
+      .resolve(RpcService)
+      ?.requestBuilder("api/v1/Chats/answer/sources")
+      .searchParam("chat_uid", this.chat.id)
+      .searchParam("uid", this.id)
+      .searchParam("step_id", step.id)
+      .sendGet()
+
+    // check response status
+    if (ResponseUtils.isFail(response)) {
+      await ResponseUtils.throwError(`Failed to get sources for ${type.toString()}, answer: ${this.id}, organization: ${this.chat.organization.id}`, response)
+    }
+
+    // parse sources from the server's response
+    const sources = (await response!.json()).sources as SourceDto[]
+
+    return sources
+  }
+
+  async fetch(): Promise<void> {
+    // fetch answer from position 0
+    const position = 0
+    // fetch answer
+    const response = await this.context
+      .resolve(RpcService)
+      ?.requestBuilder("api/v1/Chats/answer/fetch")
+      .searchParam("chatId", this.chat.id)
+      .searchParam("questionId", this.id)
+      .searchParam("position", position.toString())
+      .sendGet()
+
+    // check response status
+    if (ResponseUtils.isFail(response)) {
+      await ResponseUtils.throwError(`Failed to fetch answer ${this.id}`, response)
+    }
+
+    // parse answer from the server's response
+    const answer = (await response!.json()) as FetchAnswerResponse
+
+    // update answer
+    this._status = <AnswerStatus>answer.status
+    this._steps = <AnswerStepDto[]>answer.steps
+  }
+
+  async fetchTokens(type: StepType, token_start_at: number): Promise<FetchTokensResponse> {
+    // fetch answer
+    await this.fetch()
+    // get step
+    const step = this.getStep(type)
+
+    // check step
+    if (!step) {
+      throw new Error(`Step with type ${type.toString()} is not found`)
+    }
+
+    // get tokens
+    const response = await this.context
+      .resolve(RpcService)
+      ?.requestBuilder("api/v1/Chats/answer/fetch/tokens")
+      .searchParam("chat_uid", this.chat.id)
+      .searchParam("uid", this.id)
+      .searchParam("step_id", step.id)
+      .searchParam("token_start_at", token_start_at.toString())
+      .sendGet()
+
+    // check response status
+    if (ResponseUtils.isFail(response)) {
+      await ResponseUtils.throwError(`Failed to get sources for ${type.toString()}`, response)
+    }
+
+    // parse tokens from the server's response
+    const tokens = (await response!.json()) as FetchTokensResponse
+
+    return tokens
+  }
+
+  async cancel(): Promise<void> {
+    // send request to the server
+    const response = await this.context
+      .resolve(RpcService)
+      ?.requestBuilder("api/v1/Chats/answer/cancel")
+      .sendPutJson({
+        chat_id: this.chat.id,
+        uid: this.id
+      })
+
+    // check response status
+    if (ResponseUtils.isFail(response)) {
+      await ResponseUtils.throwError("Failed to cancel a question", response)
+    }
+  }
+}

package/src/storages/chats/answer.ts

@@ -0,0 +1,42 @@
+import {
+  AnswerStatus,
+  FetchTokensResponse,
+  SourceDto,
+  StepType
+} from "../../dto/chatResponse"
+
+export type AnswerId = string
+export type StepId = string
+
+export abstract class Answer {
+
+  /**
+   * Answer id.
+   */
+  abstract get id(): AnswerId
+
+  /**
+   * Answer status.
+   */
+  abstract get status(): AnswerStatus
+
+  /**
+   * Answer sources.
+   */
+  abstract sources(type: StepType): Promise<SourceDto[]>
+
+  /**
+   * Fetch answer.
+   */
+  abstract fetch(): Promise<void>
+
+  /**
+   * Fetch answer.
+   */
+  abstract fetchTokens(type: StepType, tokenStartAt: number): Promise<FetchTokensResponse>
+
+  /**
+   * Cancel answer
+   */
+  abstract cancel(): Promise<void>
+}