package com.ruoyi.basic.controller;

import com.ruoyi.basic.pojo.EnterpriseInfo;
import com.ruoyi.basic.service.IEnterpriseInfoService;
import com.ruoyi.framework.web.domain.AjaxResult;
import com.ruoyi.other.service.TempFileService;
import io.swagger.annotations.ApiOperation;
import lombok.AllArgsConstructor;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;

@RestController
@RequestMapping("/system/enterpriseInfo")
@AllArgsConstructor
public class EnterpriseInfoController {

    private final IEnterpriseInfoService enterpriseInfoService;
    private final TempFileService tempFileService;

    /**
     * 获取企业信息
     */
    @GetMapping("/getInfo")
    public AjaxResult getInfo() {
        return AjaxResult.success(enterpriseInfoService.getEnterpriseInfo());
    }

    /**
     * 保存或更新企业信息
     */
    @PostMapping("/save")
    public AjaxResult save(@RequestBody EnterpriseInfo enterpriseInfo) {
        boolean result = enterpriseInfoService.saveOrUpdateInfo(enterpriseInfo);
        if (result) {
            return AjaxResult.success("保存成功");
        }
        return AjaxResult.error("保存失败");
    }

    /**
     * 上传企业Logo
     */
    @PostMapping("/uploadLogo")
    @ApiOperation(value = "上传企业Logo")
    public AjaxResult uploadLogo(@RequestParam("file") MultipartFile file) {
        try {
            return AjaxResult.success(tempFileService.uploadFile(file, 0));
        } catch (IOException e) {
            return AjaxResult.error("上传失败：" + e.getMessage());
        }
    }

    /**
     * 上传二维码
     */
    @PostMapping("/uploadQrCode")
    @ApiOperation(value = "上传二维码")
    public AjaxResult uploadQrCode(@RequestParam("file") MultipartFile file) {
        try {
            return AjaxResult.success(tempFileService.uploadFile(file, 1));
        } catch (IOException e) {
            return AjaxResult.error("上传失败：" + e.getMessage());
        }
    }
}