package cn.stylefeng.guns.tenant.service;

import cn.stylefeng.guns.base.db.context.DataSourceContext;
import cn.stylefeng.guns.base.db.entity.DatabaseInfo;
import cn.stylefeng.guns.base.db.factory.DataBaseInfoFactory;
import cn.stylefeng.guns.base.db.model.params.DatabaseInfoParam;
import cn.stylefeng.guns.base.db.service.DatabaseInfoService;
import cn.stylefeng.guns.base.db.util.DbUtil;
import cn.stylefeng.guns.base.db.util.SqlRunUtil;
import cn.stylefeng.guns.base.pojo.page.LayuiPageFactory;
import cn.stylefeng.guns.base.pojo.page.LayuiPageInfo;
import cn.stylefeng.guns.base.tenant.entity.TenantInfo;
import cn.stylefeng.guns.base.tenant.model.params.TenantInfoParam;
import cn.stylefeng.guns.base.tenant.model.result.TenantInfoResult;
import cn.stylefeng.guns.base.tenant.service.TenantInfoService;
import cn.stylefeng.guns.sys.core.util.SaltUtil;
import cn.stylefeng.guns.tenant.mapper.TenantInfoMapper;
import cn.stylefeng.roses.core.config.properties.DruidProperties;
import cn.stylefeng.roses.core.data.SqlExe;
import cn.stylefeng.roses.core.util.SpringContextHolder;
import cn.stylefeng.roses.core.util.ToolUtil;
import cn.stylefeng.roses.kernel.model.exception.ServiceException;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.io.Serializable;
import java.util.List;

import static cn.stylefeng.guns.base.tenant.consts.TenantConstants.TENANT_DB_PREFIX;

/**
 * <p>
 * 租户表 服务实现类
 * </p>
 *
 * @author stylefeng
 * @since 2019-06-16
 */
@Service
public class TenantInfoServiceImpl extends ServiceImpl<TenantInfoMapper, TenantInfo> implements TenantInfoService {

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void add(TenantInfoParam param) {

        //创建租户数据库
        DruidProperties druidProperties = SpringContextHolder.getBean(DruidProperties.class);
        String databaseName = TENANT_DB_PREFIX + param.getCode();
        DbUtil.createDatabase(druidProperties, databaseName);

        //创建租户的数据源记录
        DatabaseInfoService databaseInfoService = null;
        try {
            databaseInfoService = SpringContextHolder.getBean(DatabaseInfoService.class);
        } catch (Exception e) {
            throw new ServiceException(500, "请先开启数据源容器模块！");
        }
        DatabaseInfoParam dataBaseInfo = DataBaseInfoFactory.createDataBaseInfo(druidProperties, databaseName);
        databaseInfoService.add(dataBaseInfo);

        //初始化租户的数据库
        SqlRunUtil.runClassPathSql("tenant_init.sql", databaseName);

        //插入租户记录
        TenantInfo entity = getEntity(param);
        entity.setDbName(databaseName);
        this.save(entity);

        //切换数据源到新的租户,初始化新租户的用户名和密码
        String randomSalt = SaltUtil.getRandomSalt();
        String md5 = SaltUtil.md5Encrypt(param.getAdminPassword(), randomSalt);

        SqlExe.update(DataSourceContext.getDataSources().get(databaseName), "update sys_user set password = ?, salt = ?  where account = 'admin'", md5, randomSalt);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public void delete(TenantInfoParam param) {

        //获取租户信息
        TenantInfo tenantInfo = this.getById(param.getTenantId());

        //删除租户信息
        this.removeById(getKey(param));

        //删除对应的数据源
        DatabaseInfoService databaseInfoService = null;
        try {
            databaseInfoService = SpringContextHolder.getBean(DatabaseInfoService.class);
        } catch (Exception e) {
            throw new ServiceException(500, "请先开启数据源容器模块！");
        }
        databaseInfoService.remove(new QueryWrapper<DatabaseInfo>().eq("db_name", tenantInfo.getDbName()));
    }

    @Override
    public void update(TenantInfoParam param) {
        TenantInfo oldEntity = getOldEntity(param);
        TenantInfo newEntity = getEntity(param);
        ToolUtil.copyProperties(newEntity, oldEntity);
        this.updateById(newEntity);
    }

    @Override
    public TenantInfoResult findBySpec(TenantInfoParam param) {
        return null;
    }

    @Override
    public List<TenantInfoResult> findListBySpec(TenantInfoParam param) {
        return null;
    }

    @Override
    public LayuiPageInfo findPageBySpec(TenantInfoParam param) {
        Page pageContext = getPageContext();
        IPage page = this.baseMapper.customPageList(pageContext, param);
        return LayuiPageFactory.createPageInfo(page);
    }

    @Override
    public TenantInfo getByCode(String code) {
        return this.getOne(new QueryWrapper<TenantInfo>().eq("code", code));
    }

    private Serializable getKey(TenantInfoParam param) {
        return param.getTenantId();
    }

    private Page getPageContext() {
        return LayuiPageFactory.defaultPage();
    }

    private TenantInfo getOldEntity(TenantInfoParam param) {
        return this.getById(getKey(param));
    }

    private TenantInfo getEntity(TenantInfoParam param) {
        TenantInfo entity = new TenantInfo();
        ToolUtil.copyProperties(param, entity);
        return entity;
    }

}