From 5c9c3fbb5314da872f24a3b8ef005ae73edf35b9 Mon Sep 17 00:00:00 2001
From: Tsiry Sandratraina <tsiry.sndr@fluentci.io>
Date: Fri, 19 Jul 2024 13:18:08 +0000
Subject: [PATCH] [dragonflydb] run dragonfly without docker on linux

---
 dragonflydb/src/helpers.rs | 43 ++++++++++++++++++++++++++++++++------
 1 file changed, 37 insertions(+), 6 deletions(-)

diff --git a/dragonflydb/src/helpers.rs b/dragonflydb/src/helpers.rs
index b71fa8a..dab18ea 100644
--- a/dragonflydb/src/helpers.rs
+++ b/dragonflydb/src/helpers.rs
@@ -2,6 +2,38 @@ use anyhow::Error;
 use fluentci_pdk::dag;
 
 pub fn setup() -> Result<String, Error> {
+    let os = dag().get_os()?;
+    let arch = dag().get_arch()?;
+
+    let path = dag().get_env("PATH")?;
+    let home = dag().get_env("HOME")?;
+    dag().set_envs(vec![(
+        "PATH".into(),
+        format!("{}/.local/bin:{}", home, path),
+    )])?;
+
+    if os == "linux" {
+        let version = dag().get_env("DRAGONFLY_VERSION")?;
+        if version.is_empty() {
+            dag().set_envs(vec![("DRAGONFLY_VERSION".into(), "v1.20.1".into())])?;
+        }
+        dag().set_envs(vec![("ARCH".into(), arch.into())])?;
+
+        dag()
+        .pkgx()?
+        .with_exec(vec!["mkdir", "-p", "$HOME/.local/bin"])?
+        .with_exec(vec!["type dragonfly || pkgx wget https://github.com/dragonflydb/dragonfly/releases/download/$DRAGONFLY_VERSION/dragonfly-$ARCH.tar.gz"])?
+        .with_exec(vec!["type dragonfly > /dev/null 2> /dev/null || pkgx tar -xvzf dragonfly-$ARCH.tar.gz"])?
+        .with_exec(vec!["type dragonfly > /dev/null 2> /dev/null || mv dragonfly-$ARCH $HOME/.local/bin/dragonfly"])?
+        .with_exec(vec!["[ -f dragonfly-$ARCH.tar.gz ] && rm -rf dragonfly-$ARCH.tar.gz LICENSE.md || true"])?
+        .stdout()?;
+    }
+
+    let command = match os.as_str() {
+        "linux" => "dragonfly --port $DRAGONFLY_PORT $DRAGONFLY_ARGS",
+        _ => "pkgx docker run -p $DRAGONFLY_PORT:6379 --ulimit memlock=-1 docker.dragonflydb.io/dragonflydb/dragonfly"
+    };
+
     dag()
         .pipeline("setup")?
         .with_exec(vec!["mkdir", "-p", ".fluentci/dragonflydb"])?
@@ -15,12 +47,11 @@ pub fn setup() -> Result<String, Error> {
     let stdout = dag()
         .pkgx()?
         .with_workdir(".fluentci/dragonflydb")?
-        .with_packages(vec![
-            "github.com/darthsim/overmind",
-            "github.com/tmux/tmux"])?
-        .with_exec(vec![
-            "grep -q dragonflydb Procfile || echo -e 'dragonflydb: pkgx docker run -p $DRAGONFLY_PORT:6379 --ulimit memlock=-1 docker.dragonflydb.io/dragonflydb/dragonfly\\n' >> Procfile",
-        ])?
+        .with_packages(vec!["github.com/darthsim/overmind", "github.com/tmux/tmux"])?
+        .with_exec(vec![&format!(
+            "grep -q dragonflydb: Procfile || echo -e 'dragonflydb: {} \\n' >> Procfile",
+            command
+        )])?
         .stdout()?;
 
     Ok(stdout)