package com.xmzs.midjourney.util;

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.text.CharSequenceUtil;
import com.xmzs.midjourney.exception.BannedPromptException;
import lombok.experimental.UtilityClass;

import java.io.File;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.Locale;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

@UtilityClass
public class BannedPromptUtils {
	private static final String BANNED_WORDS_FILE_PATH = "/home/spring/config/banned-words.txt";
	private final List<String> BANNED_WORDS;

	static {
		List<String> lines;
		File file = new File(BANNED_WORDS_FILE_PATH);
		if (file.exists()) {
			lines = FileUtil.readLines(file, StandardCharsets.UTF_8);
		} else {
			var resource = BannedPromptUtils.class.getResource("/banned-words.txt");
			lines = FileUtil.readLines(resource, StandardCharsets.UTF_8);
		}
		BANNED_WORDS = lines.stream().filter(CharSequenceUtil::isNotBlank).toList();
	}

	public static void checkBanned(String promptEn) throws BannedPromptException {
		String finalPromptEn = promptEn.toLowerCase(Locale.ENGLISH);
		for (String word : BANNED_WORDS) {
			Matcher matcher = Pattern.compile("\\b" + word + "\\b").matcher(finalPromptEn);
			if (matcher.find()) {
				int index = CharSequenceUtil.indexOfIgnoreCase(promptEn, word);
				throw new BannedPromptException(promptEn.substring(index, index + word.length()));
			}
		}
	}

}