sdls 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 134c42aa266627ae45c1e085d2b33e36dda4a1a54ff91aac0c7c8af973f55ddd
+  data.tar.gz: 39ebe584b534cc372ec2de8c4c7f486e4b2075734fb8bab14207f1cea2ca9a18
+SHA512:
+  metadata.gz: 0a87ee27110e3e2adcecc1ee5ebdc22208a45edd552dbc2319c918f7120e78bae355fe54ee61e796536cc2e50f6c7661d91edcff9eef099152a8ada29c30f1f5
+  data.tar.gz: fe91dc17db9b7123d927583f3b10cfaec1265f1802ff279880a99123f5a8fe57f920072cafc41429bc2907dedfbcd5507c7c4945388178925ffea8c53870b394

data/bin/sdls

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require_relative "../lib/sdls"
+
+SDLS::CLI.start(ARGV)

data/lib/sdls/cli.rb

@@ -0,0 +1,89 @@
+# frozen_string_literal: true
+
+require "thor"
+require "tty-prompt"
+require "clipboard"
+
+module SDLS
+  class CLI < Thor
+    DEFAULT_CONFIG_PATH = File.expand_path("~/.config/sdls.yml")
+
+    def initialize(*args, config_path: nil, **kwargs)
+      super(*args, **kwargs)
+      @config_path = config_path || ENV["SDLS_CONFIG_PATH"] || DEFAULT_CONFIG_PATH
+    end
+
+    no_commands do
+      def current_config
+        @config ||= SDLS::Config.load(@config_path)
+      end
+
+      def client
+        SDLS::Client.new(
+          host: current_config.host,
+          username: current_config.username,
+          password: current_config.password,
+          op_item_name: current_config.op_item_name
+        )
+      end
+
+      def extract_torrent_name(magnet)
+        uri = URI.parse(magnet)
+        query = URI.decode_www_form(uri.query || "")
+        dn_param = query.find { |key, _| key == "dn" }
+        dn_param&.last
+      rescue URI::InvalidURIError
+        nil
+      end
+    end
+
+    desc "version", "Display the SDLS tool version"
+    def version
+      puts SDLS::VERSION
+    end
+
+    desc "config", "Display the current configuration"
+    def config
+      puts "Current config:"
+      puts "  host: #{current_config.host}"
+      puts "  username: #{current_config.username}"
+      puts "  password: [REDACTED]"
+      puts "  op_item_name: #{current_config.op_item_name}" if current_config.op_item_name
+      puts "  directories: #{current_config.directories.join(", ")}" if current_config.directories.any?
+    end
+
+    desc "connect", "Verify connectivity and authentication with the server"
+    def connect
+      sid = client.authenticate
+      if sid
+        puts "Connection successful. Session ID: #{sid.slice(0, 8)}..."
+      else
+        puts "Connection failed. Please check your credentials or server status."
+        exit 1
+      end
+    end
+
+    desc "add [MAGNET]", "Add a magnet link to Synology Download Station"
+    def add(magnet = nil)
+      magnet ||= Clipboard.paste.strip
+
+      unless magnet&.start_with?("magnet:")
+        warn "Invalid or missing magnet link."
+        exit 1
+      end
+
+      name = extract_torrent_name(magnet)
+      puts "Adding torrent: #{name}" if name
+
+      prompt = TTY::Prompt.new
+      destination = prompt.select("Choose download directory", current_config.directories, default: current_config.directories.first)
+
+      success = client.create_download(magnet: magnet, destination: destination)
+      exit 1 unless success
+    end
+
+    def self.exit_on_failure?
+      true
+    end
+  end
+end

data/lib/sdls/client.rb

