TestObject.nitro.ts

import {
  type HybridObject,
  type AnyMap,
  type Sync,
  type CustomType,
  type AnyHybridObject,
  type UInt64,
  type Int64,
} from 'react-native-nitro-modules'
import type { TestView } from './TestView.nitro'
import type { SomeExternalObject } from 'react-native-nitro-test-external'
import type { Child } from './Child.nitro'
import type { Base } from './Base.nitro'

// Tuples become `std::tuple<...>` in C++.
// In contrast to arrays, they are length-checked, and can have different types inside them.
export type Float2 = [number, number]
export type Float3 = [number, number, number]
export type TestTuple = [number, string, boolean]

// Variants can have aliases/names
export type NamedVariant = string | Car

// A discriminating string union becomes an `enum` in C++.
// This one is string-backed.
export type Powertrain = 'electric' | 'gas' | 'hybrid'
type TruckPowertrain = Powertrain | 'diesel'

// A classic TypeScript enum also becomes an `enum` in C++.
// This one is number-backed.
export enum OldEnum {
  FIRST,
  SECOND,
  THIRD,
}

// Backed by numbers, but with custom number values.
export enum WeirdNumbersEnum {
  A = 0,
  B = 32,
  C = 64,
}

// A plain interface that does not inherit from `HybridObject` becomes a `struct` in C++.
// They can only have properties (get + set). No methods or native state.
export interface Car {
  year: number
  make: string
  model: string
  power: number
  powertrain: Powertrain
  driver?: Person
  passengers: Person[]
  isFast?: boolean
  favouriteTrack?: string
  performanceScores: number[]
  someVariant?: number | string
}

// A `type T = { ... }` declaration is the same as a `interface T { ... }` - it's a `struct` in C++.
export type Person = {
  name: string
  age: number
}

interface JsStyleStruct {
  value: number
  onChanged: (num: number) => void
}
export interface WrappedJsStruct {
  value: JsStyleStruct
  items: JsStyleStruct[]
}
export interface OptionalWrapper {
  optionalArrayBuffer?: ArrayBuffer
  optionalString?: string
}
// A wide optional struct mixing primitive optionals with string-union enum optionals.
export interface OptionalEnumWrapper {
  count?: number
  weight?: number
  ttl?: number
  jitter?: number
  retries?: number
  delayMs?: number
  timeoutMs?: number
  ratio?: number
  threshold?: number
  enabled?: boolean
  active?: boolean
  shouldBuffer?: boolean
  shouldRetry?: boolean
  verbose?: boolean
  tier?: Powertrain
  region?: Powertrain
  stage?: Powertrain
  tone?: Powertrain
}
export interface ExternalObjectStruct {
  someExternal: SomeExternalObject
}
export type StringOrExternal = string | SomeExternalObject

interface OptionalCallback {
  callback?: (() => void) | number
}

interface SecondMapWrapper {
  second: Record<string, string>
}
interface MapWrapper {
  map: Record<string, string>
  secondMap: SecondMapWrapper
}
export type CustomString = CustomType<
  string,
  'CustomString',
  { include: 'CustomString.hpp' }
>

type CoreTypesVariant =
  | ArrayBuffer
  | Promise<number>
  | ((value: number) => void)
  | WrappedJsStruct
  | Date
  | AnyMap

// Prefer `interface` + `extends` over `type` so TS doesn't flatten it
interface PartialPerson extends Partial<Person> {}

// This is an `interface` we're going to use as a base in both of our `HybridObject`s later.
// In this case, the `HybridObject`s will just flatten out and copy over all properties here.
// There is no separate type for `SharedTestObjectProps` on the native side.
interface SharedTestObjectProps {
  // Test Primitives
  numberValue: number
  boolValue: boolean
  stringValue: string
  int64Value: Int64
  uint64Value: UInt64
  nullValue: null
  optionalString?: string
  stringOrUndefined: string | undefined
  stringOrNull: string | null
  optionalArray?: string[]
  optionalEnum?: Powertrain
  optionalOldEnum?: OldEnum
  optionalEnumPlusOne?: TruckPowertrain
  optionalCallback?: (value: number) => void

  // Kotlin/Swift simplify boolean names (has*/is*)
  readonly hasBoolean: boolean
  readonly isBoolean: boolean
  hasBooleanWritable: boolean
  isBooleanWritable: boolean

  // Basic function tests
  simpleFunc(): void
  addNumbers(a: number, b: number): number
  addStrings(a: string, b: string): string
  multipleArguments(num: number, str: string, boo: boolean): void
  bounceNull(value: null): null

