sleeproom 0.1.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 0154be3d205feeac7e626a12dce7c76e54253c9731d926e35aab1478e046f5f5
+  data.tar.gz: 93ed191055cbe4af6c73890bb50daadef4296857ba1f90f2b924547aefbfa92d
+SHA512:
+  metadata.gz: d9d8f2c0ddb0cad2c30c8bb9c3f9719eb9585c1895b0e114f97f11d534c52372925225a9b61dfbb89be05605e6e6637e801ba547417626dfde035b1b7ea94620
+  data.tar.gz: 5d1b0e70df00ae7e069f9af9a41c751445d8a9ab24f0223f573ab7e8e61064eb2ab82a91193ebf65c0e94face9e906254928b85335bd72ca6b37621fd2281a76

data/.gitignore

@@ -0,0 +1,15 @@
+/.bundle/
+/.yardoc
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+/.vscode/
+
+# rspec failure tracking
+.rspec_status
+
+# config
+/config/base.yml

data/.rspec

@@ -0,0 +1,3 @@
+--format documentation
+--color
+--require spec_helper

data/.rubocop.yml

@@ -0,0 +1,4 @@
+inherit_from: .rubocop_todo.yml
+inherit_gem:
+  rubocop-github:
+    - config/default_edge.yml

data/.rubocop_todo.yml

@@ -0,0 +1,2 @@
+StringLiterals:
+  Enable: False

data/Gemfile

@@ -0,0 +1,10 @@
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in sleeproom.gemspec
+gemspec
+
+gem "rake", "~> 12.0"
+gem "rspec", "~> 3.0"
+gem "rubocop", "~> 0.79.0"
+gem "rubocop-github"
+gem "rubocop-performance", require: false

data/Gemfile.lock

