ttoken 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: f57d2d3a74e0441e24f73e7776033882771c2710d8e57979e78d7ccd859f903e
+  data.tar.gz: 604f0afb5e97d18bf223d41dc764bd94597ab23c46d89eacb59b1b4793696fcb
+SHA512:
+  metadata.gz: 5b4999a535b8fc61cc4cf3749c1099b46a7c73a8fe18cec75ab6a28c7e618a640e03860fb39c5b389bccb26d3eff39cec1fe7f676d58e73cdb0f81d938e2736b
+  data.tar.gz: 22581b23450161e2db3fbbb6d83cfaa81b63aef06ee1ec958540f0ce7200e193cb62fd00af24318efc020bf2bdac4db6d260a5a731f2fe1b1d300e12a5e5346c

data/.gitignore

@@ -0,0 +1,9 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+*.gem

data/Gemfile

@@ -0,0 +1,5 @@
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in ttoken.gemspec
+gemspec
+

data/Gemfile.lock

@@ -0,0 +1,23 @@
+GEM
+  remote: https://rubygems.org/
+  specs:
+    addressable (2.7.0)
+      public_suffix (>= 2.0.2, < 5.0)
+    fileutils (1.4.1)
+    openssl (2.2.0)
+    public_suffix (4.0.5)
+    rotp (6.0.0)
+      addressable (~> 2.7)
+    yaml (0.1.0)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  fileutils
+  openssl
+  rotp
+  yaml
+
+BUNDLED WITH
+   2.1.4

data/README.md

@@ -0,0 +1,52 @@
+# Ttoken
+
+Ttoken is a command-line program to generate time-based token/OTP.
+
+Apart from generating time-based OTP, it provides the feature of creating pin+token and copies it to the clipboard so that user can enter it anywhere by just the click of Ctrl+v.  
+
+## Installation
+
+```ruby
+# yum install ruby rubygems
+
+# gem install ttoken
+```
+
+## Usage
+
+Gem provides the executable command-line program - ttoken. 
+
+1) To encrypt and save your password/pin in ttoken use the below command. 
+
+```ruby
+# ttoken --encrypt
+```
+
+2) Create the file `/etc/ttoken/ttoken.yml` and enter the below content. 
+
+```ruby
+# Issuer - The site or app you are creating the OTP
+:issuer: 'Red Hat'
+
+# Time-based token provided by issuer
+:token: "1234abcdxyz"
+
+# Enable pin+token
+:pinplustoken: yes
+
+```
+Make sure you change the issuer and token. 
+
+
+
+3) Running the command will generate pin+token and will copy it to the clipboard. 
+
+
+```ruby
+# ttoken
+```
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/patilsuraj767/ttoken.
+

data/bin/ttoken

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH.unshift(File.expand_path('../../lib', __FILE__))
+
+require 'ttoken'
+
+CONFIG_FILE = '/etc/ttoken/ttoken.yml'.freeze
+ENCRYPTION_KEY_FILE = File.expand_path('~/.ttoken/ttoken_encryption_key.yml')
+
+Ttoken.setup
+Ttoken.run
+
+
+

data/config/ttoken.yml.example

@@ -0,0 +1,8 @@
+# Enable pin+token
+:pinplustoken: yes
+
+# Issuer - The site or app you are creating the OTP
+:issuer: 'Red Hat'
+
+# Time-based token provided by issuer
+:token: '1234abcdxyz'

data/lib/ttoken.rb

@@ -0,0 +1,54 @@
+require "ttoken/version"
+require 'rotp'
+require 'clipboard'
+require 'io/console'
+
+module Ttoken
+  require 'ttoken/config'
+  require 'ttoken/encrypt'
+  require 'ttoken/store'
+
+  class << self
+    attr_accessor :config
+
+    def setup
+      self.config = Config.new
+    end
+
+    def run
+      encrypt = Encrypt.new
+      if ARGV.length > 0 &&  ARGV[0] == '--encrypt'
+        pin = STDIN.getpass("Password/Pin:")
+        encrypted_pin = encrypt.encrypt_pin(pin)
+        Store.save_password(encrypted_pin)
+      else
+        token = generate_token(config.token, config.issuer)
+        if config.pinplustoken
+          password = Store.retrive_password
+          decrypted_pin = encrypt.decrypt_pin(password)
+          send_to_clipboard(decrypted_pin + token)
+        else
+          send_to_clipboard(token)
+        end
+      end
+    end
+
+    def generate_token(token, issuer)
+      begin
+        ROTP::TOTP.new(token, issuer: issuer).now
+      rescue StandardError => e
+        raise " \nSomething went wrong. Please check if all parameters in /etc/ttoken/ttoken.yml are correct.\n Error: #{e.message}"
+      end
+    end
+
+    def send_to_clipboard(text)
+      begin
+        system("echo -n #{text} | xclip -selection clipboard")
+        puts 'Data copied to clipboard. Ready to paste.'
+      rescue StandardError => e
+        raise "Problem in copying data to clipboard. Error: #{e.message}"
+      end
+    end
+
+  end
+end

data/lib/ttoken/config.rb

