package com.thinkbridge.demo.auth.thinkbridge;

import com.luamas.thinkbridge.common.security.oauth2.IRegisteredClientService;
import com.luamas.thinkbridge.common.security.oauth2.Oauth2Client;
import com.luamas.thinkbridge.common.security.oauth2.Oauth2ClientConvertor;
import com.thinkbridge.demo.auth.service.IOauth2ClientService;
import lombok.RequiredArgsConstructor;
import org.springframework.security.oauth2.server.authorization.client.RegisteredClient;
import org.springframework.security.oauth2.server.authorization.client.RegisteredClientRepository;

import java.util.Collection;
import java.util.function.Function;

@RequiredArgsConstructor
public class MyRegisteredClientRepository implements RegisteredClientRepository, IRegisteredClientService<String> {
    private final IOauth2ClientService myOauth2ClientService;
    @Override
    public void save(RegisteredClient registeredClient) {
        MyOauth2Client myOauth2Client = this.myOauth2ClientService.lambdaQuery().eq(Oauth2Client::getClientId, registeredClient.getClientId()).one();
        MyOauth2Client newOauth2Client = new MyOauth2Client();
        newOauth2Client = Oauth2ClientConvertor.fromRegisteredClient(registeredClient, newOauth2Client);
        if (myOauth2Client != null) {
            newOauth2Client.setId(myOauth2Client.getId());
        }
        this.myOauth2ClientService.saveOrUpdate(newOauth2Client);
    }

    @Override
    public RegisteredClient findById(String id) {
        Oauth2Client myOauth2Client = this.myOauth2ClientService.getById(id);
        return Oauth2ClientConvertor.toRegisteredClient(myOauth2Client);
    }

    @Override
    public RegisteredClient findByClientId(String clientId) {
        Oauth2Client myOauth2Client = this.myOauth2ClientService.lambdaQuery().eq(Oauth2Client::getClientId, clientId).one();
        return Oauth2ClientConvertor.toRegisteredClient(myOauth2Client);
    }


    @Override
    public boolean removeById(String id) {
        return myOauth2ClientService.removeById(id);
    }

    @Override
    public boolean removeByIds(Collection<String> ids) {
        return myOauth2ClientService.removeByIds(ids);
    }
}

package com.thinkbridge.demo.auth.service;

import com.baomidou.mybatisplus.spring.service.IService;
import com.thinkbridge.demo.auth.thinkbridge.MyOauth2Client;

public interface IOauth2ClientService extends IService<MyOauth2Client> {

}

package com.thinkbridge.demo.auth.service;

import com.baomidou.mybatisplus.spring.service.impl.ServiceImpl;
import com.thinkbridge.demo.auth.thinkbridge.MyOauth2Client;
import com.thinkbridge.demo.auth.mapper.MyOauth2ClientMapper;
import org.springframework.security.crypto.factory.PasswordEncoderFactories;
import org.springframework.security.crypto.password.PasswordEncoder;
import org.springframework.stereotype.Service;

@Service
public class Oauth2ClientServiceImp extends ServiceImpl<MyOauth2ClientMapper, MyOauth2Client> implements IOauth2ClientService {
    private static final PasswordEncoder passwordEncoder = PasswordEncoderFactories.createDelegatingPasswordEncoder();

    @Override
    public boolean save(MyOauth2Client myOauth2Client) {
        if (!myOauth2Client.getClientSecret().startsWith("{")) {
            myOauth2Client.setClientSecret(passwordEncoder.encode(myOauth2Client.getClientSecret()));
        }
        return super.save(myOauth2Client);
    }
}

package com.thinkbridge.demo.auth.thinkbridge;


import com.baomidou.mybatisplus.annotation.*;
import com.luamas.thinkbridge.common.security.oauth2.Oauth2Client;
import lombok.Getter;
import lombok.Setter;
import lombok.ToString;

import java.io.Serial;
import java.io.Serializable;
import java.time.LocalDateTime;


@Getter
@Setter
@ToString
@TableName("sys_oauth2_client")
public class MyOauth2Client implements Oauth2Client<String>, Serializable {

    @Serial
    private static final long serialVersionUID = 1L;

    /
     * ID
     */
    @TableId(value = "id", type = IdType.AUTO)
    private String id;

    /
     * 客户端ID
     */
    private String clientId;

    /
     * 颁发时间
     */
    private LocalDateTime clientIdIssuedAt;

    /
     * 客户端密钥
     */
    private String clientSecret;

    /
     * 密钥过期时间
     */
    private LocalDateTime clientSecretExpiresAt;

    /
     * 客户端名称
     */
    private String clientName;

    /
     * 客户端认证方式，逗号分隔
     */
    private String clientAuthenticationMethods;

    /
     * 授权方式，逗号分隔
     */
    private String authorizationGrantTypes;

    /
     * 重定向url，逗号分隔
     */
    private String redirectUris;

    /
     * 退出后的重定向url，逗号分隔
     */
    private String postLogoutRedirectUris;

    /
     * 授权范围，逗号分隔
     */
    private String scopes;

    /
     * 客户端配置，json字符串
     */
    private String clientSettings;

    /
     * token配置，json字符串
     */
    private String tokenSettings;

    /
     * 创建人
     */
    @TableField(value="create_by",fill = FieldFill.INSERT)
    private String createBy;

    /
     * 创建时间
     */
    @TableField(value="create_time",fill = FieldFill.INSERT)
    private LocalDateTime createTime;

    /
     * 更新人
     */
    @TableField(value="update_by",fill = FieldFill.INSERT_UPDATE)
    private String updateBy;

    /
     * 更新时间
     */
    @TableField(value="update_time",fill = FieldFill.INSERT_UPDATE)
    private LocalDateTime updateTime;

    /
     * 是否删除 0 否 1 是
     */
    @TableLogic(value = "false", delval = "true")
    @TableField(value="is_delete")
    private Boolean deleted;
}

package com.thinkbridge.demo.auth.thinkbridge;


import com.baomidou.mybatisplus.annotation.*;
import lombok.Builder;
import lombok.Getter;
import lombok.Setter;
import lombok.ToString;

import java.io.Serial;
import java.io.Serializable;
import java.math.BigDecimal;
import java.util.Date;


@Getter
@Setter
@ToString
@Builder
@TableName(value = "sys_user_info")
public class MyOauth2User implements Serializable {

    @Serial
    private static final long serialVersionUID = 1L;


    @TableId(value = "id", type = IdType.AUTO)
    private Long id;

    @TableField(value = "status")
    private Integer status; // 账号状态：1启用 2禁用 3预留

    @TableField(value = "user_account")
    private String userAccount; // 用户名，唯一

    @TableField(value = "user_password")
    private String userPassword; // 用户密码

    private String wxUnionId;
    private String wxOpenId;
    //租户id
    private Long tenantId;

    @TableLogic
    private Integer deleted;

    /
     * 创建人
     */
    @TableField(value="create_by",fill = FieldFill.INSERT)
    private String createBy;

    /
     * 创建时间
     */
    @TableField(value="create_time",fill = FieldFill.INSERT)
    private Date createTime;

    /
     * 更新人
     */
    @TableField(value="update_by",fill = FieldFill.INSERT_UPDATE)
    private String updateBy;

    /
     * 更新时间
     */
    @TableField(value="update_time",fill = FieldFill.INSERT_UPDATE)
    private Date updateTime;
}