Unimarc/unimarc/unimarc/SearchModel/KwangjuCityEduLibrarySearcher.cs

using System;
using System.Threading.Tasks;
using OpenQA.Selenium;
using OpenQA.Selenium.Chrome;
using OpenQA.Selenium.Support.UI;
using System.Text.RegularExpressions;
using WebDriverManager;
using WebDriverManager.DriverConfigs.Impl;
using System.IO;
using System.Runtime.InteropServices;
using UniMarc.SearchModel;
using OpenQA.Selenium.Chromium;

namespace BokBonCheck
{
    public  class KwangjuCityEduLibrarySearcher : ILibrarySearcher
    {
        public int No { get; set; }

        public string AreaCode { get; set; } = string.Empty;
        public string SiteName { get; protected set; } = "광주시교육청통합도서관";
        public string SiteUrl => "https://lib.gen.go.kr/main/site/search/bookSearch.do#simple";

        public bool HttpApiMode { get; set; } = false;

        private ChromiumDriver _driver;

        public KwangjuCityEduLibrarySearcher(int no, string areaCode, string areaName)
        {
            this.No = no;
            this.AreaCode = areaCode;
            this.SiteName = $"광주시교육청통합도서관({areaName})";
        }

        public async Task StartDriver(bool showBrowser = false)
        {
            if (_driver == null)
            {
                try
                {
                    if (SeleniumHelper.IsReady == false) await SeleniumHelper.Download();
                    _driver = await SeleniumHelper.CreateDriver(ShowBrowser:showBrowser);
                    Console.WriteLine("KwangjuCityLibrarySearcher Driver 초기화 완료");
                }
                catch (Exception ex)
                {
                    Console.WriteLine($"KwangjuCityLibrarySearcher Driver 초기화 실패: {ex.Message}");
                    throw new InvalidOperationException($"KwangjuCityLibrarySearcher Driver 초기화에 실패했습니다: {ex.Message}", ex);
                }
            }
        }

        public void StopDriver()
        {
            if (_driver != null)
            {
                _driver.Quit();
                _driver.Dispose();
                _driver = null;
            }
        }

        virtual protected void SelectLibrary(WebDriverWait wait)
        {
            try
            {
                // 콤보박스(select) 요소 찾기
                var selectElement = wait.Until(d => d.FindElement(By.CssSelector("#manage_code")));
                var select = new OpenQA.Selenium.Support.UI.SelectElement(selectElement);

                // value가 "MA"인 옵션 선택
                select.SelectByValue(AreaCode);
            }
            catch
            {
                // 예외 처리 (필요시 로깅 등)
            }
        }

        public async Task<BookSearchResult> SearchAsync(string searchTerm)
        {
            var result = new BookSearchResult
            {
                SiteName = SiteName,
                SearchTerm = searchTerm,
                SearchTime = DateTime.Now
            };

            try
            {
                if (_driver == null)
                    throw new InvalidOperationException("드라이버가 시작되지 않았습니다. StartDriver()를 먼저 호출하세요.");

                _driver.Navigate().GoToUrl(SiteUrl);

                // 페이지 로딩 대기
                var wait = new WebDriverWait(_driver, TimeSpan.FromSeconds(15));

                // 모든 감지 방법이 끝나면 크롬창 최소화
                // whale 브라우저가 최소화되어 우선해제
                //IntPtr chromeWindow = FindWindow("Chrome_WidgetWin_1", null);
                //if (chromeWindow != IntPtr.Zero)
                //{
                //    ShowWindow(chromeWindow, SW_MINIMIZE);
                //}

                //대상도서관 선택
                SelectLibrary(wait);

                // 검색창 찾기
                IWebElement searchBox = null;
                try
                {
                    // 여러 가능한 선택자 시도
                    var selectors = new[]
                    {
                        "input[id='search_txt']",
                        "input[type='text']",
                    };

                    foreach (var selector in selectors)
                    {
                        try
                        {
                            searchBox = wait.Until(d =>
                            {
                                var el = d.FindElement(By.CssSelector(selector));
                                return (el != null && el.Displayed && el.Enabled) ? el : null;
                            });
                            break;
                        }
                        catch
                        {
                            continue;
                        }
                    }

                    if (searchBox == null)
                    {
                        throw new Exception("검색창을 찾을 수 없습니다.");
                    }
                }
                catch (Exception ex)
                {
                    throw new Exception($"검색창 찾기 실패: {ex.Message}");
                }

                // 혹시 readonly라면 속성 제거
                ((IJavaScriptExecutor)_driver).ExecuteScript("arguments[0].removeAttribute('readonly')", searchBox);

                // 입력 전 clear
                searchBox.Clear();
                searchBox.SendKeys(searchTerm);

                // 검색 버튼 클릭
                IWebElement searchButton = null;
                try
                {
                    var buttonSelectors = new[]
                    {
                        "input[type='submit']",
                    };
                    foreach (var selector in buttonSelectors)
                    {
                        try
                        {
                            searchButton = _driver.FindElement(By.CssSelector(selector));
                            break;
                        }
                        catch
                        {
                            continue;
                        }
                    }

                    if (searchButton == null)
                    {
                        // Enter 키로 검색 시도
                        searchBox.SendKeys(Keys.Enter);
                    }
                    else
                    {
                        searchButton.Click();
                    }
                }
                catch (Exception ex)
                {
                    // Enter 키로 검색 시도
                    searchBox.SendKeys(Keys.Enter);
                }

                // 페이지 변경을 감지하는 메서드
                await WaitForPageChange(new WebDriverWait(_driver, TimeSpan.FromSeconds(15)));

                // 검색 결과 수 추출
                var resultCount = ExtractBookCount(_driver);

                result.BookCount = resultCount;
                result.IsSuccess = true;
            }
            catch (Exception ex)
            {
                result.IsSuccess = false;
                result.ErrorMessage = ex.Message;
                result.BookCount = 0;
            }

            return result;
        }

        private int ExtractBookCount(IWebDriver driver)
        {
            try
            {
                // div.search-result 내부의 span에서 '전체 N' 텍스트 추출
                var resultDiv = driver.FindElement(By.CssSelector("div.ndls_result"));
                var span = resultDiv.FindElement(By.XPath(".//span[contains(text(),'전체')]"));
                string text = span.Text; // 예: "전체 5 "
                var match = System.Text.RegularExpressions.Regex.Match(text, @"전체\s*(\d+)");
                if (match.Success)
                {
                    return int.Parse(match.Groups[1].Value);
                }
                return 0;
            }
            catch
            {
                return 0;
            }
        }

        // 페이지 변경을 감지하는 메서드
        public async Task WaitForPageChange(WebDriverWait wait)
        {
            try
            {
                // 방법 4: 페이지 로딩 상태 확인
                wait.Until(d =>
                {
                    var readyState = ((IJavaScriptExecutor)d).ExecuteScript("return document.readyState");
                    return readyState.Equals("complete");
                });

                // 방법 5: 특정 텍스트가 페이지에 나타날 때까지 대기
                wait.Until(d =>
                {
                    var elm = d.FindElement(By.TagName("body"));
                    if (elm == null) return false;
                    var pageText = elm.Text;
                    return pageText.Contains("전체") || pageText.Contains("건") || pageText.Contains("검색결과");
                });
            }
            catch (Exception ex)
            {
                // 모든 감지 방법이 실패하면 최소한의 대기 시간 적용
                await Task.Delay(2000);
                throw new Exception($"페이지 변경 감지 실패: {ex.Message}");
            }
        }
    }
}