refactor: convert to workspace

lnvps_api/src/data_migration/ip6_init.rs

@@ -0,0 +1,53 @@
+use crate::data_migration::DataMigration;
+use crate::provisioner::{LNVpsProvisioner, NetworkProvisioner};
+use ipnetwork::IpNetwork;
+use lnvps_db::LNVpsDb;
+use std::future::Future;
+use std::pin::Pin;
+use std::str::FromStr;
+use std::sync::Arc;
+use chrono::Utc;
+use log::info;
+
+pub struct Ip6InitDataMigration {
+    db: Arc<dyn LNVpsDb>,
+    provisioner: Arc<LNVpsProvisioner>,
+}
+
+impl Ip6InitDataMigration {
+    pub fn new(db: Arc<dyn LNVpsDb>, provisioner: Arc<LNVpsProvisioner>) -> Ip6InitDataMigration {
+        Self { db, provisioner }
+    }
+}
+
+impl DataMigration for Ip6InitDataMigration {
+    fn migrate(&self) -> Pin<Box<dyn Future<Output = anyhow::Result<()>> + Send>> {
+        let db = self.db.clone();
+        let provisioner = self.provisioner.clone();
+        Box::pin(async move {
+            let net = NetworkProvisioner::new(db.clone());
+            let vms = db.list_vms().await?;
+            for vm in vms {
+                if vm.expires < Utc::now() {
+                    continue;
+                }
+                let host = db.get_host(vm.host_id).await?;
+                let ips = db.list_vm_ip_assignments(vm.id).await?;
+                // if no ipv6 address is picked already pick one
+                if ips.iter().all(|i| {
+                        IpNetwork::from_str(&i.ip)
+                            .map(|i| i.is_ipv4())
+                            .unwrap_or(false)
+                    })
+                {
+                    let ips_pick = net.pick_ip_for_region(host.region_id).await?;
+                    if let Some(mut v6) = ips_pick.ip6 {
+                        info!("Assigning ip {} to vm {}", v6.ip, vm.id);
+                        provisioner.assign_available_v6_to_vm(&vm, &mut v6).await?;
+                    }
+                }
+            }
+            Ok(())
+        })
+    }
+}