rust-lang-nursery#8: Create project in db

Author: Jakub Matraszek

migrations/20170125121136_create_contributors/down.sql

@@ -0,0 +1 @@
+DROP TABLE contributors;

migrations/20170125121136_create_contributors/up.sql

@@ -0,0 +1,10 @@
+CREATE TABLE contributors (
+    id SERIAL NOT NULL,
+    name VARCHAR,
+    email VARCHAR
+);
+
+ALTER TABLE ONLY contributors
+    ADD CONSTRAINT contributors_pkey PRIMARY KEY (id);
+
+CREATE UNIQUE INDEX contributors_id_idx ON contributors USING btree (id);

migrations/20170125121346_create_projects/down.sql

@@ -0,0 +1 @@
+DROP TABLE projects;

migrations/20170125121346_create_projects/up.sql

@@ -0,0 +1,11 @@
+CREATE TABLE projects (
+    id SERIAL NOT NULL,
+    name VARCHAR NOT NULL,
+    path VARCHAR NOT NULL,
+    github_link VARCHAR NOT NULL
+);
+
+ALTER TABLE ONLY projects
+    ADD CONSTRAINT projects_pkey PRIMARY KEY (id);
+
+CREATE UNIQUE INDEX projects_id_idx ON projects USING btree (id);

migrations/20170125123356_add_project_to_releases/down.sql

@@ -0,0 +1,2 @@
+ALTER TABLE ONLY releases
+	DROP COLUMN project_id;

migrations/20170125123356_add_project_to_releases/up.sql

@@ -0,0 +1,8 @@
+ALTER TABLE ONLY releases
+	ADD COLUMN project_id integer NOT NULL;
+
+ALTER TABLE ONLY releases
+	ADD CONSTRAINT projects
+	FOREIGN KEY (project_id)
+	REFERENCES projects (id)
+	ON DELETE CASCADE;

src/bin/populate.rs

@@ -38,7 +38,19 @@ fn main() {
         .arg(Arg::with_name("filepath")
             .short("p")
             .long("path")
-            .help("filepath of the rust source code")
+            .help("filepath of the source code")
+            .takes_value(true)
+            .required(true))
+        .arg(Arg::with_name("name")
+            .short("n")
+            .long("name")
+            .help("name of the project")
+            .takes_value(true)
+            .required(true))
+        .arg(Arg::with_name("github_link")
+            .short("l")
+            .long("link")
+            .help("GitHub link of the project")
             .takes_value(true)
             .required(true))
         .get_matches();
@@ -67,9 +79,22 @@ fn main() {
         }
     }
 
+    // get name
+    let name = matches.value_of("name").unwrap();
+    println!("Project name: {}", name);
+
     // get path to git repo
     let path = matches.value_of("filepath").unwrap();
-    println!("Path to rust repo: {}", path);
+    println!("Path to project's repo: {}", path);
+
+    // get github link
+    let link = matches.value_of("github_link").unwrap();
+    println!("GitHub link: {}", link);
+
+    // create project
+    contributors::create_project(&connection, name, path, link);
+
+    panic!("Project created!");
 
     // create releases
 

src/lib.rs

@@ -20,6 +20,7 @@ use std::process::Command;
 pub mod schema;
 pub mod models;
 
+use self::models::{Project, NewProject};
 use self::models::{Commit, NewCommit};
 use self::models::{Release, NewRelease};
 
@@ -49,6 +50,20 @@ pub fn create_commit<'a>(conn: &PgConnection, sha: &'a str, author_name: &'a str
         .expect("Error saving new commit")
 }
 
+pub fn create_project(conn: &PgConnection, name: &str, path: &str, github_link: &str) -> Project {
+    use schema::projects;
+
+    let new_project = NewProject {
+        name: name,
+        path: path,
+        github_link: github_link
+    };
+
+    diesel::insert(&new_project).into(projects::table)
+        .get_result(conn)
+        .expect("Error saving new project")
+}
+
 
 pub fn create_release(conn: &PgConnection, version: &str) -> Release {
     use schema::releases;

src/models.rs

@@ -1,3 +1,12 @@
+#[derive(Debug,Identifiable,Queryable,Associations)]
+#[has_many(releases)]
+pub struct Project {
+    pub id: i32,
+    pub name: String,
+    pub path: String,
+    pub github_link: String,
+}
+
 #[derive(Debug,Identifiable,Queryable,Associations)]
 #[belongs_to(Release)]
 pub struct Commit {
@@ -10,11 +19,22 @@ pub struct Commit {
 
 #[derive(Debug,Identifiable,Queryable,Associations)]
 #[has_many(commits)]
+#[belongs_to(Project)]
 pub struct Release {
     pub id: i32,
     pub version: String,
+    pub project_id: i32,
 }
 
+use super::schema::projects;
+
+#[derive(Insertable)]
+#[table_name="projects"]
+pub struct NewProject<'a> {
+    pub name: &'a str,
+    pub path: &'a str,
+    pub github_link: &'a str,
+}
 use super::schema::commits;
 
 #[derive(Insertable)]