  // Arrays
  bounceStrings(array: string[]): string[]
  bounceNumbers(array: number[]): number[]
  bounceStructs(array: Person[]): Person[]
  bounceNestedArray(array: Person[][]): Person[][]
  bouncePartialStruct(person: PartialPerson): PartialPerson
  sumUpAllPassengers(cars: Car[]): string
  bounceEnums(array: Powertrain[]): Powertrain[]
  complexEnumCallback(
    array: Powertrain[],
    callback: (array: Powertrain[]) => void
  ): void

  // Arrays (complex)
  bounceHybridObjects(array: Child[]): Child[]
  bounceFunctions(functions: (() => void)[]): (() => void)[]
  bounceMaps(maps: AnyMap[]): AnyMap[]
  bouncePromises(promises: Promise<number>[]): Promise<number>[]
  bounceArrayBuffers(arrayBuffers: ArrayBuffer[]): ArrayBuffer[]

  // Maps
  createMap(): AnyMap
  mapRoundtrip(map: AnyMap): AnyMap
  getMapKeys(map: AnyMap): string[]
  mergeMaps(a: AnyMap, b: AnyMap): AnyMap
  copyAnyMap(map: AnyMap): AnyMap

  // Typed Maps (records)
  bounceMap(
    map: Record<string, number | boolean>
  ): Record<string, number | boolean>
  bounceSimpleMap(map: Record<string, number>): Record<string, number>
  extractMap(mapWrapper: MapWrapper): MapWrapper['map']

  // Errors
  funcThatThrows(): number
  funcThatThrowsBeforePromise(): Promise<void>
  throwError(error: Error): void

  // Optional parameters
  tryOptionalParams(num: number, boo: boolean, str?: string): string
  tryMiddleParam(num: number, boo: boolean | undefined, str: string): string
  tryOptionalEnum(value?: Powertrain): Powertrain | undefined
  tryTrailingOptional(num: number, str: string, boo?: boolean): boolean

  // Variants
  someVariant: number | string

  // Dates
  add1Hour(date: Date): Date
  currentDate(): Date

  // Promises
  calculateFibonacciSync(value: number): Int64
  calculateFibonacciAsync(value: number): Promise<Int64>
  wait(seconds: number): Promise<void>
  promiseThrows(): Promise<void>
  promiseReturnsInstantly(): Promise<number>
  promiseReturnsInstantlyAsync(): Promise<number>
  promiseThatResolvesVoidInstantly(): Promise<void>
  promiseThatResolvesToUndefined(): Promise<number | undefined>

  // Complex Promises
  awaitAndGetPromise(promise: Promise<number>): Promise<number>
  awaitAndGetComplexPromise(promise: Promise<Car>): Promise<Car>
  awaitPromise(promise: Promise<void>): Promise<void>

  // Callbacks
  callCallback(callback: () => void): void
  callCallbackThatReturnsPromiseVoid(
    callback: () => Promise<void>
  ): Promise<void>
  callAll(first: () => void, second: () => void, third: () => void): void
  callWithOptional(
    value: number | undefined,
    callback: (maybe: number | undefined) => void
  ): void
  callSumUpNTimes(callback: () => number, n: number): Promise<number>
  callbackAsyncPromise(callback: () => Promise<number>): Promise<number>
  callbackAsyncPromiseBuffer(
    callback: () => Promise<ArrayBuffer>
  ): Promise<ArrayBuffer>
  getComplexCallback(): (value: number) => void
  twoOptionalCallbacks(
    value: number,
    first?: (value: number) => void,
    second?: (value: string) => void
  ): void
  errorCallback(onError: (error: Error) => void): void
  createNativeCallback(
    wrappingJsCallback: (num: number) => void
  ): (num: number) => void

  // Callbacks that return values
  getValueFromJSCallbackAndWait(getValue: () => number): Promise<number>
  getValueFromJsCallback(
    callback: () => string,
    andThenCall: (valueFromJs: string) => void
  ): Promise<void>

  // Objects
  getCar(): Car
  isCarElectric(car: Car): boolean
  areCarsEqual(a: Car, b: Car): boolean
  getDriver(car: Car): Person | undefined
  bounceCar(car: Car): Car
  jsStyleObjectAsParameters(params: JsStyleStruct): void
  bounceWrappedJsStyleStruct(value: WrappedJsStruct): WrappedJsStruct
  bounceOptionalWrapper(wrapper: OptionalWrapper): OptionalWrapper
  bounceOptionalCallback(value: OptionalCallback): OptionalCallback
  bounceOptionalEnumStruct(
    value?: OptionalEnumWrapper
  ): OptionalEnumWrapper | undefined

