package com.qianwen.core.oss;

import com.obs.services.ObsClient;
import com.obs.services.model.GetObjectRequest;
import com.obs.services.model.ObjectMetadata;
import com.obs.services.model.ObsObject;
import com.obs.services.model.PutObjectResult;
import java.io.InputStream;
import java.math.BigDecimal;
import java.util.List;
import com.qianwen.core.oss.model.BladeFile;
import com.qianwen.core.oss.model.OssFile;
import com.qianwen.core.oss.props.OssProperties;
import com.qianwen.core.oss.rule.OssRule;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;

/* loaded from: blade-starter-oss-9.3.0.0-SNAPSHOT.jar:org/springblade/core/oss/HuaweiObsTemplate.class */
public class HuaweiObsTemplate implements OssTemplate {
    private final ObsClient obsClient;
    private final OssProperties ossProperties;
    private final OssRule ossRule;

    public HuaweiObsTemplate(final ObsClient obsClient, final OssProperties ossProperties, final OssRule ossRule) {
        this.obsClient = obsClient;
        this.ossProperties = ossProperties;
        this.ossRule = ossRule;
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public void makeBucket(String bucketName) {
        if (!bucketExists(bucketName)) {
            this.obsClient.createBucket(getBucketName(bucketName));
        }
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public void removeBucket(String bucketName) {
        this.obsClient.deleteBucket(getBucketName(bucketName));
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public boolean bucketExists(String bucketName) {
        return this.obsClient.headBucket(getBucketName(bucketName));
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public void copyFile(String bucketName, String fileName, String destBucketName) {
        this.obsClient.copyObject(getBucketName(bucketName), fileName, getBucketName(destBucketName), fileName);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public void copyFile(String bucketName, String fileName, String destBucketName, String destFileName) {
        this.obsClient.copyObject(getBucketName(bucketName), fileName, getBucketName(destBucketName), destFileName);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public OssFile statFile(String fileName) {
        return statFile(this.ossProperties.getBucketName(), fileName);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public OssFile statFile(String bucketName, String fileName) {
        ObjectMetadata stat = this.obsClient.getObjectMetadata(getBucketName(bucketName), fileName);
        OssFile ossFile = new OssFile();
        ossFile.setName(fileName);
        ossFile.setLink(fileLink(ossFile.getName()));
        ossFile.setHash(stat.getContentMd5());
        ossFile.setLength(stat.getContentLength().longValue());
        ossFile.setPutTime(stat.getLastModified());
        ossFile.setContentType(stat.getContentType());
        return ossFile;
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public String filePath(String fileName) {
        return getOssHost(getBucketName()).concat("/").concat(fileName);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public String filePath(String bucketName, String fileName) {
        return getOssHost(getBucketName(bucketName)).concat("/").concat(fileName);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public String fileLink(String fileName) {
        return getOssHost().concat("/").concat(fileName);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public String fileLink(String bucketName, String fileName) {
        return getOssHost(getBucketName(bucketName)).concat("/").concat(fileName);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public InputStream getObject(String bucketName, String fileName, Long offset, Long length) {
        return get(bucketName, fileName, offset, length);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public InputStream getObject(String fileName, Long offset, Long length) {
        return get(this.ossProperties.getBucketName(), fileName, offset, length);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public InputStream getObject(String bucketName, String fileName) {
        return get(bucketName, fileName);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public InputStream getObject(String fileName) {
        return getObject(this.ossProperties.getBucketName(), fileName);
    }

    public InputStream get(String bucketName, String key) {
        ObsObject ossObject = this.obsClient.getObject(getBucketName(bucketName), key);
        return ossObject.getObjectContent();
    }

    public InputStream get(String bucketName, String key, Long offset, Long length) {
        GetObjectRequest getObjectRequest = new GetObjectRequest(getBucketName(bucketName), key);
        getObjectRequest.setRangeStart(offset);
        getObjectRequest.setRangeEnd(Long.valueOf(new BigDecimal(offset.longValue()).add(new BigDecimal(length.longValue())).longValue()));
        ObsObject ossObject = this.obsClient.getObject(getObjectRequest);
        return ossObject.getObjectContent();
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public BladeFile putFile(MultipartFile file) {
        return putFile(this.ossProperties.getBucketName(), file.getOriginalFilename(), file);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public BladeFile putFile(String fileName, MultipartFile file) {
        return putFile(this.ossProperties.getBucketName(), fileName, file);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public BladeFile putFile(String bucketName, String fileName, MultipartFile file) {

		try {
			return putFile(bucketName, fileName, file.getInputStream());
		} catch (Exception ex) {
			throw new RuntimeException(ex);
		}
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public BladeFile putFile(String fileName, InputStream stream) {
        return putFile(this.ossProperties.getBucketName(), fileName, stream);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public BladeFile putFile(String bucketName, String fileName, InputStream stream) {
        return put(bucketName, stream, fileName, false);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public void removeFile(String fileName) {
        this.obsClient.deleteObject(getBucketName(), fileName);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public void removeFile(String bucketName, String fileName) {
        this.obsClient.deleteObject(getBucketName(bucketName), fileName);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public void removeFiles(List<String> fileNames) {
        fileNames.forEach(this::removeFile);
    }

    @Override // com.qianwen.core.oss.OssTemplate
    public void removeFiles(String bucketName, List<String> fileNames) {
        fileNames.forEach(fileName -> {
            removeFile(getBucketName(bucketName), fileName);
        });
    }

    public BladeFile put(String bucketName, InputStream stream, String key, boolean cover) {
        makeBucket(bucketName);
        String key2 = getFileName(key);
        if (cover) {
            this.obsClient.putObject(getBucketName(bucketName), key2, stream);
        } else {
            PutObjectResult response = this.obsClient.putObject(getBucketName(bucketName), key2, stream);
            for (int retry = 0; StringUtils.isEmpty(response.getEtag()) && retry < 5; retry++) {
                response = this.obsClient.putObject(getBucketName(bucketName), key2, stream);
            }
        }
        BladeFile file = new BladeFile();
        file.setOriginalName(key);
        file.setName(key2);
        file.setLink(fileLink(bucketName, key2));
        return file;
    }

    private String getFileName(String originalFilename) {
        return this.ossRule.fileName(originalFilename);
    }

    private String getBucketName() {
        return getBucketName(this.ossProperties.getBucketName());
    }

    private String getBucketName(String bucketName) {
        return this.ossRule.bucketName(bucketName);
    }

    public String getOssHost(String bucketName) {
        String prefix = this.ossProperties.getEndpoint().contains("https://") ? "https://" : "http://";
        return prefix + getBucketName(bucketName) + "." + this.ossProperties.getEndpoint().replaceFirst(prefix, "");
    }

    public String getOssHost() {
        return getOssHost(this.ossProperties.getBucketName());
    }
}