feat: ref codes

closes #8

src/api/model.rs

@@ -207,6 +207,7 @@ pub struct CreateVmRequest {
     pub template_id: u64,
     pub image_id: u64,
     pub ssh_key_id: u64,
+    pub ref_code: Option<String>
 }
 
 #[derive(Serialize, Deserialize, JsonSchema)]

src/api/routes.rs

@@ -361,14 +361,20 @@ async fn v1_create_vm_order(
     auth: Nip98Auth,
     db: &State<Arc<dyn LNVpsDb>>,
     provisioner: &State<Arc<LNVpsProvisioner>>,
-    req: Json<CreateVmRequest>,
+    req: Json<CreateVmRequest>
 ) -> ApiResult<ApiVmStatus> {
     let pubkey = auth.event.pubkey.to_bytes();
     let uid = db.upsert_user(&pubkey).await?;
 
     let req = req.0;
     let rsp = provisioner
-        .provision(uid, req.template_id, req.image_id, req.ssh_key_id)
+        .provision(
+            uid,
+            req.template_id,
+            req.image_id,
+            req.ssh_key_id,
+            req.ref_code,
+        )
         .await?;
     ApiData::ok(vm_to_status(db, rsp, None).await?)
 }

src/mocks.rs

@@ -173,7 +173,6 @@ impl LNVpsDb for MockDb {
                     pubkey: pubkey.to_vec(),
                     created: Utc::now(),
                     email: None,
-                    contact_nip4: false,
                     contact_nip17: false,
                     contact_email: false,
                 },
@@ -193,7 +192,6 @@ impl LNVpsDb for MockDb {
             u.email = user.email.clone();
             u.contact_email = user.contact_email;
             u.contact_nip17 = user.contact_nip17;
-            u.contact_nip4 = user.contact_nip4;
         }
         Ok(())
     }
@@ -377,6 +375,7 @@ impl LNVpsDb for MockDb {
                 disk_id: vm.disk_id,
                 mac_address: vm.mac_address.clone(),
                 deleted: false,
+                ref_code: vm.ref_code.clone(),
             },
         );
         Ok(max_id + 1)

src/provisioner/lnvps.rs

@@ -249,6 +249,7 @@ impl LNVpsProvisioner {
         template_id: u64,
         image_id: u64,
         ssh_key_id: u64,
+        ref_code: Option<String>,
     ) -> Result<Vm> {
         let user = self.db.get_user(user_id).await?;
         let template = self.db.get_vm_template(template_id).await?;
@@ -272,6 +273,7 @@ impl LNVpsProvisioner {
 
         let client = get_host_client(&pick_host, &self.provisioner_config)?;
         let mut new_vm = Vm {
+            id: 0,
             host_id: pick_host.id,
             user_id: user.id,
             image_id: image.id,
@@ -280,7 +282,9 @@ impl LNVpsProvisioner {
             created: Utc::now(),
             expires: Utc::now(),
             disk_id: pick_disk.id,
-            ..Default::default()
+            mac_address: "NOT FILLED YET".to_string(),
+            deleted: false,
+            ref_code,
         };
 
         // ask host client to generate the mac address
@@ -468,7 +472,9 @@ mod tests {
         };
         let ssh_key = db.insert_user_ssh_key(&new_key).await?;
 
-        let vm = provisioner.provision(user_id, 1, 1, ssh_key).await?;
+        let vm = provisioner
+            .provision(user_id, 1, 1, ssh_key, Some("mock-ref".to_string()))
+            .await?;
         println!("{:?}", vm);
         provisioner.spawn_vm(vm.id).await?;
 
@@ -477,6 +483,7 @@ mod tests {
         assert_eq!(1, arp.len());
         let arp = arp.first().unwrap();
         assert_eq!(&vm.mac_address, &arp.mac_address);
+        assert_eq!(vm.ref_code, Some("mock-ref".to_string()));
         assert_eq!(ROUTER_BRIDGE, arp.interface.as_ref().unwrap());
         println!("{:?}", arp);
 

src/worker.rs

@@ -241,9 +241,6 @@ impl Worker {
                 sender.send(msg).await?;
             }
         }
-        if user.contact_nip4 {
-            // TODO: send nip4 dm
-        }
         if user.contact_nip17 {
             if let Some(c) = self.nostr.as_ref() {
                 let sig = c.signer().await?;