cpu-cli-controller 0.2.0

use itertools::Itertools;

use crate::utils::{
    activate_cores, deactivate_cores, get_all_core_nums, get_nums_from_ranges, show_cores,
};
use crate::{CoresArgs, OptionalCoresArgs};

pub fn activate_cmd(args: &OptionalCoresArgs) {
    let all_core_nums = get_all_core_nums();
    let mut core_nums = match &args.cores {
        Some(ranges) => get_nums_from_ranges(ranges.to_string()),
        None => all_core_nums.clone(),
    };
    if !core_nums.iter().all(|num| all_core_nums.contains(num)) {
        panic!(
            "Error: the cores selected must be among {:?}",
            all_core_nums
        );
    }
    if args.no_duplicate {
        let mut core_nums: Vec<_> = core_nums.iter().unique().collect();
    }
    if args.sort {
        core_nums.sort()
    }
    activate_cores(core_nums);
}

pub fn deactivate_cmd(args: &CoresArgs) {
    let all_core_nums = get_all_core_nums();
    let mut core_nums = get_nums_from_ranges(args.cores.to_string());
    if !core_nums.iter().all(|num| all_core_nums.contains(num)) {
        panic!(
            "Error: the cores selected must be among {:?}",
            all_core_nums
        );
    }
    if args.no_duplicate {
        let mut core_nums: Vec<_> = core_nums.iter().unique().collect();
    }
    if args.sort {
        core_nums.sort()
    }
    deactivate_cores(core_nums);
}

pub fn show_cmd(args: &OptionalCoresArgs) {
    let all_core_nums = get_all_core_nums();
    let mut core_nums = match &args.cores {
        Some(ranges) => get_nums_from_ranges(ranges.to_string()),
        None => all_core_nums.clone(),
    };
    if !core_nums.iter().all(|num| all_core_nums.contains(num)) {
        panic!(
            "Error: the cores selected must be among {:?}",
            all_core_nums
        );
    }
    if args.no_duplicate {
        let mut core_nums: Vec<_> = core_nums.iter().unique().collect();
    }
    if args.sort {
        core_nums.sort()
    }
    show_cores(core_nums);
}