focus.rb 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: be3ca51dc59f5a69a6b15b2381b7596776c7a85b656f2402db409ef11a095e0a
-  data.tar.gz: dfed043bd5a771f9ecc8ad7e22eb898349bbece9cb5c9520da50cf37256e7390
+  metadata.gz: 9dae74d714f49abf09ba3d42f7831f05760346563dd4cce4a233b6f0975020c4
+  data.tar.gz: a2b268db17e8ff009b7d1b672499f5fbbb7cfff6b0f76a66c2c352c3a7767325
 SHA512:
-  metadata.gz: a551d74afb999598545732fe17a94f0e96581b44a6b06e3ea6cce87fab20f66ed51cda65715238abc8c1612fb468b2f72f4774e4ad81ed96310bd35c2c897311
-  data.tar.gz: 3230ffd2acd77850eaf32bd2eb01b3bf150bd531beaa291c3d3cee11cf5f2821ab1307cef3f0e16dc78f56cb0ce32bf29c70b1769835868afa85faa3e2a9f612
+  metadata.gz: 04d3b1ebf9df3df68e3fb2f79fa3ce793b46346b886163ba61cd040d2c467ccaabc64cc25e6a90527dbbb08d3ea941da773172eed5b36e762022f0ca6db57810
+  data.tar.gz: 8f66742d9f3684b84555548b26a983bf70eb34e81f411ef720bc2e58a910b95cd003c6686ddc4c29564344c6d56fc683a0e9f1a69f913a61db9ed28550f62658

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
@@ -25,6 +28,7 @@ Help to focus on your work, block leasure website while you need working, unbloc
 
   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) },
+  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,

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.0
 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