import { Injectable } from "@nestjs/common";
import { InjectModel } from "@nestjs/mongoose";
import { Model } from "mongoose";
import { Custodial } from "./entities/custodial.entity";
import { CustodialDocument } from "./schemas/custodial.schema";

@Injectable()
export class InternalCustodialService {
  constructor(
    @InjectModel(Custodial.name)
    private readonly custodialModel: Model<CustodialDocument>,
  ) {}
  async create(createCustodial: Custodial): Promise<CustodialDocument> {
    const custodial = new this.custodialModel({
      ...createCustodial,
    });
    return custodial.save();
  }
  async findOne(address: string): Promise<CustodialDocument> {
    return this.custodialModel.findOne({ address });
  }
}