mochizuki 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 9267c9d3ffaa34e028901b0c0492651c7df87425dc1a23ce1223ae3c11a9c294
+  data.tar.gz: 2271c29c305f2de09d0ab6258028c694970d05111dad001fa91e3c9d5e71f45b
+SHA512:
+  metadata.gz: 476cdfc1d02ccef3802e88559747af2357d91c8aa81473a35c46b63d291554706f7a617c778f364c469afeecb994a87351ff5c21f403dcb8c94f41debe8f5804
+  data.tar.gz: b35815bcfab76a140026ac1ef026d101b51ca498c7b09c86e399058b354faf5aff0355b6bce24b3bb8060c04b844bafa60599d8cd8e6b9c16ff0b15a22649e9f

data/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2020 Kowalski Dark
+
+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,59 @@
+# Mochizuki
+
+同济大学能源管理中心电费自动查询 telegram bot
+
+Manually/auto query and send alarms to channel when the power gets below your threshold.
+
+![3](./screenshots/3.jpg)
+![4](./screenshots/4.jpg)
+
+## Installation
+
+### From RubyGems
+
+```
+gem install mochizuki
+```
+
+### Using docker
+
+**Recommended**
+
+See [mochizuki-docker](https://github.com/DarkKowalski/mochizuki-docker)
+
+## Configuration
+
+Place a `mochizuki.conf` in your directory and run `mochizuki`
+
+```ruby
+Mochizuki.configure do |config|
+  config.bot_token = 'your_bot_token'
+  config.channel = '@your_channel'
+  config.query_interval = '300s'
+  config.alarm_threshold = '60' # kWh
+
+  config.campus = '1'
+  config.building = '1'
+  config.floor = '1'
+  config.dorm = '1'
+end
+```
+
+**How to get the last four attributes**
+
+Go to [同济大学能源管理中心](http://202.120.163.129:88) and fill the form
+
+![1](./screenshots/1.png)
+
+Open your dev tools and look at the bottom on the right
+
+![2](./screenshots/2.png)
+
+Those shitty words mean
+
+```
+drlouming  ->  campus
+drceng     ->  building
+dr_ceng    ->  floor
+drfangjian ->  dorm
+```

data/bin/mochizuki

@@ -0,0 +1,7 @@
+#!/bin/ruby
+# frozen_string_literal: true
+
+require'mochizuki'
+
+load('mochizuki.conf')
+Mochizuki::Bot.new.start

data/lib/mochizuki.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require_relative 'mochizuki/error'
+
+require_relative 'mochizuki/version'
+require_relative 'mochizuki/logging'
+require_relative 'mochizuki/configurable'
+
+require_relative 'mochizuki/auto_query'
+require_relative 'mochizuki/fetcher'
+require_relative 'mochizuki/status'
+require_relative 'mochizuki/telegram_bot'

data/lib/mochizuki/auto_query.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require 'rufus-scheduler'
+
+module Mochizuki
+  class AutoQuery
+    def initialize(logger = Mochizuki.logger, config = Mochizuki.config)
+      @logger = logger
+      @config = config
+
+      Mochizuki::Fetcher.new.fetch # update Mochizuki.status
+    end
+
+    def alarm
+      scheduler = Rufus::Scheduler.new
+      scheduler.every @config.query_interval.to_s do
+        power = Mochizuki::Fetcher.new.fetch
+        @logger.info "Auto query, #{power} kWh remaining"
+        if Mochizuki.status.auto_alarm_enabled?
+          yield(power)
+          Mochizuki.status.alarmed_before = true
+        end
+      end
+    end
+  end
+end

data/lib/mochizuki/configurable.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+module Mochizuki
+  module Configurable
+    def self.included(base)
+      base.extend ClassMethods
+    end
+
+    module ClassMethods
+      def config
+        @config ||= Configuration.new
+      end
+
+      def reset_config
+        @config = Configuration.new
+      end
+
+      def configure
+        yield(config)
+      end
+    end
+
+    class Configuration
+      attr_accessor :bot_token, :channel,
+                    :api_host, :api_port,
+                    :query_interval, :alarm_threshold,
+                    :campus, :building, :floor, :dorm
+
+      def initialize
+        @bot_token = nil
+        @channel = nil
+
+        @api_host = '202.120.163.129' # https://nyglzx.tongji.edu.cn
+        @api_port = '88'
+
+        @query_interval = '300s'
+        @alarm_threshold = '60' # kWh
+
+        # shitty names from legacy code
+        @campus = nil   # -> drlouming
+        @building = nil # -> drceng
+        @floor = nil    # -> dr_ceng
+        @dorm = nil     # -> drfangjian
+      end
+    end
+  end
+end
+
+module Mochizuki
+  include Mochizuki::Configurable
+end

data/lib/mochizuki/error.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+module Mochizuki
+  class Error < StandardError
+  end
+end

data/lib/mochizuki/fetcher.rb

@@ -0,0 +1,81 @@
+# frozen_string_literal: true
+
+require 'faraday'
+require 'uri'
+require 'nokogiri'
+
+module Mochizuki
+  class Fetcher
+    def initialize(logger = Mochizuki.logger, config = Mochizuki.config)
+      @logger = logger
+      @config = config
+
+      @uri = URI("http://#{@config.api_host}:#{config.api_port}")
+
+      @request_body = {
+        '__EVENTTARGET' => '',
+        '__EVENTARGUMENT' => '',
+        '__LASTFOCUS' => '',
+        '__VIEWSTATE' => '',
+        '__VIEWSTATEGENERATOR' => 'CA0B0334',
+        'drlouming' => '',
+        'drceng' => '',
+        'dr_ceng' => '',
+        'drfangjian' => '',
+        'radio' => 'usedR',
+        'ImageButton1.x' => '0',
+        'ImageButton1.y' => '0'
+      }
+
+      @cookie = nil
+    end
+
+    def fetch
+      fetch_viewstate
+      fetch_cookie
+      power = fetch_power
+
+      if power.nil?
+        @logger.warn 'Failed to query.'
+        return
+      end
+
+      Mochizuki.update_status(power)
+      power
+    end
+
+    private
+
+    def fetch_viewstate
+      @request_body['drlouming'] = @config.campus
+      @request_body['drceng'] = @config.building
+      @request_body['dr_ceng'] = @config.floor
+      @request_body['drfangjian'] = @config.dorm
+
+      3.times do
+        resp = Faraday.post(@uri) do |req|
+          req.headers = { 'Content-Type' => 'application/x-www-form-urlencoded' }
+          req.body = URI.encode_www_form(@request_body)
+        end
+        html = Nokogiri::HTML(resp.body)
+        @request_body['__VIEWSTATE'] = html.at_css('input#__VIEWSTATE')['value']
+      end
+    end
+
+    def fetch_cookie
+      resp = Faraday.post(@uri) do |req|
+        req.headers = { 'Content-Type' => 'application/x-www-form-urlencoded' }
+        req.body = URI.encode_www_form(@request_body)
+      end
+      @cookie = resp.headers['set-cookie'].split('; ')[0]
+    end
+
+    def fetch_power
+      resp = Faraday.get("#{@uri}/usedRecord1.aspx") do |req|
+        req.headers = { 'Cookie' => @cookie, 'Content-Type' => 'application/x-www-form-urlencoded' }
+      end
+      html = Nokogiri::HTML(resp.body)
+      html.xpath('//h6').text.scan(/(\d+[,.]\d+)/)[0][0]
+    end
+  end
+end

data/lib/mochizuki/logging.rb

@@ -0,0 +1,21 @@
+# frozen_string_literal: true
+
+require 'logger'
+
+module Mochizuki
+  module Logging
+    def self.included(base)
+      base.extend ClassMethods
+    end
+
+    module ClassMethods
+      def logger
+        @logger ||= Logger.new($stdout)
+      end
+    end
+  end
+end
+
+module Mochizuki
+  include Logging
+end

data/lib/mochizuki/status.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+module Mochizuki
+  module Status
+    def self.included(base)
+      base.extend ClassMethods
+    end
+
+    module ClassMethods
+      def status
+        @status ||= Status.new
+      end
+
+      def reset_status
+        @status = Status.new
+      end
+
+      def update_status(power)
+        last = Mochizuki.status.below_threshold
+        current = power.to_f < @config.alarm_threshold.to_f
+        @logger.info "Status, @below_threshold, last: #{last}, current: #{current}"
+
+        Mochizuki.status.below_threshold = current
+
+        return unless !current && last # re-enabled auto alarm
+
+        Mochizuki.status.alarmed_before = false
+        @logger.info "Status, @alarmed_before, #{@alarmed_before}, auto alarm enabled"
+      end
+    end
+
+    class Status
+      attr_accessor :below_threshold, :alarmed_before
+
+      def initialize(logger = Mochizuki.logger)
+        @logger = logger
+        @below_threshold = nil
+        @alarmed_before = false
+      end
+
+      def auto_alarm_enabled?
+        if @below_threshold.nil?
+          @logger.error "Unable to check status, @below_threshold can't be nil"
+          raise Mochizuki::Error, 'Invalid @below_threshold'
+        end
+
+        @below_threshold && !@alarmed_before
+      end
+    end
+  end
+end
+
+module Mochizuki
+  include Status
+end

data/lib/mochizuki/telegram_bot.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require 'telegram/bot'
+
+module Mochizuki
+  class Bot
+    def initialize(logger = Mochizuki.logger, config = Mochizuki.config)
+      @config = config
+      @logger = logger
+    end
+
+    def start
+      Telegram::Bot::Client.run(@config.bot_token) do |bot|
+        Mochizuki::AutoQuery.new.alarm do |power|
+          bot.api.send_message(chat_id: @config.channel,
+                               text: "Dorm: #{@config.dorm}: #{power} kWh, " \
+                               "lower than threshold #{@config.alarm_threshold}")
+        end
+        @logger.info 'Auto query enabled'
+
+        bot.listen do |msg|
+          case msg.text
+          when '/start'
+            bot.api.send_message(chat_id: msg.chat.id,
+                                 text: 'I am a telegram bot written in ruby, '\
+                                       'details at https://github.com/DarkKowalski/mochizuki-bot')
+            @logger.info "ID: #{msg.chat.id} uses /start"
+          when '/query'
+            dorm = @config.dorm
+            begin
+              power = Mochizuki::Fetcher.new.fetch
+              send_msg = "Dorm #{dorm}: #{power} kWh"
+            rescue StandardError
+              send_msg = 'Failed to query.'
+            end
+            bot.api.send_message(chat_id: msg.chat.id, text: send_msg)
+          else
+            bot.api.send_message(chat_id: msg.chat.id, text: "Use '/query' to get remaining power")
+            @logger.warn "ID: #{msg.chat.id} uses #{msg.text}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/mochizuki/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Mochizuki
+  VERSION = '0.0.2'
+end

metadata

@@ -0,0 +1,155 @@
+--- !ruby/object:Gem::Specification
+name: mochizuki
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Kowalski Dark
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-09-19 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.1
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.10.10
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.10.10
+- !ruby/object:Gem::Dependency
+  name: rufus-scheduler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.6.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.6.0
+- !ruby/object:Gem::Dependency
+  name: telegram-bot-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.10.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.10.1
+- !ruby/object:Gem::Dependency
+  name: ci_reporter_minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.14.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.14.1
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 13.0.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 13.0.1
+description: Get the amount of electricity remaining for dorms from Tongxinxun
+email:
+- DarkKowalski2012@gmail.com
+executables:
+- mochizuki
+extensions: []
+extra_rdoc_files: []
+files:
+- LICENSE
+- README.md
+- bin/mochizuki
+- lib/mochizuki.rb
+- lib/mochizuki/auto_query.rb
+- lib/mochizuki/configurable.rb
+- lib/mochizuki/error.rb
+- lib/mochizuki/fetcher.rb
+- lib/mochizuki/logging.rb
+- lib/mochizuki/status.rb
+- lib/mochizuki/telegram_bot.rb
+- lib/mochizuki/version.rb
+homepage: https://github.com/darkkowalski/mochizuki-bot
+licenses:
+- MIT
+metadata:
+  bug_tracker_uri: https://github.com/darkkowalski/mochizuki-bot/issues
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.7.1
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.2
+signing_key: 
+specification_version: 4
+summary: Tongxinyun Electircity Remaining
+test_files: []