  // ArrayBuffers
  createArrayBuffer(): ArrayBuffer
  createArrayBufferFromNativeBuffer(copy: boolean): ArrayBuffer
  copyBuffer(buffer: ArrayBuffer): ArrayBuffer
  getBufferLastItem(buffer: ArrayBuffer): number
  setAllValuesTo(buffer: ArrayBuffer, value: number): void
  createArrayBufferAsync(): Promise<ArrayBuffer>
  bounceArrayBuffer(buffer: ArrayBuffer): ArrayBuffer

  // Complex variants
  passVariant(
    either: number | string | number[] | string[] | boolean
  ): number | string
  getVariantEnum(variant: OldEnum | boolean): OldEnum | boolean
  bounceVariantUnionEnum(variant: Powertrain | Car): Powertrain | Car
  getVariantWeirdNumbersEnum(
    variant: WeirdNumbersEnum | boolean
  ): WeirdNumbersEnum | boolean
  getVariantObjects(variant: Person | Car): Person | Car
  passNamedVariant(variant: NamedVariant): NamedVariant
  passAllEmptyObjectVariant(
    variant: OptionalWrapper | Base
  ): OptionalWrapper | Base
  bounceComplexVariant(variant: CoreTypesVariant): CoreTypesVariant

  // Inheritance
  createChild(): Child
  createBase(): Base
  createBaseActualChild(): Base
  bounceChild(child: Child): Child
  bounceBase(base: Base): Base
  bounceChildBase(child: Child): Base
  castBase(base: Base): Child

  // Sync funcs
  callbackSync(callback: Sync<() => number>): number

  // Views
  getIsViewBlue(view: TestView): boolean

  // External HybridObjects
  bounceExternalHybrid(externalObject: SomeExternalObject): SomeExternalObject
  createInternalObject(): SomeExternalObject
  bounceExternalStruct(
    externalStruct: ExternalObjectStruct
  ): ExternalObjectStruct
  bounceExternalVariant(variant: StringOrExternal): StringOrExternal
  createExternalVariantFromFunc(
    factory: Sync<() => SomeExternalObject>
  ): SomeExternalObject
}

// This is a C++-based `HybridObject`.
// Since it inherited from the `SharedTestObjectProps` interface,
// it will be flattened out and every property/method will be added here.
export interface TestObjectCpp
  extends HybridObject<{ ios: 'c++'; android: 'c++' }>, SharedTestObjectProps {
  // Complex Variants + Tuples
  getVariantTuple(variant: Float2 | Float3): Float2 | Float3

  // Tuples
  someTuple: [number, string]
  flip(tuple: Float3): Float3
  passTuple(tuple: TestTuple): [number, string, boolean]

  // Type-specifics
  readonly thisObject: TestObjectCpp
  newTestObject(): TestObjectCpp
  // Returns a HybridObject from a Promise resolved off the JS thread. Used to reproduce the data
  // race between resolve() and JSIConverter::toJSI.
  newTestObjectAsync(): Promise<TestObjectCpp>
  optionalHybrid?: TestObjectCpp
  getVariantHybrid(variant: TestObjectCpp | Person): TestObjectCpp | Person
  getVariantHybridEnum(
    variant: TestObjectCpp | Powertrain
  ): TestObjectCpp | Powertrain

  // Any HybridObject
  bounceAnyHybrid(object: AnyHybridObject): AnyHybridObject

  // Custom C++ JSI Converters
  bounceCustomType(value: CustomString): CustomString
}

// This is a Swift/Kotlin-based `HybridObject`.
// Since it inherited from the `SharedTestObjectProps` interface,
// it will be flattened out and every property/method will be added here.
export interface TestObjectSwiftKotlin
  extends
    HybridObject<{ ios: 'swift'; android: 'kotlin' }>,
    SharedTestObjectProps {
  // Type-specifics
  readonly thisObject: TestObjectSwiftKotlin
  newTestObject(): TestObjectSwiftKotlin
  newTestObjectAsync(): Promise<TestObjectSwiftKotlin>
  optionalHybrid?: TestObjectSwiftKotlin
  getVariantHybrid(
    variant: TestObjectSwiftKotlin | Person
  ): TestObjectSwiftKotlin | Person
}