Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 5e64029b authored by Zhi Dou's avatar Zhi Dou
Browse files

aconfig: add ForceReadOnly mode to aconfig rust codegen 

This commit adds ForceReadOnly mode to rust codegen.

Bug: 316357759
Test: atest aconfig.test aconfig.prod_mode.test.rust
aconfig.test_mode.test.rust aconfig.exported_mode.test.rust
aconfig.force_read_only_mode.test.rust

Change-Id: I4cf98a6e9fc908fc372d4e68098a516808718c57
parent 1a8281cc

tools/aconfig/Android.bp

+18 −0
Original line number Diff line number Diff line
@@ -276,3 +276,21 @@ rust_test { 
    ],

    test_suites: ["general-tests"],

}



rust_aconfig_library {

    name: "libaconfig_test_rust_library_with_force_read_only_mode",

    crate_name: "aconfig_test_rust_library",

    aconfig_declarations: "aconfig.test.flags",

    mode: "force-read-only",

}



rust_test {

    name: "aconfig.force_read_only_mode.test.rust",

    srcs: [

        "tests/aconfig_force_read_only_mode_test.rs"

    ],

    rustlibs: [

        "libaconfig_test_rust_library_with_force_read_only_mode",

    ],

    test_suites: ["general-tests"],

}

tools/aconfig/src/codegen/rust.rs

+87 −4
Original line number Diff line number Diff line
@@ -44,10 +44,10 @@ where 
    template.add_template(

        "rust_code_gen",

        match codegen_mode {

            CodegenMode::Production => include_str!("../../templates/rust_prod.template"),

            CodegenMode::Test => include_str!("../../templates/rust_test.template"),

            CodegenMode::Exported => include_str!("../../templates/rust_exported.template"),

            CodegenMode::ForceReadOnly => todo!(),

            CodegenMode::Exported | CodegenMode::ForceReadOnly | CodegenMode::Production => {

                include_str!("../../templates/rust.template")

            }

        },

    )?;

    let contents = template.render("rust_code_gen", &context)?;
@@ -553,6 +553,84 @@ pub fn enabled_fixed_ro_exported() -> bool { 
pub fn enabled_ro_exported() -> bool {

    PROVIDER.enabled_ro_exported()

}

"#;



    const FORCE_READ_ONLY_EXPECTED: &str = r#"

//! codegenerated rust flag lib



/// flag provider

pub struct FlagProvider;



impl FlagProvider {

    /// query flag disabled_ro

    pub fn disabled_ro(&self) -> bool {

        false

    }



    /// query flag disabled_rw

    pub fn disabled_rw(&self) -> bool {

        false

    }



    /// query flag disabled_rw_in_other_namespace

    pub fn disabled_rw_in_other_namespace(&self) -> bool {

        false

    }



    /// query flag enabled_fixed_ro

    pub fn enabled_fixed_ro(&self) -> bool {

        true

    }



    /// query flag enabled_ro

    pub fn enabled_ro(&self) -> bool {

        true

    }



    /// query flag enabled_rw

    pub fn enabled_rw(&self) -> bool {

        true

    }

}



/// flag provider

pub static PROVIDER: FlagProvider = FlagProvider;



/// query flag disabled_ro

#[inline(always)]

pub fn disabled_ro() -> bool {

    false

}



/// query flag disabled_rw

#[inline(always)]

pub fn disabled_rw() -> bool {

    false

}



/// query flag disabled_rw_in_other_namespace

#[inline(always)]

pub fn disabled_rw_in_other_namespace() -> bool {

    false

}



/// query flag enabled_fixed_ro

#[inline(always)]

pub fn enabled_fixed_ro() -> bool {

    true

}



/// query flag enabled_ro

#[inline(always)]

pub fn enabled_ro() -> bool {

    true

}



/// query flag enabled_rw

#[inline(always)]

pub fn enabled_rw() -> bool {

    true

}

"#;



    fn test_generate_rust_code(mode: CodegenMode) {
@@ -570,7 +648,7 @@ pub fn enabled_ro_exported() -> bool { 
                    CodegenMode::Production => PROD_EXPECTED,

                    CodegenMode::Test => TEST_EXPECTED,

                    CodegenMode::Exported => EXPORTED_EXPECTED,

                    codegen::CodegenMode::ForceReadOnly => todo!(),

                    CodegenMode::ForceReadOnly => FORCE_READ_ONLY_EXPECTED,

                },

                &String::from_utf8(generated.contents).unwrap()

            )
@@ -591,4 +669,9 @@ pub fn enabled_ro_exported() -> bool { 
    fn test_generate_rust_code_for_exported() {

        test_generate_rust_code(CodegenMode::Exported);

    }



    #[test]

    fn test_generate_rust_code_for_force_read_only() {

        test_generate_rust_code(CodegenMode::ForceReadOnly);

    }

}

tools/aconfig/templates/rust_exported.template

deleted100644 → 0
+0 −35
Original line number Diff line number Diff line 
//! codegenerated rust flag lib



/// flag provider

pub struct FlagProvider;



lazy_static::lazy_static! \{

{{ for flag in template_flags }}

    /// flag value cache for {flag.name}

    static ref CACHED_{flag.name}: bool = flags_rust::GetServerConfigurableFlag(

        "aconfig_flags.{flag.device_config_namespace}",

        "{flag.device_config_flag}",

        "false") == "true";

{{ endfor }}

}



impl FlagProvider \{

{{ for flag in template_flags }}

    /// query flag {flag.name}

    pub fn {flag.name}(&self) -> bool \{

        *CACHED_{flag.name}

    }

{{ endfor }}



}



/// flag provider

pub static PROVIDER: FlagProvider = FlagProvider;



{{ for flag in template_flags }}

/// query flag {flag.name}

#[inline(always)]

pub fn {flag.name}() -> bool \{

    PROVIDER.{flag.name}()

}

{{ endfor }}

tools/aconfig/tests/aconfig_force_read_only_mode_test.rs

0 → 100644
+10 −0
Original line number Diff line number Diff line 
#[cfg(not(feature = "cargo"))]

#[test]

fn test_flags() {

    assert!(!aconfig_test_rust_library::disabled_ro());

    assert!(!aconfig_test_rust_library::disabled_rw());

    assert!(!aconfig_test_rust_library::disabled_rw_in_other_namespace());

    assert!(aconfig_test_rust_library::enabled_fixed_ro());

    assert!(aconfig_test_rust_library::enabled_ro());

    assert!(aconfig_test_rust_library::enabled_rw());

}