sleeproom 0.1.1

data/lib/sleeproom/record/write_status.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+require "async/queue"
+module SleepRoom
+  module Record
+    class WriteStatus
+      attr_accessor :queue
+      def initialize
+        @queue = Async::Queue.new
+      end
+
+      def run
+        Async do
+          while status = @queue.dequeue
+            status[:update] = Time.now
+            old_status = SleepRoom.load_config(:status)
+            room = status[:room]
+            if old_status.find { |h| h[:room] == room }
+              new_status = old_status.delete_if { |h| h[:room] == room }
+              new_status.push(status)
+            else
+              new_status = old_status.push(status)
+            end
+            SleepRoom.write_config_file(:status, new_status)
+          end
+        end
+      end
+
+      def add(status)
+        Async do
+          @queue.enqueue(status)
+        end
+      end
+    end
+  end
+end

data/lib/sleeproom/utils.rb

@@ -0,0 +1,214 @@
+# frozen_string_literal: true
+
+require "configatron"
+require "colorize"
+require "fileutils"
+require "yaml"
+
+module SleepRoom
+  class Error < StandardError; end
+  # @return [String]
+  def self.root_path
+    Dir.pwd
+  end
+
+  # @return [String]
+  def self.user_path
+    Dir.home
+  end
+
+  # @return [String]
+  def self.sleeproom_dir
+    File.join(user_path, "sleeproom")
+  end
+
+  def self.working_dir
+    Dir.pwd
+  end
+
+  # @return [String]
+  def self.config_dir
+    File.join(user_path, ".config", "sleeproom")
+  end
+
+  # @param filename [String]
+  # @return [String]
+  def self.config_path(config)
+    name = {
+      status: "status.yml",
+      base: "base.yml",
+      record: "record.yml",
+      pid: "tmp/pid.yml"
+    }
+    file = name[config].to_s
+    raise Error if file.empty?
+    return File.join(config_dir, file) if config == :base
+
+    if load_config(:base)[:config_path] == "USER_CONFIG"
+      File.join(config_dir, file)
+    else
+      File.join(load_config(:base)[:config_path], file)
+    end
+  end
+
+  # @param filename [Symbol]
+  # @param settings [Hash]
+  # @return [Boolean]
+  def self.write_config_file(config, settings)
+    file = File.new(config_path(config), "w")
+    file.puts(YAML.dump(settings))
+    file.close
+  end
+
+  def self.create_config_file(config, settings)
+    path = config_path(config)
+    return false if File.exist?(path)
+    mkdir(File.dirname(path)) unless Dir.exist?(File.dirname(path))
+    write_config_file(config, settings)
+  end
+
+  def self.load_config(config)
+    raise Error if config.empty? || !File.exist?(config_path(config))
+
+    YAML.load_file(config_path(config))
+  rescue Error => e
+    init_config
+    retry
+  end
+
+  def self.mkdir(path)
+    FileUtils.mkdir_p(path) unless Dir.exist?(path)
+  end
+
+  def self.init_base
+    base = {
+      web: {
+        use: true,
+        server: "localhost",
+        port: 3000
+      },
+      proxy: {
+        use: false,
+        server: "localhost",
+        port: 8080,
+        type: "socks5"
+      },
+      record: {
+        all: true,
+        wanted: [],
+        unwanted: []
+      },
+      config_path: "USER_CONFIG",
+      save_path: "#{sleeproom_dir}/archive",
+      working_path: "#{sleeproom_dir}/working",
+      default_save_name: "%ROOMNAME%-%TIME%.ts",
+      minyami: {
+        threads: 8,
+        retries: 999,
+      },
+      logger: {
+        console: true,
+        file: {
+          use: true,
+          path: "#{sleeproom_dir}/log"
+        },
+        websocket: {
+          log: true,
+          ping_log: false
+        }
+      }
+    }
+    create_config_file(:base, base)
+  end
+
+  def self.init_config
+    mkdir(config_dir) unless Dir.exist?(config_dir)
+
+    mkdir("#{config_dir}/tmp") unless Dir.exist?("#{config_dir}/tmp")
+    
+    init_base
+    
+    record = {
+      "default" => []
+    }
+
+    create_config_file(:record, record)
+    create_config_file(:status, [])
+    write_config_file(:pid, nil)
+  end
+
+  # @return [Boolean]
+  def self.reload_config
+    configs = %i[base]
+    configs.each do |config|
+      configatron.configure_from_hash(YAML.load_file(config_path(config)))
+    end
+    true
+  rescue Errno::ENOENT => e
+    error("Could not load file. \n#{e.message}")
+    false
+  end
+
+  def self.settings
+    configatron
+  end
+  def self.load_status
+    SleepRoom.load_config(:status)
+  rescue Error
+    create_status
+    retry
+  end
+
+  def self.load_pid
+    SleepRoom.load_config(:pid)
+  rescue Error
+    create_pid(nil)
+    retry
+  end
+
+  def self.running?(pid=nil)
+    pid = SleepRoom.load_config(:pid) if pid.nil?
+    Process.getpgid(pid)
+    true
+  rescue
+    false
+  end
+
+  def self.load_record
+    SleepRoom.load_config(:record)
+  rescue Error
+    create_record
+    retry
+  end
+
+  def self.create_status(status = [])
+    SleepRoom.create_config_file(:status, status)
+  end
+
+  def self.create_record(record = {default: []})
+    SleepRoom.create_config_file(:record, record)
+  end
+
+  def self.create_pid(pid)
+    SleepRoom.create_config_file(:pid, pid)
+    SleepRoom.write_config_file(:pid, pid)
+  end
+  
+  # @param string [String]
+  # @return [nil]
+  def self.info(string)
+    STDOUT.puts("[INFO] #{string}".colorize(:white))
+  end
+
+  # @param string [String]
+  # @return [nil]
+  def self.warning(string)
+    warn("[WARN] #{string}".colorize(:yellow))
+  end
+
+  # @param string [String]
+  # @return [nil]
+  def self.error(string)
+    STDOUT.puts("[ERROR] #{string}".colorize(:red))
+  end
+end