@@ -0,0 +1,126 @@
+PATH
+  remote: .
+  specs:
+    sleeproom (0.1.1)
+      async
+      async-http-faraday
+      async-websocket
+      colorize
+      configatron
+      terminal-table
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    activesupport (6.0.3.1)
+      concurrent-ruby (~> 1.0, >= 1.0.2)
+      i18n (>= 0.7, < 2)
+      minitest (~> 5.1)
+      tzinfo (~> 1.1)
+      zeitwerk (~> 2.2, >= 2.2.2)
+    ast (2.4.0)
+    async (1.26.1)
+      console (~> 1.0)
+      nio4r (~> 2.3)
+      timers (~> 4.1)
+    async-http (0.52.4)
+      async (~> 1.25)
+      async-io (~> 1.28)
+      async-pool (~> 0.2)
+      protocol-http (~> 0.20.0)
+      protocol-http1 (~> 0.13.0)
+      protocol-http2 (~> 0.14.0)
+    async-http-faraday (0.9.0)
+      async-http (~> 0.42)
+      faraday
+    async-io (1.30.0)
+      async (~> 1.14)
+    async-pool (0.3.2)
+      async (~> 1.25)
+    async-websocket (0.15.0)
+      async-http (~> 0.51)
+      async-io (~> 1.23)
+      protocol-websocket (~> 0.7.0)
+    colorize (0.8.1)
+    concurrent-ruby (1.1.6)
+    configatron (4.5.1)
+    console (1.8.2)
+    diff-lcs (1.3)
+    faraday (1.0.1)
+      multipart-post (>= 1.2, < 3)
+    i18n (1.8.3)
+      concurrent-ruby (~> 1.0)
+    jaro_winkler (1.5.4)
+    minitest (5.14.1)
+    multipart-post (2.1.1)
+    nio4r (2.5.2)
+    parallel (1.19.1)
+    parser (2.7.1.3)
+      ast (~> 2.4.0)
+    protocol-hpack (1.4.2)
+    protocol-http (0.20.0)
+    protocol-http1 (0.13.0)
+      protocol-http (~> 0.19)
+    protocol-http2 (0.14.0)
+      protocol-hpack (~> 1.4)
+      protocol-http (~> 0.18)
+    protocol-websocket (0.7.4)
+      protocol-http (~> 0.2)
+      protocol-http1 (~> 0.2)
+    rack (2.2.2)
+    rainbow (3.0.0)
+    rake (12.3.3)
+    rspec (3.9.0)
+      rspec-core (~> 3.9.0)
+      rspec-expectations (~> 3.9.0)
+      rspec-mocks (~> 3.9.0)
+    rspec-core (3.9.2)
+      rspec-support (~> 3.9.3)
+    rspec-expectations (3.9.2)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-mocks (3.9.1)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.9.0)
+    rspec-support (3.9.3)
+    rubocop (0.79.0)
+      jaro_winkler (~> 1.5.1)
+      parallel (~> 1.10)
+      parser (>= 2.7.0.1)
+      rainbow (>= 2.2.2, < 4.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 1.4.0, < 1.7)
+    rubocop-github (0.16.0)
+      rubocop (<= 0.82.0)
+      rubocop-performance (~> 1.0)
+      rubocop-rails (~> 2.0)
+    rubocop-performance (1.6.1)
+      rubocop (>= 0.71.0)
+    rubocop-rails (2.5.2)
+      activesupport
+      rack (>= 1.1)
+      rubocop (>= 0.72.0)
+    ruby-progressbar (1.10.1)
+    terminal-table (1.8.0)
+      unicode-display_width (~> 1.1, >= 1.1.1)
+    thread_safe (0.3.6)
+    timers (4.3.0)
+    tzinfo (1.2.7)
+      thread_safe (~> 0.1)
+    unicode-display_width (1.6.1)
+    zeitwerk (2.3.0)
+
+PLATFORMS
+  ruby
+  x64-mingw32
+
+DEPENDENCIES
+  rake (~> 12.0)
+  rspec (~> 3.0)
+  rubocop (~> 0.79.0)
+  rubocop-github
+  rubocop-performance
+  sleeproom!
+
+BUNDLED WITH
+   2.1.4

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2020 Koell
+
+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 @@
+# SLEEPROOM
+
+## Installation
+    $ gem install sleeproom
+
+## Dependency
+* [Minyami](https://github.com/Last-Order/Minyami)
+
+## Usage
+    $ ln -s sleeproom.service /etc/systemd/system/
+    $ sudo service sleeproom start
+    $ sleeproom status
+
+默认配置文件保存在 ~/.config/sleeproom/base.yml
+
+## TODO
+* [ ] 代理支持
+* [ ] Web GUI
+* [ ] 日志保存
+
+## Service example
+```
+[Unit]
+Description=SLEEPROOM
+After=network.target
+
+[Service]
+Type=simple
+User=root
+WorkingDirectory=[WorkingDirectory]
+ExecStart=[ExecStart]
+TimeoutSec=300
+Restart=always
+
+[Install]
+WantedBy=multi-user.target
+```
+
+## License
+The gem is available as open source under the terms of the [MIT License](LICENSE.txt).

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/sleeproom

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+require "sleeproom/cli"
+
+cli = SleepRoom::CLI.new(ARGV)
+cli.run

data/lib/sleeproom/async/websocket.rb

@@ -0,0 +1,20 @@
+require "async/websocket/connection"
+class WebSocketConnection < Async::WebSocket::Connection
+  def read
+    if buffer = super
+      parse(buffer)
+    end
+  end
+  
+  def write(object)
+    super(dump(object))
+  end
+  
+  def parse(buffer)
+    return buffer
+  end
+  
+  def dump(object)
+    return object
+  end
+end

data/lib/sleeproom/cli.rb

@@ -0,0 +1,78 @@
+# frozen_string_literal: true
+
+require "optparse"
+require "yaml"
+require "sleeproom/record"
+
+module SleepRoom
+  class CLI
+    # @param argv [Array]
+    def initialize(argv)
+      @options = {}
+      build
+      unless argv.empty?
+        @parser.parse!(argv)
+        action = argv.shift
+        if action == "status"
+          SleepRoom::Record::Tasks.status
+        elsif action == "start"
+          SleepRoom::Record::Tasks.start
+        elsif action == "download"
+          raise "未实现"
+        elsif action == "exit"
+          SleepRoom::Record::Tasks.stop
+        end
+        exit(0)
+      else
+        puts @parser
+        exit(0)
+      end
+    end
+
+    # @return [void]
+    def run
+      SleepRoom::Record::Tasks.start
+    end
+
+    # @return [void]
+    def build
+      @parser = OptionParser.new do |opt|
+        opt.version = "SleepRoom / #{SleepRoom::VERSION}"
+        opt.banner = "#{opt.version}"
+        opt.banner += "\nUsage: sleeproom [Options]\n\n"
+
+        opt.banner += "Action:\n"
+        opt.banner += "status".rjust(10)
+        opt.banner += "显示任务状态".rjust(33)
+        opt.banner += "\n"
+        opt.banner += "exit".rjust(8)
+        opt.banner += "关闭任务队列".rjust(35)
+        opt.banner += "\n\nCommands:\n"
+
+        opt.on("-a ROOM, NAME", "--add ROOM, GROUP", Array, "添加到监视列表") do |room|
+          SleepRoom::Record::Tasks.add(room[0].to_s, room[1].to_s)
+        end
+
+        opt.on("-r", "--remove [URL]", "从监视列表移除") do |room|
+          SleepRoom::Record::Tasks.remove(room)
+        end
+
+        opt.on("-v", "--verbose", "Print log") do
+          @options[:verbose] = true
+        end
+
+        opt.on("-d", "daemon", "Daemonize the script into the background") do
+          @options[:daemon] = true
+        end
+
+        opt.on_tail("--version", "Print version") do
+          STDOUT.puts(opt.version)
+        end
+
+        opt.on_tail("-h", "--help", "Print help") do
+          STDOUT.puts(opt)
+        end
+      end
+    end
+  end
+end

data/lib/sleeproom/record.rb

@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+require "json"
+require "configatron"
+require "sleeproom/utils"
+require "sleeproom/version"
+require "sleeproom/record/record"
+require "sleeproom/record/tasks"
+require "sleeproom/record/websocket"
+require "sleeproom/record/api/api"
+require "async"
+require "shellwords"
+module SleepRoom
+  module Record
+    # Okite!!!
+    # @param url [String]
+    # @return [Boolean]
+    def self.call_minyami(url:, is_live: true, threads: configatron.minyami.threads, output:, retries: configatron.minyami.retries)
+      command = "minyami -d #{Shellwords.escape(url)}"
+      command += " --retries #{retries.to_i}" if retries
+      command += " --threads #{threads.to_i}" if threads
+      command += " --live" if is_live
+      output = File.join(configatron.save_path, output)
+      command += " --output #{Shellwords.escape(output)}" if output
+      download_dir_check(output)
+      pid = exec_command(command, output)
+      return pid
+    end
+
+    # @param command [String]
+    # @return [Boolean]
+    def self.exec_command(command, output)
+      SleepRoom.info("Call command: #{command}")
+      SleepRoom.info("STDOUT: #{output}.out , STDERR: #{output}.err")
+      pid = spawn(command, out: "#{output}.out", err: "#{output}.err")
+      SleepRoom.info("PID: #{pid}")
+      Process.detach(pid)
+      return pid
+    end
+
+    def self.download_dir_check(output)
+      dir = File.dirname(output)
+      if !Dir.exist?(dir)
+        SleepRoom.info("#{dir} does not exist, creating...")
+        SleepRoom.mkdir(dir)
+      end
+    end
+  end
+end

data/lib/sleeproom/record/api/api.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "async/http/faraday"
+require "async/http/faraday/default"
+require_relative "room"
+require_relative "room_api"
+require_relative "streaming_api"
+
+module SleepRoom
+  module Record
+    module API
+      class Error < StandardError; end
+      ROOM_URL = "https://www.showroom-live.com"
+      ROOM_API = "https://www.showroom-live.com/api/room/status"
+      STREAMING_API = "https://www.showroom-live.com/api/live/streaming_url"
+
+      USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.61 Safari/537.36"
+
+      def self.get(url)
+        Async do
+          http = Faraday.get(url, nil, {"User-Agent": USER_AGENT})
+          if http.status == 200
+            @json = JSON.parse(http.body)
+          else
+            raise Error, "HTTP Error: #{http.status}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/sleeproom/record/api/room.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module SleepRoom
+  module Record
+    module API
+      class Room
+        def initialize(room_name)
+          @url = ROOM_URL + "/" + room_name
+        end
+
+        def get(task: Async::Task.current)
+          @json = API.get(@url).wait
+        end
+      end
+    end
+  end
+end