package com.example;

import org.drools.core.event.DefaultAgendaEventListener;
import org.kie.api.KieServices;
import org.kie.api.runtime.KieContainer;
import org.kie.api.runtime.KieSession;

public class SecurityEncryptionExample {

    public static void main(String[] args) {
        // Drools KieServices 인스턴스 생성
        KieServices kieServices = KieServices.Factory.get();

        // KieContainer를 사용하여 kmodule.xml 파일에 정의된 규칙을 로드
        KieContainer kieContainer = kieServices.getKieClasspathContainer();

        // KieSession 생성
        KieSession kieSession = kieContainer.newKieSession();

        // 보안 암호화 실행을 위한 데이터 객체 생성
        SecurityData securityData = new SecurityData("Some important data to encrypt");

        // 데이터 객체를 KieSession에 주입
        kieSession.insert(securityData);

        // KieSession에 이벤트 리스너 추가
        kieSession.addEventListener(new DefaultAgendaEventListener() {
            @Override
            public void afterMatchFired(org.kie.api.event.rule.AfterMatchFiredEvent event) {
                // 보안 암호화 로직을 호출하는 비즈니스 룰 실행
                encryptData(securityData);
            }
        });

        // 규칙 실행
        kieSession.fireAllRules();

        // 보안 암호화가 적용된 데이터 출력
        System.out.println("Encrypted Data: " + securityData.getData());

        // KieSession 종료
        kieSession.dispose();
    }

    private static void encryptData(SecurityData securityData) {
        // 보안 암호화 로직을 구현
        // securityData 객체의 데이터를 암호화하고 값을 업데이트
        String encryptedData = performEncryption(securityData.getData());
        securityData.setData(encryptedData);
    }

    private static String performEncryption(String data) {
        // 실제 암호화 로직을 구현하는 메서드
        // 예를 들어, AES 암호화 알고리즘을 사용하여 데이터를 암호화
        // 여기서는 암호화하지 않고 단순히 입력 데이터를 반환하는 예제를 사용
        return data;
    }

    public static class SecurityData {
        private String data;

        public SecurityData(String data) {
            this.data = data;
        }

        public String getData() {
            return data;
        }

        public void setData(String data) {
            this.data = data;
        }
    }
}