src/stories/bin/discovermgr/src/local_action_provider.rs

// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#![allow(dead_code)]

use crate::indexing::load_module_facet;
use crate::models::{Action, FuchsiaFulfillment, Parameter};
use failure::Error;
use fidl_fuchsia_sys_index::{ComponentIndexMarker, ComponentIndexProxy};
use fuchsia_component::client::{launch, launcher, App};
use fuchsia_syslog::macros::*;

const COMPONENT_INDEX_URL: &str =
    "fuchsia-pkg://fuchsia.com/component_index#meta/component_index.cmx";

/// Starts and connects to the component index service.
// TODO: combine with version in package_suggestions_provider
fn get_index_service() -> Result<(App, ComponentIndexProxy), Error> {
    let app = launch(&launcher()?, COMPONENT_INDEX_URL.to_string(), None)?;
    let service = app.connect_to_service::<ComponentIndexMarker>()?;
    Ok((app, service))
}

/// Gets a list of all components on the system
// TODO: combine with version in package_suggestions_provider
async fn get_components() -> Result<Vec<String>, Error> {
    let (_app, index_service) = get_index_service().map_err(|e| {
        fx_log_err!("Failed to connect to index service");
        e
    })?;
    // Empty string returns all components.
    let index_response = index_service.fuzzy_search("").await.map_err(|e| {
        fx_log_err!("Fuzzy search error from component index: {:?}", e);
        e
    })?;
    index_response.or_else(|e| {
        fx_log_err!("Fuzzy search error from component index: {:?}", e);
        Ok(vec![])
    })
}

// Gets a vector of all actions on the system
pub async fn get_local_actions() -> Result<Vec<Action>, Error> {
    let urls = get_components().await?;
    let mut result = vec![];
    for url in urls {
        let module_facet = load_module_facet(&url).await?;
        for intent_filter in module_facet.intent_filters.into_iter() {
            // Convert from Indexing to Models
            result.push(Action {
                name: intent_filter.action,
                parameters: intent_filter
                    .parameters
                    .iter()
                    .map(|(k, v)| Parameter { name: k.to_string(), parameter_type: v.to_string() })
                    .collect(),
                action_display: intent_filter.action_display,
                web_fulfillment: None,
                fuchsia_fulfillment: Some(FuchsiaFulfillment { component_url: url.to_string() }),
            })
        }
    }
    Ok(result)
}

#[cfg(test)]
mod test {
    use {super::*, fuchsia_async as fasync};

    const TEST_COMPONENT_URL: &str =
        "fuchsia-pkg://fuchsia.com/discovermgr_tests#meta/discovermgr_bin_test.cmx";

    // Verify the actions in this component test manifest
    #[fasync::run_singlethreaded(test)]
    async fn test_component_index() -> Result<(), Error> {
        // TODO fix tests -- failing only in CQ
        let result = get_local_actions().await.unwrap_or_else(|e| {
            fx_log_err!("Error! {:?}", e);
            vec![]
        });
        #[allow(unused)]
        let actions: Vec<&Action> = result
            .iter()
            .filter(|a| match &a.fuchsia_fulfillment {
                Some(v) => v.component_url == TEST_COMPONENT_URL,
                None => false,
            })
            .collect();
        // assert_eq!(actions.len(), 3, "Expecting to find 3 actions in discovermgr test cmx file");
        Ok(())
    }
}