From 960b14fa20aa13820b793209a9e07fd83caef247 Mon Sep 17 00:00:00 2001
From: SteveLauC <stevelauc@outlook.com>
Date: Sun, 7 Jul 2024 10:37:07 +0800
Subject: [PATCH] feat: support Poetry (#790)

---
 src/config.rs        | 1 +
 src/main.rs          | 1 +
 src/steps/generic.rs | 6 ++++++
 3 files changed, 8 insertions(+)

diff --git a/src/config.rs b/src/config.rs
index 150e8325..0f572dd1 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -125,6 +125,7 @@ pub enum Step {
     Pkgin,
     PlatformioCore,
     Pnpm,
+    Poetry,
     Powershell,
     Protonup,
     Pyenv,
diff --git a/src/main.rs b/src/main.rs
index cbaa632d..e219f207 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -415,6 +415,7 @@ fn run() -> Result<()> {
     runner.execute(Step::Lensfun, "Lensfun's database update", || {
         generic::run_lensfun_update_data(&ctx)
     })?;
+    runner.execute(Step::Poetry, "Poetry", || generic::run_poetry(&ctx))?;
 
     if should_run_powershell {
         runner.execute(Step::Powershell, "Powershell Modules Update", || {
diff --git a/src/steps/generic.rs b/src/steps/generic.rs
index 9a70d494..d305cbbc 100644
--- a/src/steps/generic.rs
+++ b/src/steps/generic.rs
@@ -1002,3 +1002,9 @@ pub fn run_lensfun_update_data(ctx: &ExecutionContext) -> Result<()> {
             .status_checked_with_codes(&[EXIT_CODE_WHEN_NO_UPDATE])
     }
 }
+
+pub fn run_poetry(ctx: &ExecutionContext) -> Result<()> {
+    let poetry = require("poetry")?;
+    print_separator("Poetry");
+    ctx.run_type().execute(poetry).args(["self", "update"]).status_checked()
+}