PART 04 — テーブル名・操作種別の判別
JSqlParser / 正規表現でCRUD分類

import re
from dataclasses import dataclass

@dataclass
class CrudEntry:
    table_name: str
    operations: set  # {'C', 'R', 'U', 'D'} の部分集合

# CRUD 操作を判別するパターン
_CRUD_PATTERNS = [
    ('C', re.compile(r'^\s*INSERT\s+INTO\s+([A-Z][A-Z0-9_-]*)', re.IGNORECASE)),
    ('R', re.compile(r'^\s*SELECT\b.*?\bFROM\s+([A-Z][A-Z0-9_-]*)', re.IGNORECASE | re.DOTALL)),
    ('R', re.compile(r'^\s*DECLARE\s+\w+\s+CURSOR\b.*?\bFROM\s+([A-Z][A-Z0-9_-]*)', re.IGNORECASE | re.DOTALL)),
    ('U', re.compile(r'^\s*UPDATE\s+([A-Z][A-Z0-9_-]*)', re.IGNORECASE)),
    ('D', re.compile(r'^\s*DELETE\s+FROM\s+([A-Z][A-Z0-9_-]*)', re.IGNORECASE)),
]

def classify_crud_regex(sql: str) -> list[CrudEntry]:
    """正規表現でSQL文のCRUD種別とテーブル名を判別（フォールバック用）"""
    results = {}
    for op, pattern in _CRUD_PATTERNS:
        m = pattern.match(sql)
        if m:
            table = m.group(1).upper()
            if table not in results:
                results[table] = CrudEntry(table_name=table, operations=set())
            results[table].operations.add(op)
    return list(results.values())

import net.sf.jsqlparser.parser.CCJSqlParserUtil;
import net.sf.jsqlparser.statement.*;
import net.sf.jsqlparser.statement.select.*;
import net.sf.jsqlparser.statement.insert.Insert;
import net.sf.jsqlparser.statement.update.Update;
import net.sf.jsqlparser.statement.delete.Delete;
import net.sf.jsqlparser.util.TablesNamesFinder;
import java.util.*;

/**
 * 標準入力から SQL を読み込み、テーブル名と操作種別を
 * "TABLE_NAME:OPERATION" 形式で標準出力に書き出すラッパー。
 * Python から subprocess で呼び出して使う。
 */
public class CrudAnalyzer {
    public static void main(String[] args) throws Exception {
        Scanner sc = new Scanner(System.in);
        while (sc.hasNextLine()) {
            String sql = sc.nextLine().trim();
            if (sql.isEmpty()) continue;
            try {
                Statement stmt = CCJSqlParserUtil.parse(sql);
                TablesNamesFinder finder = new TablesNamesFinder();
                List<String> tables = finder.getTableList(stmt);
                String op = detectOperation(stmt);
                for (String t : tables) {
                    System.out.println(t.toUpperCase() + ":" + op);
                }
            } catch (Exception e) {
                System.err.println("PARSE_ERROR: " + sql.substring(0, Math.min(60, sql.length())));
            }
        }
    }

    private static String detectOperation(Statement stmt) {
        if (stmt instanceof Select)  return "R";
        if (stmt instanceof Insert)  return "C";
        if (stmt instanceof Update)  return "U";
        if (stmt instanceof Delete)  return "D";
        return "?";
    }
}

import subprocess

def classify_crud_jsqlparser(sql_list: list[str], jar_path: str) -> dict[str, list[CrudEntry]]:
    """
    JSqlParser ラッパーJARを呼び出してSQL群を一括解析する。
    戻り値: {sql文字列: [CrudEntry, ...]} の辞書
    """
    proc = subprocess.run(
        ['java', '-cp', jar_path, 'CrudAnalyzer'],
        input='\n'.join(sql_list),
        capture_output=True,
        text=True,
        encoding='utf-8'
    )

    results = {sql: {} for sql in sql_list}
    lines = proc.stdout.strip().splitlines()
    idx = 0
    for sql in sql_list:
        entries = {}
        # 各SQL の出力行数はテーブル数に依存するため、
        # PARSE_ERROR が出るまで or 次のSQLまで読む（簡易実装）
        while idx < len(lines) and not lines[idx].startswith('PARSE_ERROR'):
            parts = lines[idx].split(':')
            if len(parts) == 2:
                table, op = parts[0].strip(), parts[1].strip()
                if table not in entries:
                    entries[table] = CrudEntry(table_name=table, operations=set())
                entries[table].operations.add(op)
            idx += 1
        results[sql] = list(entries.values())

    return results

def classify_crud(sql: str, jar_path: str | None = None) -> list[CrudEntry]:
    """
    JSqlParser（利用可能な場合）で解析し、失敗時は正規表現でフォールバック。
    """
    if jar_path:
        try:
            result = classify_crud_jsqlparser([sql], jar_path)
            entries = result.get(sql, [])
            if entries:
                return entries
        except Exception:
            pass
    # フォールバック
    return classify_crud_regex(sql)