import { basicApi } from './apiBasic'
import { define } from '../define/define'
import { promises as fspromises } from 'fs'

export class SdApi {
  constructor() {
    this.baseUrl = global.config?.webui_api_url
    this.sd_setting = null
  }

  /**
   * 获取当前SD的服务器中所有的lora信息
   * @returns
   */
  async getAllLoras(baseURL = null) {
    let url = this.baseUrl + 'sdapi/v1/loras'
    if (baseURL != null) {
      url = baseURL + 'sdapi/v1/loras'
    }
    return await basicApi.get(url)
  }

  /**
   * 获取当前的所有的checkpoint模型
   * @param {*} baseURL
   */
  async getAllSDModel(baseURL = null) {
    let url = this.baseUrl + 'sdapi/v1/sd-models'
    if (baseURL != null) {
      url = baseURL + 'sdapi/v1/sd-models'
    }
    return await basicApi.get(url)
  }

  /**
   * 获取当前连接的所有的samplers（采样器）
   * @param {*} baseURL
   * @returns
   */
  async getAllSamplers(baseURL = null) {
    try {
      let url = this.baseUrl + 'sdapi/v1/samplers'
      if (baseURL != null) {
        url = baseURL + 'sdapi/v1/samplers'
      }
      return await basicApi.get(url)
    } catch (error) {
      throw error
    }
  }

  async txt2img(data, baseURL = null) {
    try {
      if (this.sd_setting == null) {
        this.sd_setting = JSON.parse(await fspromises.readFile(define.sd_setting, 'utf-8'))
        this.baseUrl = this.sd_setting.setting.webui_api_url
      }

      // 加上通用前缀
      data.prompt = this.sd_setting.webui.prompt + data.prompt

      data.negative_prompt = this.sd_setting.webui.negative_prompt
      data.sampler_name = this.sd_setting.webui.sampler_name
      data.cfg_scale = this.sd_setting.webui.cfg_scale
      data.n_iter = 1
      data.steps = this.sd_setting.webui.steps
      data.save_images = false
      data.batch_size = data.batch_size ? data.batch_size : 1

      if (data.width == null) {
        data.width = 512
      }
      if (data.height == null) {
        data.height = 512
      }

      let url = this.baseUrl + 'sdapi/v1/txt2img'
      if (baseURL != null) {
        url = baseURL + 'sdapi/v1/txt2img'
      }
      let res = await basicApi.post(url, data)
      return res
    } catch (error) {
      throw error
    }
  }
}