focus.rb 1.0.1 → 1.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: be3ca51dc59f5a69a6b15b2381b7596776c7a85b656f2402db409ef11a095e0a
-  data.tar.gz: dfed043bd5a771f9ecc8ad7e22eb898349bbece9cb5c9520da50cf37256e7390
+  metadata.gz: 1df428809adafd42b392926175afb6d8b4ba2919b2aa089d55fe92f8a9f6bc4b
+  data.tar.gz: 1c7218d6b73f81b58a6d9ca6dbafaa26b0b5df57d6a464cf23774886cdf5e7e4
 SHA512:
-  metadata.gz: a551d74afb999598545732fe17a94f0e96581b44a6b06e3ea6cce87fab20f66ed51cda65715238abc8c1612fb468b2f72f4774e4ad81ed96310bd35c2c897311
-  data.tar.gz: 3230ffd2acd77850eaf32bd2eb01b3bf150bd531beaa291c3d3cee11cf5f2821ab1307cef3f0e16dc78f56cb0ce32bf29c70b1769835868afa85faa3e2a9f612
+  metadata.gz: 53c969c851742f456d2fd003e1d738ebf51bac2e1f162c7b351aa7f360659f21b13c702563ae77684848bd5a65ad377ed3fb3649f1ed9e9e9cec9bd9ef972934
+  data.tar.gz: 5893453871608649f0b26bcd84d753acb4768e2d9f256f9f10827bb19d49abe9e45375ea4e37203ad42d1b7f065af3feab4d852b65e13e0fe9dafde24c0341f5

data/bin/focus

@@ -1,5 +1,7 @@
 #!/usr/bin/env ruby
 
+$LOAD_PATH << File.join(Dir.pwd, "lib")
+
 require "colorize"
 require "optparse"
 require "focus/config"
@@ -10,6 +12,7 @@ config_path = ".config/focus.rb/config.yaml"
 action = :list_block_hosts
 args = []
 options_parser = nil
+dry = false
 
 ARGV.options do |opts|
   options_parser = opts
@@ -21,10 +24,11 @@ Help to focus on your work, block leasure website while you need working, unbloc
   opts.on("-c=PATH", "--config=PATH", "Change the default config path") { |path| config_path = path }
 
   opts.on("-a=BLOCK", "--add-block-host=BLOCK", "Add a new host to block") { |block| action = :add_block_host; args = [block] }
-  opts.on("-d=BLOCK", "--delete-block-host=BLOCK", "Delete a host to block") { |block| action = :delete_block_host; args = [block] }
+  opts.on("--delete-block-host=BLOCK", "Delete a host to block") { |block| action = :delete_block_host; args = [block] }
 
   opts.on("-b", "--block", "Block non-work website") { action = :block }
   opts.on("-r", "--restore", "Restore non-work website") { action = :restore }
+  opts.on("-d", "--dry", "Don't write on real files") { dry = true }
 
   opts.separator ""
 
@@ -34,13 +38,15 @@ Help to focus on your work, block leasure website while you need working, unbloc
 end
 
 config = Focus::Config.new(filepath: config_path)
+host_file = "/etc/hosts"
+host_file = "/dev/null" if dry
 
 ACTIONS = {
   list_block_hosts: lambda { puts config.block_hosts },
   add_block_host: lambda { config.add_block_host(*args) },
-  delete_block_host: lambda { config.add_block_host(*args) },
-  block: lambda { Focus::Host.new.block!(config) },
-  restore: lambda { Focus::Host.new.restore!(config) },
+  delete_block_host: lambda { config.delete_block_host(*args) },
+  block: lambda { Focus::Host.new(host_file).block!(config) },
+  restore: lambda { Focus::Host.new(host_file).restore!(config) },
   help: lambda { puts options_parser }
 }
 

data/lib/focus/config.rb

@@ -3,15 +3,20 @@ require "yaml"
 module Focus
   class Config
     BASE_PATH = ENV["HOME"]
-    DEFAULT_BLOCK_HOSTS = %w(reddit.com youtube.com okcupid.com)
+    DEFAULT_BLOCK_HOSTS = %w(
+      reddit.com youtube.com facebook.com instagram.com twitter.com
+      linkedin.com snapchat.com tumblr.com pinterest.com tiktok.com
+    )
 
     def initialize(filepath:)
       @filepath = File.join(BASE_PATH, filepath)
       dir = File.dirname(@filepath)
       if !File.exist?(dir)
+        puts "Create config directory #{dir}"
         Dir.mkdir(dir)
       end
       if !File.exist?(@filepath)
+        puts "Initialize default configuration #{@filepath}"
         write(
           {
             block_hosts: DEFAULT_BLOCK_HOSTS,
@@ -45,7 +50,7 @@ module Focus
     end
 
     def delete_block_host(blocks)
-      puts "This operation removes \"#{blocks}\" to the list of blocked hosts.".yellow
+      puts "This operation deletes \"#{blocks}\" to the list of blocked hosts.".yellow
       puts "This operation does not unblock the hosts. Execute -b to refresh the blocking or -r to reset it.".yellow
       blocks.split(",").each do |block|
       data = read

data/lib/focus/host.rb

@@ -37,7 +37,9 @@ module Focus
       end
       @all_lines << "### FOCUS:stop (automaticaly generated) ###"
       @all_lines << ""
-      File.write(@file, @all_lines.join("\n"))
+
+      command = "sudo su -c 'echo \"#{@all_lines.join("\n")}\" > #{@file}'"
+      `#{command}`
     end
 
     def block!(config)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: focus.rb
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.1
 platform: ruby
 authors:
 - Arthur Poulet
@@ -9,7 +9,7 @@ autorequire:
 bindir:
 - bin
 cert_chain: []
-date: 2021-09-21 00:00:00.000000000 Z
+date: 2023-06-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: colorize
@@ -55,7 +55,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.27
+rubygems_version: 3.3.22
 signing_key:
 specification_version: 4
 summary: A tool to help you focus on work