dentimoer_setup/

lib.rs

use std::process::Command;
use std::fs;
use std::io;
use std::path::Path;
use regex::Regex;
use std::thread;
use std::time::Duration;

pub fn install_essential_equipments() -> io::Result<()> {
    let choco_install_cmd = "Set-ExecutionPolicy Bypass -Scope Process -Force; [System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor 3072; iex ((New-Object System.Net.WebClient).DownloadString('https://community.chocolatey.org/install.ps1'))";
    
    let opencv_include_paths = "C:\\tools\\opencv\\build\\include";
    let opencv_world_version = get_opencv_num_and_version_test();
    let opencv_link_paths = "C:\\tools\\opencv\\x64\\vc16\\lib";
    let opencv_bin_paths = "C:\\tools\\opencv\\build\\x64\\vc16\\bin";

    // 첫 번째 명령어 실행 및 완료 대기
    match call_admin_powershell("ls") {
        Ok(exit_code) => {
            println!("First ls command completed with exit code: {}", exit_code);
            thread::sleep(Duration::from_secs(2));
            
            // dir 명령어 실행
            match call_admin_powershell("dir") {
                Ok(exit_code) => {
                    println!("Dir command completed with exit code: {}", exit_code);
                    thread::sleep(Duration::from_secs(2));
                    
                    // 두 번째 ls 명령어 실행
                    match call_admin_powershell("ls") {
                        Ok(exit_code) => {
                            println!("Second ls command completed with exit code: {}", exit_code);
                            Ok(())
                        },
                        Err(e) => {
                            eprintln!("Second ls command failed: {}", e);
                            Err(e)
                        }
                    }
                },
                Err(e) => {
                    eprintln!("Dir command failed: {}", e);
                    Err(e)
                }
            }
        },
        Err(e) => {
            eprintln!("First ls command failed: {}", e);
            Err(e)
        }
    }
}

pub fn call_admin_powershell(cmd: &str) -> io::Result<u32> {
    let output = Command::new("powershell")
        .arg("-Command")
        .arg(format!(
            "Start-Process PowerShell -ArgumentList '-Command', '{}' -Verb RunAs -Wait",
            cmd
        ))
        .output()?;

    if output.status.success() {
        println!("Command '{}' executed successfully.", cmd);
        Ok(output.status.code().unwrap_or(0) as u32)
    } else {
        let error_message = String::from_utf8_lossy(&output.stderr);
        eprintln!("Command '{}' failed: {}", cmd, error_message);
        Err(io::Error::new(
            io::ErrorKind::Other,
            format!("Command execution failed: {}", error_message)
        ))
    }
}

pub fn get_opencv_num_and_version_test() -> Option<u32> {
    let opencv_lib_path = "C:\\tools\\opencv\\build\\x64\\lib"; // OpenCV 라이브러리 경로
    let opencv_world_prefix = "opencv_world";

    // 라이브러리 경로가 존재하는지 확인
    if !Path::new(opencv_lib_path).exists() {
        eprintln!("OpenCV 라이브러리 경로가 존재하지 않습니다: {}", opencv_lib_path);
        return None;
    }

    // 디렉토리에서 파일 목록 읽기
    match fs::read_dir(opencv_lib_path) {
        Ok(entries) => {
            // opencv_world 뒤의 숫자를 찾기 위한 정규식
            let re = Regex::new(&format!(r"{}(\d+)\.lib", opencv_world_prefix)).unwrap();

            for entry in entries {
                if let Ok(entry) = entry {
                    let file_name = entry.file_name();
                    let file_name = file_name.to_string_lossy();

                    // 정규식에 매칭되는 파일 찾기
                    if let Some(captures) = re.captures(&file_name) {
                        if let Some(version_str) = captures.get(1) {
                            if let Ok(version) = version_str.as_str().parse::<u32>() {
                                println!("발견된 OpenCV 라이브러리: {}", file_name);
                                println!("추출된 버전 숫자: {}", version);
                                return Some(version); // 버전을 u16 값으로 반환
                            } else {
                                eprintln!("버전 숫자가 u16 형식으로 변환되지 않습니다.");
                                return None;
                            }
                        }
                    }
                }
            }

            println!("opencv_world 라이브러리를 찾을 수 없습니다.");
            None
        }
        Err(e) => {
            eprintln!("디렉토리를 읽을 수 없습니다: {}", e);
            None
        }
    }
}

pub fn set_opencv_env_test(opencv_version: u32) { // for windows 11
    let opencv_include_paths = "C:\\tools\\opencv\\build\\include";
    let opencv_link_string = format!("opencv_world{}", opencv_version);
    let opencv_link_libs = opencv_link_string.as_str();
    let opencv_link_paths = "C:\\tools\\opencv\\build\\x64\\lib";
    let opencv_bin_paths = "C:\\tools\\opencv\\build\\x64\\vc16\\bin";

    // 명령어를 실행하기 위한 PowerShell 스크립트 생성
    let script = format!(
        r#"
        [Environment]::SetEnvironmentVariable('OPENCV_INCLUDE_PATHS', '{0}', [EnvironmentVariableTarget]::Machine)
        [Environment]::SetEnvironmentVariable('OPENCV_LINK_LIBS', '{1}', [EnvironmentVariableTarget]::Machine)
        [Environment]::SetEnvironmentVariable('OPENCV_LINK_PATHS', '{2}', [EnvironmentVariableTarget]::Machine)
        $currentPath = [Environment]::GetEnvironmentVariable('Path', [EnvironmentVariableTarget]::Machine)
        if (-not ($currentPath -like '*{3}*')) {{
            [Environment]::SetEnvironmentVariable('Path', \"$currentPath;{3}\", [EnvironmentVariableTarget]::Machine)
        }}
        "#,
        opencv_include_paths, opencv_link_libs, opencv_link_paths, opencv_bin_paths
    );

    // PowerShell 명령 실행
    let output = Command::new("powershell")
        .args(["-Command", &script])
        .output()
        .expect("Failed to execute PowerShell command");

    // 결과 출력
    if output.status.success() {
        println!("환경 변수가 성공적으로 설정되었습니다.");
    } else {
        eprintln!(
            "오류 발생: {}",
            String::from_utf8_lossy(&output.stderr)
        );
    }
}