adjust service crate naming (#440)

Author: ctaggart

services/autorust/codegen/examples/gen_mgmt.rs

@@ -1,7 +1,6 @@
 // cargo run --example gen_mgmt --release
 // https://github.com/Azure/azure-rest-api-specs/blob/master/specification/compute/resource-manager
 use autorust_codegen::{self, cargo_toml, config_parser::to_mod_name, get_mgmt_readmes, lib_rs, path, Config, PropertyName, SpecReadme};
-use heck::SnakeCase;
 use std::{collections::HashSet, fs, path::PathBuf};
 
 const OUTPUT_FOLDER: &str = "../mgmt";
@@ -324,7 +323,7 @@ fn main() -> Result<()> {
 }
 
 fn gen_crate(spec: &SpecReadme) -> Result<()> {
-    let service_name = &get_service_name(spec.spec());
+    let service_name = &spec.service_name();
     let crate_name = &format!("azure_mgmt_{}", service_name);
     let output_folder = &path::join(OUTPUT_FOLDER, service_name).map_err(|source| Error::PathError { source })?;
 
@@ -407,7 +406,3 @@ fn gen_crate(spec: &SpecReadme) -> Result<()> {
 
     Ok(())
 }
-
-fn get_service_name(spec_folder: &str) -> String {
-    spec_folder.to_snake_case().replace("-", "_")
-}

services/autorust/codegen/examples/gen_svc.rs

@@ -1,7 +1,6 @@
 // cargo run --example gen_svc --release
 // https://github.com/Azure/azure-rest-api-specs/blob/master/specification/batch/data-plane
 use autorust_codegen::{self, cargo_toml, config_parser::to_mod_name, get_svc_readmes, lib_rs, path, Config, PropertyName, SpecReadme};
-use heck::SnakeCase;
 use std::{collections::HashSet, fs, path::PathBuf};
 
 const OUTPUT_FOLDER: &str = "../svc";
@@ -170,7 +169,7 @@ fn gen_crate(spec: &SpecReadme) -> Result<()> {
         return Ok(());
     }
 
-    let service_name = &get_service_name(spec.spec());
+    let service_name = &spec.service_name();
     let crate_name = &format!("azure_svc_{}", service_name);
     let output_folder = &path::join(OUTPUT_FOLDER, service_name).map_err(|source| Error::PathError { source })?;
 
@@ -261,7 +260,3 @@ fn gen_crate(spec: &SpecReadme) -> Result<()> {
 
     Ok(())
 }
-
-fn get_service_name(spec_folder: &str) -> String {
-    spec_folder.to_snake_case().replace("-", "_").replace(".", "_")
-}

services/autorust/codegen/src/lib.rs

@@ -180,6 +180,9 @@ impl SpecReadme {
     pub fn spec(&self) -> &str {
         self.spec.as_str()
     }
+    pub fn service_name(&self) -> String {
+        get_service_name(&self.spec)
+    }
     pub fn readme(&self) -> &Path {
         self.readme.as_path()
     }
@@ -226,3 +229,21 @@ pub fn get_svc_readmes() -> Result<Vec<SpecReadme>> {
     });
     Ok(readmes)
 }
+
+fn get_service_name(spec_name: &str) -> String {
+    spec_name.replace("azure", "").replace("_", "").replace("-", "").to_lowercase()
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_service_name() {
+        assert_eq!("activedirectory", get_service_name("azureactivedirectory"));
+        assert_eq!("cosmosdb", get_service_name("cosmos_db"));
+        assert_eq!("datalakestore", get_service_name("datalake_store"));
+        assert_eq!("kusto", get_service_name("azure-kusto"));
+        assert_eq!("enterpriseknowledgegraph", get_service_name("EnterpriseKnowledgeGraph"));
+    }
+}