#!/bin/bash
#
# Copyright(c) 2025 Huawei Technologies
# SPDX-License-Identifier: BSD-3-Clause
#

. $(dirname $3)/conf_framework.sh

check() {
	cur_name=$(basename $2)
	config_file_path=$1
	if compile_module $cur_name "preempt_model_voluntary();" "linux/preempt.h" &&
	   compile_module $cur_name "preempt_model_none();" "linux/preempt.h"
	then
		echo $cur_name "1" >> $config_file_path
	else
		echo $cur_name "2" >> $config_file_path
	fi
}

apply() {
    case "$1" in
    "1")
		add_function "
	static inline int cas_preempt_model_voluntary(void)
	{
		return preempt_model_voluntary();
	}"
		add_function "
	static inline int cas_preempt_model_none(void)
	{
		return preempt_model_none();
	}" ;;

    "2")
		add_function "
	static inline int cas_preempt_model_voluntary(void)
	{
		return IS_ENABLED(CONFIG_PREEMPT_VOLUNTARY);
	}"
		add_function "
	static inline int cas_preempt_model_none(void)
	{
		return IS_ENABLED(CONFIG_PREEMPT_NONE);
	}" ;;

    *)
        exit 1
    esac
}

conf_run $@