@@ -0,0 +1,34 @@
+require 'yaml'
+module Ttoken
+  class Config
+    attr_accessor :config_file, :token, :issuer, :pinplustoken
+
+    def initialize
+      @config_file = config_file_path
+      @options = load_config
+      @token = @options.fetch(:token, nil)
+      @issuer = @options.fetch(:issuer, nil)
+      @pinplustoken = @options.fetch(:pinplustoken, false)
+    end
+
+    def config_file_path
+      File.exist?(CONFIG_FILE) ? CONFIG_FILE : File.join(source_path, 'config/ttoken.yml')
+    end
+
+    def load_config
+      if File.exist?(config_file)
+        YAML.load(File.open(config_file)) || {}
+      else
+        puts "Config file #{config_file} not found, using default configuration"
+        {}
+      end
+    rescue StandardError => e
+      raise "Couldn't load configuration file. Error: #{e.message}"
+    end
+
+    def source_path
+      File.expand_path('../../..', __FILE__)
+    end
+
+  end
+end

data/lib/ttoken/encrypt.rb

@@ -0,0 +1,44 @@
+require 'openssl'
+class Encrypt
+  CIPHER = OpenSSL::Cipher.new('aes-256-cbc')
+
+  def encrypt_pin(pin)
+    if is_encryptable?(pin)
+      cipher = CIPHER.encrypt
+      cipher.key = encryption_key
+      s = cipher.update(pin) + cipher.final
+      s.unpack('H*')[0].upcase
+    end
+  end
+
+  def decrypt_pin(pin)
+    if is_decryptable?(pin)
+      cipher = CIPHER.decrypt
+      cipher.key = encryption_key
+      s = [pin].pack("H*").unpack("C*").pack("c*")
+      begin
+        cipher.update(s) + cipher.final
+      rescue StandardError => e
+      raise "\n Problem in decrypting password. To resolve the issue encrypt and save the password again using command # ttoken --encrypt. \n Error: #{e.message}"
+      end
+    end
+  end
+
+  def is_encryptable?(str)
+    !encryption_key.nil? && !str.nil?
+  end
+
+  def is_decryptable?(str)
+    !encryption_key.nil? && !str.nil?
+  end
+
+  def encryption_key
+    if File.exist?(ENCRYPTION_KEY_FILE)
+      Store.retrive_encryption_key
+    else
+      key = CIPHER.random_key
+      Store.save_encryption_key(key)
+      key
+    end
+  end
+end

data/lib/ttoken/store.rb

@@ -0,0 +1,44 @@
+require 'yaml'
+class Store
+  PASSWORD_FILE = File.expand_path('~/.ttoken/ttoken.yml')
+
+  def self.save_password(password)
+    create_dir
+    data = {password: password}
+    begin
+      File.write(PASSWORD_FILE, data.to_yaml)
+      puts 'Password saved successfully. Use command #ttoken to generate pin+token'
+    rescue StandardError => e
+      raise "Unable to write password file. Error: #{e.message}"
+    end
+  end
+
+  def self.retrive_password
+    if File.exist?(PASSWORD_FILE)
+      pf = YAML.load(File.read(PASSWORD_FILE))
+      pf[:password]
+    else
+      puts 'Password file does not exits. Save the password/pin using command # ttoken --encrypt'
+      exit 2
+    end
+  end
+
+  def self.retrive_encryption_key
+    if File.exist?(ENCRYPTION_KEY_FILE)
+      pf = YAML.load(File.read(ENCRYPTION_KEY_FILE))
+      pf[:encryption_key]
+    end
+  end
+
+  def self.save_encryption_key(encryption_key)
+      create_dir
+      data = {encryption_key: encryption_key}
+      File.write(ENCRYPTION_KEY_FILE, data.to_yaml)
+  end
+
+  def self.create_dir
+    dir = File.expand_path('~/.ttoken')
+    Dir.mkdir(dir) unless Dir.exist?(dir)  
+  end
+
+end

data/lib/ttoken/version.rb

@@ -0,0 +1,3 @@
+module Ttoken
+  VERSION = "0.1.0"
+end

data/ttoken.gemspec

@@ -0,0 +1,23 @@
+require_relative 'lib/ttoken/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "ttoken"
+  spec.version       = Ttoken::VERSION
+  spec.authors       = ["Suraj Patil"]
+  spec.email         = ["patilsuraj767@gmail.com"]
+
+  spec.summary       = %q{ttoken is a command line utility to generate time based token/otp.}
+  spec.homepage      = "https://github.com/patilsuraj767/ttoken"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features|images)/}) }
+  end
+
+  spec.executables = ['ttoken']
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'rotp', '5.1.0'
+
+end

metadata

@@ -0,0 +1,70 @@
+--- !ruby/object:Gem::Specification
+name: ttoken
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Suraj Patil
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-06-13 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rotp
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 5.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 5.1.0
+description: 
+email:
+- patilsuraj767@gmail.com
+executables:
+- ttoken
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Gemfile.lock
+- README.md
+- bin/ttoken
+- config/ttoken.yml.example
+- lib/ttoken.rb
+- lib/ttoken/config.rb
+- lib/ttoken/encrypt.rb
+- lib/ttoken/store.rb
+- lib/ttoken/version.rb
+- ttoken.gemspec
+homepage: https://github.com/patilsuraj767/ttoken
+licenses: []
+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: []
+rubyforge_project: 
+rubygems_version: 2.7.6.2
+signing_key: 
+specification_version: 4
+summary: ttoken is a command line utility to generate time based token/otp.
+test_files: []