Lock command: save or print lockfile

Author: ysbaddaden

src/cli.cr

@@ -48,7 +48,7 @@ module Shards
         when "list"
           Commands::List.run(path, tree: args.includes?("--tree"))
         when "lock"
-          Commands::Lock.run(path)
+          Commands::Lock.run(path, print: args.includes?("--print"))
         when "outdated"
           Commands::Outdated.run(path, prereleases: args.includes?("--pre"))
         when "prune"

src/commands/lock.cr

@@ -4,29 +4,18 @@ require "../solver"
 module Shards
   module Commands
     class Lock < Command
-      def run
+      def run(print = false)
+        Shards.logger.info { "Resolving dependencies" }
+
         solver = Solver.new(spec)
         solver.prepare(development: !Shards.production?)
 
         if solution = solver.solve
-          Shards.logger.info { "Found solution:" }
-
-          puts "version: 1.1"
-          puts "shards:"
-
-          solution.sort_by!(&.name).each do |rs|
-            key = rs.resolver.class.key
-
-            puts "  #{rs.name}:"
-            puts "    #{key}: #{rs.resolver.dependency[key]}"
-
-            if rs.commit
-              puts "    commit: #{rs.commit}"
-            else
-              puts "    version: #{rs.version}" unless rs.commit
-            end
-
-            puts
+          if print
+            to_lockfile(solution, STDOUT)
+          else
+            Shards.logger.info { "Writing #{LOCK_FILENAME}" }
+            File.open(LOCK_FILENAME, "w") { |file| to_lockfile(solution, file) }
           end
         else
           solver.each_conflict do |message|
@@ -35,6 +24,26 @@ module Shards
           Shards.logger.error { "Failed to find a solution" }
         end
       end
+
+      private def to_lockfile(solution, io)
+        io << "version: 1.1\n"
+        io << "shards:\n"
+
+        solution.sort_by!(&.name).each do |rs|
+          key = rs.resolver.class.key
+
+          io << "  " << rs.name << ":\n"
+          io << "    " << key << ": " << rs.resolver.dependency[key] << '\n'
+
+          if rs.commit
+            io << "    commit: " << rs.commit << '\n'
+          else
+            io << "    version: " << rs.version << '\n'
+          end
+
+          puts
+        end
+      end
     end
   end
 end