增加后台管理,支持docker部署

ageer

2024-05-17 7fe89a931b291196b0b571c668bd3758309019d7

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package com.xmzs.midjourney.util;
 
import cn.hutool.core.text.CharSequenceUtil;
import com.xmzs.midjourney.enums.TaskAction;
import eu.maxschuster.dataurl.DataUrl;
import eu.maxschuster.dataurl.DataUrlSerializer;
import eu.maxschuster.dataurl.IDataUrlSerializer;
import lombok.experimental.UtilityClass;
 
import java.net.MalformedURLException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
 
@UtilityClass
public class ConvertUtils {
    /**
     * content正则匹配prompt和进度.
     */
    public static final String CONTENT_REGEX = ".*?\\*\\*(.*?)\\*\\*.+<@\\d+> \\((.*?)\\)";
 
    public static ContentParseData parseContent(String content) {
        return parseContent(content, CONTENT_REGEX);
    }
 
    public static ContentParseData parseContent(String content, String regex) {
        if (CharSequenceUtil.isBlank(content)) {
            return null;
        }
        Matcher matcher = Pattern.compile(regex).matcher(content);
        if (!matcher.find()) {
            return null;
        }
        ContentParseData parseData = new ContentParseData();
        parseData.setPrompt(matcher.group(1));
        parseData.setStatus(matcher.group(2));
        return parseData;
    }
 
    public static List<DataUrl> convertBase64Array(List<String> base64Array) throws MalformedURLException {
        if (base64Array == null || base64Array.isEmpty()) {
            return Collections.emptyList();
        }
        IDataUrlSerializer serializer = new DataUrlSerializer();
        List<DataUrl> dataUrlList = new ArrayList<>();
        for (String base64 : base64Array) {
            DataUrl dataUrl = serializer.unserialize(base64);
            dataUrlList.add(dataUrl);
        }
        return dataUrlList;
    }
 
    public static TaskChangeParams convertChangeParams(String content) {
        List<String> split = CharSequenceUtil.split(content, " ");
        if (split.size() != 2) {
            return null;
        }
        String action = split.get(1).toLowerCase();
        TaskChangeParams changeParams = new TaskChangeParams();
        changeParams.setId(split.get(0));
        if (action.charAt(0) == 'u') {
            changeParams.setAction(TaskAction.UPSCALE);
        } else if (action.charAt(0) == 'v') {
            changeParams.setAction(TaskAction.VARIATION);
        } else if (action.equals("r")) {
            changeParams.setAction(TaskAction.REROLL);
            return changeParams;
        } else {
            return null;
        }
        try {
            int index = Integer.parseInt(action.substring(1, 2));
            if (index < 1 || index > 4) {
                return null;
            }
            changeParams.setIndex(index);
        } catch (Exception e) {
            return null;
        }
        return changeParams;
    }
 
}