lib/makit/dotnet.rb in makit-0.0.34 vs lib/makit/dotnet.rb in makit-0.0.35

@@ -55,14 +55,36 @@
         add_package(project_path, package)
       end
     end
 
     def self.add_reference(project_path, reference_path)
+      if(!File.exist?(project_path))
+        actual_project_path = Makit::DotNet::find_project(project_path)
+        if(!File.exist?(actual_project_path))
+          raise "Project #{project_path} does not exist".colorize(:red)
+        else
+          project_path = actual_project_path
+        end
+      end
+      if(!File.exist?(reference_path))
+        actual_reference_path = Makit::DotNet::find_project(reference_path)
+        if(!File.exist?(actual_reference_path))
+          raise "Project #{reference_path} does not exist".colorize(:red)
+        else
+          reference_path = actual_reference_path
+        end
+      end
       project_content = File.read(project_path)
-      if (project_content.include?("<PackageReference Include=\"#{package}\""))
+      if (project_content.include?(File.basename(reference_path)))
         #puts "  reference ".colorize(:grey) + "#{reference_path}".colorize(:yellow) + " is in ".colorize(:grey) + "#{project_path}".colorize(:yellow)
       else
         "dotnet add #{project_path} reference #{reference_path}".run
+      end
+    end
+
+    def self.add_references(project_path, references)
+      references.each do |reference|
+        add_reference(project_path, reference)
       end
     end
 
     def self.new_solution(name)
       if File.exist? "#{name}.sln"