stockroll 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 3dc6678fdbb06fe8356c3f8b9331ef29adb58f61
+  data.tar.gz: b692ef3c18b9f405a536f7096afac406de22eca1
+SHA512:
+  metadata.gz: 251b4bfec0947e0c39b1859ff8ec078f1d5d74375fa0730aa308d8ad9f86d11c0372c3a783288faacec208d4e4fff7bd375ec014e7748c80590b888b1ef71a77
+  data.tar.gz: c54864d4be3db21c4cdc5549e82c2cbd6a21aeb6269b35834ff6c5e4d31cd03ff04174c619e2a063adeaddf546b5fd71a7dacd3ba844f77d10a639ea84d57505

data/.gitignore

@@ -0,0 +1,19 @@
+# See http://help.github.com/ignore-files/ for more about ignoring files.
+#
+# If you find yourself ignoring temporary files generated by your text editor
+# or operating system, you probably want to add a global ignore instead:
+#   git config --global core.excludesfile ~/.gitignore_global
+
+coverage
+doc
+pkg
+*.swp
+*.patch
+*.gem
+*.sh
+.DS_Store
+.bundle/*
+#Gemfile.lock
+tmp/*
+log/*
+.rvmrc

data/Gemfile

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

data/Gemfile.lock

@@ -0,0 +1,46 @@
+PATH
+  remote: .
+  specs:
+    stockroll (0.0.2)
+      colorize (~> 0.6.0)
+      curses (~> 1.0.0)
+      em-http-request (~> 1.1.2)
+      thor (~> 0.18.1)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    addressable (2.3.5)
+    colorize (0.6.0)
+    cookiejar (0.3.1)
+    curses (1.0.0)
+    diff-lcs (1.2.5)
+    em-http-request (1.1.2)
+      addressable (>= 2.3.4)
+      cookiejar
+      em-socksify (>= 0.3)
+      eventmachine (>= 1.0.3)
+      http_parser.rb (>= 0.6.0)
+    em-socksify (0.3.0)
+      eventmachine (>= 1.0.0.beta.4)
+    eventmachine (1.0.3)
+    http_parser.rb (0.6.0)
+    rake (10.1.1)
+    rspec (2.14.1)
+      rspec-core (~> 2.14.0)
+      rspec-expectations (~> 2.14.0)
+      rspec-mocks (~> 2.14.0)
+    rspec-core (2.14.7)
+    rspec-expectations (2.14.5)
+      diff-lcs (>= 1.1.3, < 2.0)
+    rspec-mocks (2.14.6)
+    thor (0.18.1)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bundler (~> 1.3)
+  rake
+  rspec
+  stockroll!

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2014 Raymond Jiang
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,40 @@
+# Stockroll
+### To show real-time stock quotes
+
+![screenshot](/screenshot.png)
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'stockroll'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install stockroll
+
+## Usage
+
+	1. Very simple to use, only type 'stock' in your console and let's fun.  
+
+	2. Type 'stock --help' to view all commands.     
+
+	3. Type 'stock add your_stock_code_xxx' , add stock code to display.   
+
+	4. Type 'stock delete your_stock_code_xxx', delete a existed stock code.  
+
+	5. Type 'stock clear' to clear all existed stock codes.  
+
+	6. Type 'q' or 'Q' to quit stock.   
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1 @@
+require "bundler/gem_tasks"

data/bin/stock

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require 'stockroll/commands'
+
+Stockroll::Commands.start

data/lib/stockroll.rb

@@ -0,0 +1,5 @@
+require "stockroll/version"
+
+module Stockroll
+  # Your code goes here...
+end

data/lib/stockroll/commands.rb

@@ -0,0 +1,126 @@
+require "thor"
+require 'find'
+require 'erb'
+require 'colorize'
+require 'stockroll/utils'
+
+module Stockroll 
+	class Commands < Thor
+		include Thor::Actions
+		CONF_FILE_PATH = File.expand_path("~/.stock_roll_codes_config",__FILE__)
+
+		desc "start", "start stock roll"
+		def start
+			codes = self.get_stock_codes
+			if codes.nil? or codes.strip.size == 0
+				isOk, cs = self.init_config
+				Stockroll::Utils.run(cs) if isOk
+			else
+				Stockroll::Utils.run(codes)
+			end
+		end
+
+		desc "add", "add a stock code, if you want add multiple CODE please separated by commas."
+		def add(codes)
+			is_valid , err_msg = self.check_codes(codes)
+			if is_valid
+				self.add_codes(codes)
+				puts "Info: add code successfully.".yellow
+			else
+				err_msg.each do |msg|
+					puts msg.red
+				end
+			end
+		end
+
+		desc "delete", "delete a stock code"
+		def delete(code)
+			self.remove_code(code)
+			puts "Info: delete is ok.".yellow
+		end
+
+		desc "clear", "clear all stock codes"
+		def clear()
+			self.remove_all_codes()
+			puts "Info: clear all codes is successed.".yellow
+		end
+
+		default_task :start
+
+		protected
+		def get_stock_codes
+			return nil unless File.exists?(CONF_FILE_PATH)
+			return IO.read(CONF_FILE_PATH)
+		end
+		def add_codes(codes)
+			raise 'Please input stock codes.' if codes.nil? or codes.strip.size == 0
+			codes = codes.gsub(/(^\s+)|(\s+$)/,'')
+			unless File.exists?(CONF_FILE_PATH)
+				File.open(CONF_FILE_PATH,'w') do |f|
+					f.write(codes)
+				end
+			else
+				content = self.get_stock_codes
+				if content.nil? or content.strip.size == 0
+					File.open(CONF_FILE_PATH,'w') do |f|
+						f.write(codes)
+					end
+				else
+					old_codes = content.split(',')
+					added_codes = codes.split(',')
+					new_codes = old_codes.concat(added_codes).uniq
+					File.open(CONF_FILE_PATH,'w') do |f|
+						f.write(new_codes.join(','))
+					end
+				end
+				
+			end
+		end
+		def remove_code(code)
+			old_codes = self.get_stock_codes
+			unless old_codes.nil?
+				n = []
+				old_codes.split(',').each do |c|
+					if c.strip != code.strip
+						n << c
+					end
+				end
+				File.open(CONF_FILE_PATH,'w') do |f|
+					f.write(n.join(','))
+				end
+			end
+		end
+		def remove_all_codes()
+			File.open(CONF_FILE_PATH,'w') do |f|
+				f.write('')
+			end
+		end
+		def check_codes(codes)
+			is_valid = true
+			err_msg = []
+			c_a = codes.split(',')
+			c_a.each do |c|
+				unless Stockroll::Utils.is_valid_stock_code(c)
+					err_msg << "Error: invalid stock code: #{c}"
+					is_valid = false
+				end
+			end
+			[is_valid, err_msg]
+		end
+		def init_config
+			print "Please input your stock codes: ".yellow
+			cs = gets.chomp
+			is_valid, err_msg = self.check_codes(cs)
+			unless is_valid
+				err_msg.each do |msg|
+					puts msg.red
+				end
+				[false, '']
+			else
+				self.add_codes(cs)
+				[true, cs]
+			end
+		end
+	end
+
+end

data/lib/stockroll/utils.rb

@@ -0,0 +1,144 @@
+
+require 'net/http'
+require 'colorize'
+require 'em-http-request'
+require "curses"
+
+include Curses
+
+module Stockroll 
+	class Utils
+		ROOT_URL = "http://hq.sinajs.cn/list="
+		class << self
+			def is_valid_stock_code(code)
+				uri = URI("#{ROOT_URL}#{code}")
+				res = Net::HTTP.get(uri).force_encoding("GBK").encode("UTF-8")
+				val = res.split('=')[1]
+				if val =~ /([^\"\n;]+)/
+					if !$1.nil? and $1.strip.size > 0
+						true
+					else
+						false
+					end
+				else
+					false
+				end
+			end
+			def run(codes)
+				codes = codes.split(',')	
+				Curses.noecho
+				Curses.init_screen
+				Curses.start_color
+				Curses.init_pair(COLOR_RED,COLOR_RED,COLOR_BLACK)
+				Curses.init_pair(COLOR_BLUE,COLOR_BLUE,COLOR_BLACK)
+				Curses.init_pair(COLOR_GREEN,COLOR_GREEN,COLOR_BLACK)
+				Curses.init_pair(COLOR_YELLOW,COLOR_YELLOW,COLOR_BLACK)
+				47.times do |i|
+					Curses.setpos(0,i)
+					Curses.addstr('#')
+				end
+				curs_set(0)
+				Curses.setpos(1,0)
+				Curses.addstr('Loading...')
+
+				another_th = Thread.new {
+					EventMachine.run do
+
+						do_work = proc{
+
+							codes.each_with_index do |code,index| 
+								http = EventMachine::HttpRequest.new("#{ROOT_URL}#{code}").get
+								http.errback { p 'Uh oh'; }
+								http.callback {
+									res = http.response.encode('utf-8')
+									val = res.split('=')[1]
+									if val =~ /([^\"\n;]+)/
+										captures = $1
+										if index == 0
+											ypos = 1
+										else
+											ypos = (index * 2) + 1
+										end
+										
+										if captures.strip != ''
+											ar = captures.split(',')
+											Curses.setpos(ypos,0)
+											Curses.attron(A_BOLD|A_UNDERLINE) {
+												Curses.addstr(ar[0])
+												clrtoeol() #clear line	
+											}
+
+											Curses.setpos(ypos,11)
+											Curses.attron(A_NORMAL|A_DIM) {
+												Curses.addstr(code)
+											}
+
+											Curses.setpos(ypos,22)
+											Curses.attron(A_NORMAL) {
+												Curses.addstr(ar[3])
+											}
+
+											Curses.setpos(ypos,32)
+											if ar[3].to_f == 0
+												diff = 0.00
+											else
+												diff = (ar[3].to_f - ar[2].to_f).round(2)
+											end
+											prefix = ' '
+											if diff >= 0
+												if diff == 0
+													Curses.attron(A_BOLD) {
+														Curses.addstr("#{prefix}#{diff.abs}")
+														Curses.addstr("   ")
+														Curses.addstr("0.00%")
+														
+													}
+												else
+													prefix = '+'
+													Curses.attron(color_pair(COLOR_RED)|A_BOLD) {
+														Curses.addstr("#{prefix}#{diff.abs}")
+														Curses.addstr("   ")
+														Curses.addstr("#{prefix}#{((diff.abs/ar[2].to_f)*100).round(2)}%")
+														
+													}
+												end
+											else
+												prefix = '-'
+												Curses.attron(color_pair(COLOR_GREEN)|A_BOLD) {
+														Curses.addstr("#{prefix}#{diff.abs}")
+														Curses.addstr("   ")
+														Curses.addstr("#{prefix}#{((diff.abs/ar[2].to_f)*100).round(2)}%")
+												}
+											end
+											Curses.refresh
+										end
+										
+									end
+
+								}
+							end
+						}
+
+						EM::PeriodicTimer.new(10) do
+							do_work.call()
+						end
+						do_work.call()
+
+					end
+
+				}
+				while TRUE
+					c = getch
+					case c
+						when /q/i
+							EM.stop if EM
+							another_th.exit()
+							Curses.close_screen
+							exit(0)
+
+					end
+				end
+			end
+		end
+	end
+end

data/lib/stockroll/version.rb

@@ -0,0 +1,3 @@
+module Stockroll
+  VERSION = "0.0.2"
+end

data/spec/spec_helper.rb

@@ -0,0 +1,18 @@
+require 'rubygems'
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+
+require "stockroll"
+
+
+RSpec.configure do |config|
+  # Use color in STDOUT
+  config.color_enabled = true
+
+  # Use color not only in STDOUT but also in pagers and files
+  config.tty = true
+
+  # Use the specified formatter
+  config.formatter = :documentation # :progress, :html, :textmate
+end

data/spec/stock_spec.rb

@@ -0,0 +1,36 @@
+# run rspec 
+require "spec_helper"
+require 'stockroll/commands'
+require 'stockroll/utils'
+
+describe Stockroll do
+
+	it 'stock code is valid' do 
+		Stockroll::Utils.is_valid_stock_code('shabc').should == false
+		Stockroll::Utils.is_valid_stock_code('sz000100').should == true
+	end
+	describe "Stock Config Operation" do
+		path = Stockroll::Commands::CONF_FILE_PATH
+		before(:each) do
+			File.open(path,'w'){|f| f.write('')} if File.exists?(path)
+		end
+		it 'read and write stock code is fine' do
+			cmd = Stockroll::Commands.new
+			cmd.send :add_codes, "sz001"
+			cmd.send :add_codes, "sz001,sz002"
+			cmd.send :add_codes, "sz003"
+			(cmd.send :get_stock_codes).should == "sz001,sz002,sz003"
+		end
+		it 'remove stock code is fine' do
+			cmd = Stockroll::Commands.new
+			cmd.send :add_codes, "sz001"
+			cmd.send :add_codes, "sz001,sz002"
+			cmd.send :remove_code, "sz002"
+			(cmd.send :get_stock_codes).should == "sz001"
+		end
+		after(:each) do
+			File.open(path,'w'){|f| f.write('')} if File.exists?(path)
+		end
+	end
+
+end

data/stockroll.gemspec

@@ -0,0 +1,28 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'stockroll/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "stockroll"
+  spec.version       = Stockroll::VERSION
+  spec.authors       = ["Robot Jiang"]
+  spec.email         = ["robot.z.jiang@gmail.com"]
+  spec.description   = %q{To show real-time stock quotes.}
+  spec.summary       = %q{To show real-time stock quotes.}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = ['stock']#spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+	spec.add_development_dependency "rspec"
+	spec.add_dependency "thor" , "~> 0.18.1"
+	spec.add_dependency "colorize" , "~> 0.6.0"
+	spec.add_dependency "em-http-request" , "~> 1.1.2"
+	spec.add_dependency "curses" , "~> 1.0.0"
+end

metadata

@@ -0,0 +1,160 @@
+--- !ruby/object:Gem::Specification
+name: stockroll
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Robot Jiang
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-02-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.18.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.18.1
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.6.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.6.0
+- !ruby/object:Gem::Dependency
+  name: em-http-request
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.1.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.1.2
+- !ruby/object:Gem::Dependency
+  name: curses
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+description: To show real-time stock quotes.
+email:
+- robot.z.jiang@gmail.com
+executables:
+- stock
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/stock
+- lib/stockroll.rb
+- lib/stockroll/commands.rb
+- lib/stockroll/utils.rb
+- lib/stockroll/version.rb
+- screenshot.png
+- spec/spec_helper.rb
+- spec/stock_spec.rb
+- stockroll.gemspec
+homepage: ''
+licenses:
+- MIT
+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.1.11
+signing_key: 
+specification_version: 4
+summary: To show real-time stock quotes.
+test_files:
+- spec/spec_helper.rb
+- spec/stock_spec.rb