database: Add links column to versions table (#5111)

Author: arlosi

migrations/2022-08-18-152941_links-column/down.sql

@@ -0,0 +1,2 @@
+ALTER TABLE versions
+    DROP COLUMN links;

migrations/2022-08-18-152941_links-column/up.sql

@@ -0,0 +1,2 @@
+ALTER TABLE versions
+    ADD COLUMN links VARCHAR NULL;

src/controllers/krate/publish.rs

@@ -171,6 +171,7 @@ pub fn publish(req: &mut dyn RequestExt) -> EndpointResult {
             file_length as i32,
             user.id,
             hex_cksum.clone(),
+            links.clone(),
         )?
         .save(&conn, &verified_email_address)?;
 

src/downloads_counter.rs

@@ -458,6 +458,7 @@ mod tests {
                 0,
                 self.user.id,
                 "0000000000000000000000000000000000000000000000000000000000000000".to_string(),
+                None,
             )
             .expect("failed to create version")
             .save(conn, "ghost@example.com")

src/models/version.rs

@@ -24,6 +24,7 @@ pub struct Version {
     pub crate_size: Option<i32>,
     pub published_by: Option<i32>,
     pub checksum: Option<String>,
+    pub links: Option<String>,
 }
 
 #[derive(Insertable, Debug)]
@@ -36,6 +37,7 @@ pub struct NewVersion {
     crate_size: Option<i32>,
     published_by: i32,
     checksum: String,
+    links: Option<String>,
 }
 
 /// The highest version (semver order) and the most recently updated version.
@@ -133,6 +135,7 @@ impl NewVersion {
         crate_size: i32,
         published_by: i32,
         checksum: String,
+        links: Option<String>,
     ) -> AppResult<Self> {
         let features = serde_json::to_value(features)?;
 
@@ -144,6 +147,7 @@ impl NewVersion {
             crate_size: Some(crate_size),
             published_by,
             checksum,
+            links,
         };
 
         new_version.validate_license(license_file)?;

src/schema.rs

@@ -994,6 +994,12 @@ table! {
         ///
         /// (Automatically generated by Diesel.)
         checksum -> Nullable<Bpchar>,
+        /// The `links` column of the `versions` table.
+        ///
+        /// Its SQL type is `Nullable<Varchar>`.
+        ///
+        /// (Automatically generated by Diesel.)
+        links -> Nullable<Varchar>,
     }
 }
 

src/tests/builders/version.rs

@@ -19,6 +19,7 @@ pub struct VersionBuilder<'a> {
     size: i32,
     yanked: bool,
     checksum: String,
+    links: Option<String>,
 }
 
 impl<'a> VersionBuilder<'a> {
@@ -43,6 +44,7 @@ impl<'a> VersionBuilder<'a> {
             size: 0,
             yanked: false,
             checksum: String::new(),
+            links: None,
         }
     }
 
@@ -94,6 +96,7 @@ impl<'a> VersionBuilder<'a> {
             self.size,
             published_by,
             self.checksum,
+            self.links,
         )?
         .save(connection, "someone@example.com")?;
 

src/worker/dump_db/dump-db.toml

@@ -214,6 +214,7 @@ license = "public"
 crate_size = "public"
 published_by = "public"
 checksum = "public"
+links = "public"
 
 [versions_published_by.columns]
 version_id = "private"

src/worker/update_downloads.rs

@@ -109,6 +109,7 @@ mod test {
             0,
             user_id,
             "0000000000000000000000000000000000000000000000000000000000000000".to_string(),
+            None,
         )
         .unwrap();
         let version = version.save(conn, "someone@example.com").unwrap();