mobx-lark 2.7.0 → 2.8.1

package/src/module/DocumentAI/index.ts

@@ -0,0 +1,61 @@
+import { makeFormData } from 'koajax';
+import { BaseModel, RESTClient, toggle } from 'mobx-restful';
+
+import { LarkData } from '../../type';
+import { TaxiInvoice, TrainInvoice, VatInvoice, VehicleInvoice } from './type';
+
+export * from './type';
+
+export abstract class DocumentAIModel extends BaseModel {
+    baseURI = 'document_ai/v1';
+
+    abstract client: RESTClient;
+
+    /**
+     * @see {@link https://open.feishu.cn/document/ai/document_ai-v1/vat_invoice/recognize}
+     */
+    @toggle('uploading')
+    async recognizeVatInvoices(file: File) {
+        const { body } = await this.client.post<
+            LarkData<{ vat_invoices: { entities: VatInvoice[] }[] }>
+        >(`${this.baseURI}/vat_invoice/recognize`, makeFormData({ file }));
+
+        return body!.data!.vat_invoices;
+    }
+
+    /**
+     * @see {@link https://open.feishu.cn/document/ai/document_ai-v1/taxi_invoice/recognize}
+     */
+    @toggle('uploading')
+    async recognizeTaxiInvoices(file: File) {
+        const { body } = await this.client.post<
+            LarkData<{ taxi_invoices: { entities: TaxiInvoice[] }[] }>
+        >(`${this.baseURI}/taxi_invoice/recognize`, makeFormData({ file }));
+
+        return body!.data!.taxi_invoices;
+    }
+
+    /**
+     * @see {@link https://open.feishu.cn/document/ai/document_ai-v1/train_invoice/recognize}
+     */
+    @toggle('uploading')
+    async recognizeTrainInvoices(file: File) {
+        const { body } = await this.client.post<
+            LarkData<{ train_invoices: { entities: TrainInvoice[] }[] }>
+        >(`${this.baseURI}/train_invoice/recognize`, makeFormData({ file }));
+
+        return body!.data!.train_invoices;
+    }
+
+    /**
+     * @see {@link https://open.feishu.cn/document/ai/document_ai-v1/vehicle_invoice/recognize}
+     */
+    @toggle('uploading')
+    async recognizeVehicleInvoice(file: File) {
+        const { body } = await this.client.post<
+            LarkData<{ vehicle_invoice: { entities: VehicleInvoice[] } }>
+        >(`${this.baseURI}/vehicle_invoice/recognize`, makeFormData({ file }));
+
+        return body!.data!.vehicle_invoice;
+    }
+}

package/src/module/DocumentAI/type.ts

@@ -0,0 +1,69 @@
+export type InvoiceEntityType =
+    | `invoice_${'code' | 'no' | 'special_seal'}`
+    | `seller_${'name' | 'taxpayer_no'}_in_seal`;
+
+export type TrafficInvoiceEntityType = 'total_amount' | 'price';
+
+export interface VatInvoice {
+    type:
+        | InvoiceEntityType
+        | `invoice_${'name' | 'date'}`
+        | `total_${'price' | 'tax'}`
+        | `${'big_' | ''}total_price_and_tax`
+        | 'check_code'
+        | `buyer_${'name' | 'taxpayer_no' | 'address_phone' | 'account'}`
+        | `seller_${'name' | 'taxpayer_no' | 'address_phone' | 'account'}`
+        | 'payee'
+        | 'password_area'
+        | 'remarks'
+        | 'reviewer'
+        | 'drawer'
+        | 'is_sealed'
+        | 'machine_num';
+    value: string;
+    items: Record<'type' | 'value', string>[][];
+}
+
+export interface TaxiInvoice {
+    type:
+        | InvoiceEntityType
+        | TrafficInvoiceEntityType
+        | 'car_number'
+        | 'start_date'
+        | `${'start' | 'end'}_time`
+        | 'distance'
+        | `${'dispatch' | 'additional'}_fee`
+        | 'is_sealed'
+        | 'title_trial';
+    value: string;
+}
+
+export interface TrainInvoice {
+    type:
+        | TrafficInvoiceEntityType
+        | 'name'
+        | 'id_num'
+        | 'time'
+        | `${'start' | 'end'}_station`
+        | `sale_${'num' | 'station'}`
+        | `${'train' | 'seat' | 'ticket'}_num`
+        | `seat_${'num' | 'cls'}`;
+    value: string;
+}
+
+export interface VehicleInvoice {
+    type:
+        | 'date'
+        | `invoice_${'code' | 'num'}`
+        | `print_${'code' | 'num'}`
+        | `buyer_${'name' | 'id'}`
+        | `saler_${'name' | 'id' | 'addr'}`
+        | 'vehicle_type'
+        | 'product_model'
+        | `${'certificate' | 'machine' | 'engine'}_num`
+        | 'vin'
+        | `tax${'' | '_rate'}`
+        | 'price'
+        | `total_price${'' | '_little'}`;
+    value: string;
+}

package/src/module/index.ts

@@ -8,3 +8,4 @@ export * from './SpreadSheet';
 export * from './BITable';
 export * from './Drive';
 export * from './Wiki';
+export * from './DocumentAI';