@@ -0,0 +1,117 @@
+require "net/http"
+require "uri"
+require "json"
+require "open3"
+
+module SDLS
+  class Client
+    def initialize(host:, username:, password:, op_item_name: nil)
+      @host = host
+      @username = username
+      @password = password
+      @op_item_name = op_item_name
+    end
+
+    def authenticate(otp: nil)
+      uri = build_uri("/webapi/auth.cgi", auth_params(otp))
+      response_body = make_request(uri)
+
+      handle_auth_response(response_body, otp)
+    rescue => e
+      warn "Authentication error: #{e.message}"
+      nil
+    end
+
+    def create_download(magnet:, destination:)
+      sid = authenticate
+      return false unless sid
+
+      uri = build_uri("/webapi/DownloadStation/task.cgi")
+      data = download_params(magnet, destination, sid)
+
+      response = Net::HTTP.post_form(uri, data)
+      body = JSON.parse(response.body)
+
+      if response.is_a?(Net::HTTPSuccess) && body["success"]
+        puts "Download created successfully in #{destination}"
+        true
+      else
+        warn "Download creation failed: #{body.inspect}"
+        false
+      end
+    end
+
+    private
+
+    def build_uri(endpoint, params = nil)
+      uri = URI.parse("#{@host}#{endpoint}")
+      uri.query = URI.encode_www_form(params) if params
+      uri
+    end
+
+    def make_request(uri)
+      response = Net::HTTP.get_response(uri)
+      raise "HTTP error: #{response.code}" unless response.is_a?(Net::HTTPSuccess)
+      JSON.parse(response.body)
+    end
+
+    def auth_params(otp)
+      params = {
+        api: "SYNO.API.Auth",
+        version: 6,
+        method: "login",
+        account: @username,
+        passwd: @password,
+        session: "FileStation",
+        format: "cookie"
+      }
+      params[:otp_code] = otp if otp
+      params
+    end
+
+    def download_params(magnet, destination, sid)
+      {
+        api: "SYNO.DownloadStation.Task",
+        version: "1",
+        method: "create",
+        session: "DownloadStation",
+        _sid: sid,
+        uri: magnet,
+        destination: destination
+      }
+    end
+
+    def handle_auth_response(body, otp)
+      return body["data"]["sid"] if body["success"]
+
+      if otp_required?(body, otp)
+        retry_with_otp
+      else
+        raise "Authentication failed: #{body.inspect}"
+      end
+    end
+
+    def otp_required?(body, current_otp)
+      current_otp.nil? && body.dig("error", "errors", "types")&.any? { |e| e["type"] == "otp" }
+    end
+
+    def retry_with_otp
+      puts "OTP required for authentication. Fetching from 1Password..."
+      fetched_otp = fetch_otp_from_1password
+      return authenticate(otp: fetched_otp) if fetched_otp
+      raise "Could not retrieve OTP from 1Password"
+    end
+
+    def fetch_otp_from_1password
+      return nil unless @op_item_name
+
+      stdout, stderr, status = Open3.capture3("op item get \"#{@op_item_name}\" --otp")
+      if status.success?
+        stdout.strip
+      else
+        warn "Failed to retrieve OTP from 1Password: #{stderr.strip}"
+        nil
+      end
+    end
+  end
+end

data/lib/sdls/config.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "yaml"
+
+module SDLS
+  REQUIRED_KEYS = %i[host username password]
+  Config = Data.define(:host, :username, :password, :op_item_name, :directories) do
+    def self.load(path)
+      raise "Configuration file not found: #{path}" unless File.exist?(path)
+
+      data = YAML.load_file(path)
+      data = {} unless data.is_a?(Hash)
+      data = symbolize_keys(data)
+
+      missing_keys = REQUIRED_KEYS - data.keys
+      nil_keys = REQUIRED_KEYS.select { |k| data[k].nil? }
+
+      if missing_keys.any? || nil_keys.any?
+        raise "Configuration file (#{path}) is missing required keys or values: #{(missing_keys + nil_keys).uniq.join(", ")}"
+      end
+
+      new(**data)
+    rescue Psych::SyntaxError => e
+      raise "Error parsing configuration file (#{path}): #{e.message}"
+    end
+
+    def self.symbolize_keys(hash)
+      hash.to_h { |k, v| [k.to_sym, v] }
+    end
+  end
+end

data/lib/sdls/version.rb

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

data/lib/sdls.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+require_relative "sdls/cli"
+require_relative "sdls/config"
+require_relative "sdls/version"
+require_relative "sdls/client"
+
+module SDLS
+end

metadata

@@ -0,0 +1,46 @@
+--- !ruby/object:Gem::Specification
+name: sdls
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Camillo Visini
+bindir: bin
+cert_chain: []
+date: 1980-01-02 00:00:00.000000000 Z
+dependencies: []
+description: A command-line interface for managing downloads on Synology Download
+  Station
+executables:
+- sdls
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/sdls
+- lib/sdls.rb
+- lib/sdls/cli.rb
+- lib/sdls/client.rb
+- lib/sdls/config.rb
+- lib/sdls/version.rb
+homepage: https://github.com/visini/sdls
+licenses:
+- MIT
+metadata: {}
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3.4'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.6.9
+specification_version: 4
+summary: Synology Download Station CLI
+test_files: []