Merge pull request #346 from HallerPatrick/master

Added pub package manager for dart projects

Author: o2sh

Diff for: Cargo.lock

@@ -33,6 +33,7 @@ image = "0.23.12"
 regex = "1"
 error-chain = "0.12"
 toml = "0.5.7"
+yaml-rust = "0.4"
 serde = "1.0.118"
 serde_json = "1.0.60"
 

Diff for: Cargo.toml

@@ -36,6 +36,10 @@ impl DependencyDetector {
             String::from("requirements.txt"),
             (package_parser::pip, package_manager::PackageManager::Pip),
         );
+        package_managers.insert(
+            String::from("pubspec.yaml"),
+            (package_parser::pub_packages, package_manager::PackageManager::Pub),
+        );
 
         DependencyDetector { package_managers }
     }

Diff for: src/onefetch/deps/mod.rs

@@ -6,6 +6,7 @@ pub enum PackageManager {
     GoModules,
     Npm,
     Pip,
+    Pub,
 }
 
 impl std::fmt::Display for PackageManager {
@@ -15,6 +16,7 @@ impl std::fmt::Display for PackageManager {
             PackageManager::GoModules => write!(f, "go modules"),
             PackageManager::Npm => write!(f, "npm"),
             PackageManager::Pip => write!(f, "pip"),
+            PackageManager::Pub => write!(f, "pub"),
         }
     }
 }

Diff for: src/onefetch/deps/package_manager.rs

@@ -1,5 +1,5 @@
 use crate::onefetch::error::*;
-use {regex::Regex, toml::Value};
+use {regex::Regex, toml::Value, yaml_rust::YamlLoader};
 
 pub fn cargo(contents: &str) -> Result<usize> {
     let parsed = contents.parse::<Value>()?;
@@ -43,3 +43,13 @@ pub fn pip(contents: &str) -> Result<usize> {
 
     Ok(count)
 }
+
+pub fn pub_packages(contents: &str) -> Result<usize> {
+    match YamlLoader::load_from_str(contents) {
+        Ok(parsed) => match &parsed[0]["dependencies"].as_hash() {
+            Some(deps) => Ok(deps.len()),
+            None => Ok(0),
+        },
+        Err(_) => Ok(0),
+    }
+}