io.github.robothy.sdwebui.sdk.services.DefaultSdWebuiBeanContainer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of sdwebui-java-sdk Show documentation

Stable Diffusion Web UI Java SDK

The newest version!

package io.github.robothy.sdwebui.sdk.services;

import com.fasterxml.jackson.databind.ObjectMapper;
import io.github.robothy.sdwebui.sdk.*;
import io.github.robothy.sdwebui.sdk.models.SdWebuiOptions;
import io.github.robothy.sdwebui.sdk.models.SystemInfo;
import org.apache.hc.client5.http.classic.HttpClient;
import org.apache.hc.client5.http.impl.classic.CloseableHttpClient;
import org.apache.hc.client5.http.impl.classic.HttpClients;

import java.util.HashMap;
import java.util.Map;

public class DefaultSdWebuiBeanContainer implements SdWebuiBeanContainer {

  private final Map, Object> services;

  private final SdWebuiOptions sdWebuiOptions;

  public DefaultSdWebuiBeanContainer(SdWebuiOptions options) {
    this.services = new HashMap<>();
    this.sdWebuiOptions = options;
    init();
  }

  @SuppressWarnings("unchecked")
  @Override
  public  T getBean(Class serviceClass) {
    T instance = (T) this.services.get(serviceClass);
    if (instance == null) {
      throw new RuntimeException("No service or service factory found for " + serviceClass);
    }

    if (instance instanceof SdWebuiBeanFactory) {
        return ((SdWebuiBeanFactory) instance).getBean(this);
    }

    return instance;
  }

  @Override
  public void register(Class serviceClass, Object service) {
    this.services.put(serviceClass, service);
  }

  private void init() {
    CloseableHttpClient closeableHttpClient = HttpClients.createDefault();
    register(SdWebuiOptions.class, sdWebuiOptions);
    register(ObjectMapper.class, new ObjectMapper());
    register(HttpClient.class, closeableHttpClient);
    register(SystemInfo.class, new CacheableSystemInfoFetcher(sdWebuiOptions.getEndpoint(), this));
    register(Txt2Image.class, new DefaultTxt2ImageService(this));
    register(Image2Image.class, new DefaultImage2ImageService(this));
    register(CommonGetService.class, new CommonGetService(this));
    register(GetSdModels.class, new DefaultGetSdModelService(this));
    register(GetFaceRestorers.class, new DefaultGetFaceRestorersService(this));
    register(ExtraImage.class, new DefaultExtraImageService(this));
  }

}