data/lib/sleeproom/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module SleepRoom
+  VERSION = "0.1.1"
+end

data/sleeproom.gemspec

@@ -0,0 +1,41 @@
+# frozen_string_literal: true
+
+require_relative "lib/sleeproom/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "sleeproom"
+  spec.version       = SleepRoom::VERSION
+  spec.authors       = ["Koell"]
+  spec.email         = ["i@wug.moe"]
+
+  spec.summary       = "sleeproom"
+  spec.homepage      = "https://github.com/KoellM/sleeproom"
+  spec.license       = "MIT"
+  spec.required_ruby_version = Gem::Requirement.new(">= 2.3.0")
+
+  spec.metadata["homepage_uri"] = spec.homepage
+  # spec.metadata["source_code_uri"] = "TODO: Put your gem's public repo URL here."
+
+  # 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(__dir__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "bin"
+  spec.executables   = ["sleeproom"]
+  spec.require_paths = ["lib"]
+
+  spec.add_runtime_dependency("colorize")
+  spec.add_runtime_dependency("async")
+  spec.add_runtime_dependency("async-websocket")
+  spec.add_runtime_dependency("configatron")
+  spec.add_runtime_dependency("async-http-faraday")
+  spec.add_runtime_dependency("terminal-table")
+
+  spec.post_install_message = <<~STR
+    SleepRoom 需要:
+      [Minyami]       https://github.com/Last-Order/minyami
+      (npm install -g minyami)
+    使用前请确保已安装上述依赖。
+  STR
+end

metadata

@@ -0,0 +1,157 @@
+--- !ruby/object:Gem::Specification
+name: sleeproom
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- Koell
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2020-06-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: colorize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: async
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: async-websocket
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: configatron
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: async-http-faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: terminal-table
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email:
+- i@wug.moe
+executables:
+- sleeproom
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".rubocop.yml"
+- ".rubocop_todo.yml"
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/sleeproom
+- lib/sleeproom/async/websocket.rb
+- lib/sleeproom/cli.rb
+- lib/sleeproom/record.rb
+- lib/sleeproom/record/api/api.rb
+- lib/sleeproom/record/api/room.rb
+- lib/sleeproom/record/api/room_api.rb
+- lib/sleeproom/record/api/streaming_api.rb
+- lib/sleeproom/record/record.rb
+- lib/sleeproom/record/tasks.rb
+- lib/sleeproom/record/websocket.rb
+- lib/sleeproom/record/write_status.rb
+- lib/sleeproom/utils.rb
+- lib/sleeproom/version.rb
+- sleeproom.gemspec
+homepage: https://github.com/KoellM/sleeproom
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/KoellM/sleeproom
+post_install_message: |
+  SleepRoom 需要:
+    [Minyami]       https://github.com/Last-Order/minyami
+    (npm install -g minyami)
+  使用前请确保已安装上述依赖。
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 2.3.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.4
+signing_key: 
+specification_version: 4
+summary: sleeproom
+test_files: []