support railway connect for timescale image (#466)

* support `railway connect` for timescale image * lowercase
railwayapp · Nov 14, 2023 · 18e6272 · 18e6272
1 parent 975a01d
commit 18e6272
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/src/commands/connect.rs b/src/commands/connect.rs
@@ -118,7 +118,10 @@ impl PluginOrService {
                     .and_then(|source| source.image)
                     .map(|image: String| image.to_lowercase())
                     .and_then(|image: String| {
-                        if image.contains("postgres") {
+                        if image.contains("postgres")
+                            || image.contains("postgis")
+                            || image.contains("timescale")
+                        {
                             Some(PluginType::postgresql)
                         } else if image.contains("redis") {
                             Some(PluginType::redis)

diff --git a/src/controllers/project.rs b/src/controllers/project.rs
@@ -51,13 +51,11 @@ pub fn get_plugin_or_service(
         .services
         .edges
         .iter()
-        .find(|edge| edge.node.name.to_lowercase() == service_or_plugin_name);
+        .find(|edge| edge.node.name.to_lowercase() == service_or_plugin_name.to_lowercase());
 
-    let plugin = project
-        .plugins
-        .edges
-        .iter()
-        .find(|edge| edge.node.friendly_name.to_lowercase() == service_or_plugin_name);
+    let plugin = project.plugins.edges.iter().find(|edge| {
+        edge.node.friendly_name.to_lowercase() == service_or_plugin_name.to_lowercase()
+    });
 
     if let Some(service) = service {
         return Ok(PluginOrService::Service(service.node.clone()));