focus.rb 1.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: be3ca51dc59f5a69a6b15b2381b7596776c7a85b656f2402db409ef11a095e0a
+  data.tar.gz: dfed043bd5a771f9ecc8ad7e22eb898349bbece9cb5c9520da50cf37256e7390
+SHA512:
+  metadata.gz: a551d74afb999598545732fe17a94f0e96581b44a6b06e3ea6cce87fab20f66ed51cda65715238abc8c1612fb468b2f72f4774e4ad81ed96310bd35c2c897311
+  data.tar.gz: 3230ffd2acd77850eaf32bd2eb01b3bf150bd531beaa291c3d3cee11cf5f2821ab1307cef3f0e16dc78f56cb0ce32bf29c70b1769835868afa85faa3e2a9f612

data/bin/focus

@@ -0,0 +1,47 @@
+#!/usr/bin/env ruby
+
+require "colorize"
+require "optparse"
+require "focus/config"
+require "focus/host"
+
+script_name = File.basename($0)
+config_path = ".config/focus.rb/config.yaml"
+action = :list_block_hosts
+args = []
+options_parser = nil
+
+ARGV.options do |opts|
+  options_parser = opts
+
+  opts.banner = "Usage: ruby #{script_name} [--start | --restore]
+Help to focus on your work, block leasure website while you need working, unblock for week-ends !
+"
+
+  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("-b", "--block", "Block non-work website") { action = :block }
+  opts.on("-r", "--restore", "Restore non-work website") { action = :restore }
+
+  opts.separator ""
+
+  opts.on("-h", "--help", "Show this help message.") { action = :help }
+
+  opts.parse!
+end
+
+config = Focus::Config.new(filepath: config_path)
+
+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) },
+  help: lambda { puts options_parser }
+}
+
+ACTIONS.fetch(action, ACTIONS[:help]).call()

data/lib/focus/config.rb

@@ -0,0 +1,58 @@
+require "yaml"
+
+module Focus
+  class Config
+    BASE_PATH = ENV["HOME"]
+    DEFAULT_BLOCK_HOSTS = %w(reddit.com youtube.com okcupid.com)
+
+    def initialize(filepath:)
+      @filepath = File.join(BASE_PATH, filepath)
+      dir = File.dirname(@filepath)
+      if !File.exist?(dir)
+        Dir.mkdir(dir)
+      end
+      if !File.exist?(@filepath)
+        write(
+          {
+            block_hosts: DEFAULT_BLOCK_HOSTS,
+          },
+        )
+      end
+    end
+
+    def read
+      YAML.load_file(@filepath)
+    end
+
+    def write(data)
+      data[:block_hosts].uniq!
+      File.write(@filepath, data.to_yaml)
+    end
+
+    def block_hosts
+      read[:block_hosts]
+    end
+
+    def add_block_host(blocks)
+      puts "This operation adds \"#{blocks}\" to the list of blocked hosts.".yellow
+      puts "This operation does not block the hosts. Execute -b to execute the blocking.".yellow
+      blocks.split(",").each do |block|
+        data = read
+        data[:block_hosts] << block
+        write(data)
+        puts "Added host \"#{block}\"!".green
+      end
+    end
+
+    def delete_block_host(blocks)
+      puts "This operation removes \"#{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[:block_hosts].delete_if { |host| host == block }
+      write(data)
+      puts "Removed host \"#{block}\"!".green
+      end
+    end
+  end
+end

data/lib/focus/host.rb

@@ -0,0 +1,63 @@
+module Focus
+  class Host
+    def initialize(file = "/etc/hosts")
+      @file = file
+      read!
+    end
+
+    def get_index
+      start_index = @all_lines.find_index { |line| /### FOCUS:start \(automaticaly generated\) ###/i === line }
+      stop_index = @all_lines.find_index { |line| /### FOCUS:stop \(automaticaly generated\) ###/i === line }
+      { start: start_index + 1, stop: stop_index - 1 } if start_index && stop_index
+    end
+
+    def read!
+      @all_lines = File.read(@file).split("\n")
+      index = get_index
+      @data = []
+      if index
+        selected_lines = @all_lines[index[:start]..index[:stop]]
+        @data = selected_lines.map { |line| line.split(/\s+/) }.map { |ip, host| { ip: ip, host: host } }
+      end
+    end
+
+    def overwrite!
+      index = get_index
+      if index
+        count = index[:stop] - index[:start]
+        @all_lines << "" if @all_lines[index[:start] - 2] != ""
+        @all_lines.slice!(index[:start] - 1, count + 3)
+      else
+        @all_lines << "" if @all_lines.last != ""
+      end
+
+      @all_lines << "### FOCUS:start (automaticaly generated) ###"
+      @data.each do |managed|
+        @all_lines << "#{managed[:ip]} #{managed[:host]}"
+      end
+      @all_lines << "### FOCUS:stop (automaticaly generated) ###"
+      @all_lines << ""
+      File.write(@file, @all_lines.join("\n"))
+    end
+
+    def block!(config)
+      puts "Blocking the hosts".green
+      read!
+      config.block_hosts.each do |block_host|
+        puts block_host.red
+        @data << { host: block_host, ip: "127.0.0.1" }
+        @data << { host: "www.#{block_host}", ip: "127.0.0.1" }
+      end
+      @data.uniq!
+      overwrite!
+      puts "Done!".green
+    end
+
+    def restore!(config)
+      puts "Restoring blocked hosts".green
+      @data = []
+      overwrite!
+      puts "Done!".green
+    end
+  end
+end

metadata

@@ -0,0 +1,62 @@
+--- !ruby/object:Gem::Specification
+name: focus.rb
+version: !ruby/object:Gem::Version
+  version: 1.0.1
+platform: ruby
+authors:
+- Arthur Poulet
+autorequire:
+bindir:
+- bin
+cert_chain: []
+date: 2021-09-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.8.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.8.1
+description: A tool to help you focus on work* Block hosts names localy
+email:
+- arthur.focus@sceptique.eu
+executables:
+- focus
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/focus
+- lib/focus/config.rb
+- lib/focus/host.rb
+homepage: https://git.sceptique.eu/Sceptique/focus.rb
+licenses:
+- GPL-3.0
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.27
+signing_key:
+specification_version: 4
+summary: A tool to help you focus on work
+test_files: []