Check if a crate is library in add_package

Author: onur

src/db/add_package.rs

@@ -8,7 +8,7 @@ use std::io::BufReader;
 use std::path::{Path, PathBuf};
 use std::fs;
 
-use cargo::core::Package;
+use cargo::core::{Package, TargetKind};
 use rustc_serialize::json::{Json, ToJson};
 use slug::slugify;
 use hyper::client::Client;
@@ -33,6 +33,10 @@ pub fn add_package_into_database(conn: &Connection,
     let rustdoc = get_rustdoc(&pkg).unwrap_or(None);
     let readme = get_readme(&pkg).unwrap_or(None);
     let (release_time, yanked, downloads) = try!(get_release_time_yanked_downloads(&pkg));
+    let is_library = match pkg.targets()[0].kind() {
+        &TargetKind::Lib(_) => true,
+        _ => false,
+    };
 
     let release_id: i32 = {
         let rows = try!(conn.query("SELECT id FROM releases WHERE crate_id = $1 AND version = $2",
@@ -45,10 +49,12 @@ pub fn add_package_into_database(conn: &Connection,
                                             rustdoc_status, test_status, license, repository_url, \
                                             homepage_url, description, description_long, readme, \
                                             authors, keywords, have_examples, downloads, files, \
-                                            doc_targets \
+                                            doc_targets, is_library \
                                         ) \
                                         VALUES ( $1,  $2,  $3,  $4, $5, $6,  $7, $8, $9, $10, \
-                                        $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21 ) \
+                                                 $11, $12, $13, $14, $15, $16, $17, $18, $19, \
+                                                 $20, $21, $22 \
+                                        ) \
                                         RETURNING id",
                                        &[&crate_id,
                                          &format!("{}", pkg.manifest().version()),
@@ -70,7 +76,8 @@ pub fn add_package_into_database(conn: &Connection,
                                          &res.have_examples,
                                          &downloads,
                                          &files,
-                                         &doc_targets.to_json()]));
+                                         &doc_targets.to_json(),
+                                         &is_library]));
             // return id
             rows.get(0).get(0)
 
@@ -80,7 +87,7 @@ pub fn add_package_into_database(conn: &Connection,
                              $8, test_status = $9, license = $10, repository_url = $11, \
                              homepage_url = $12, description = $13, description_long = $14, \
                              readme = $15, authors = $16, keywords = $17, have_examples = $18, \
-                             downloads = $19, files = $20, doc_targets = $21 \
+                             downloads = $19, files = $20, doc_targets = $21, is_library = $22 \
                              WHERE crate_id = $1 AND version = $2",
                             &[&crate_id,
                               &format!("{}", pkg.manifest().version()),
@@ -102,7 +109,8 @@ pub fn add_package_into_database(conn: &Connection,
                               &res.have_examples,
                               &downloads,
                               &files,
-                              &doc_targets.to_json()]));
+                              &doc_targets.to_json(),
+                              &is_library]));
             rows.get(0).get(0)
         }
     };

src/db/mod.rs

@@ -45,6 +45,7 @@ pub fn create_tables(conn: &Connection) -> Result<(), Error> {
             dependencies JSON, \
             target_name VARCHAR(255), \
             yanked BOOL DEFAULT FALSE, \
+            is_library BOOL DEFAULT TRUE, \
             build_status BOOL DEFAULT FALSE, \
             rustdoc_status BOOL DEFAULT FALSE, \
             test_status BOOL DEFAULT FALSE, \