Merge pull request #18 from THE-POWERNEWS/develop

1.4.0

Author: pooza

CLAUDE.md

@@ -25,6 +25,12 @@ VPS上で実行するサーバー管理ユーティリティ集。FreeBSD / Ubun
 - テスト: `bin/wb help` で動作確認可能
 - GitHub CLI: `gh`
 
+## Mastodon関連ツールの共通パターン
+
+- `config/application.yaml`の`mastodon`セクション（user, rails_env, dir）を共通設定として参照
+- tootctlの実行は`sudo -u`＋`CommandLine`で、`RAILS_ENV`環境変数とディレクトリを設定
+- 現在FreeBSD専用（Ubuntu対応は不要）
+
 ## 注意点
 
 - 外部gemのrequireは不要（`Bundler.require`で自動ロード）

README.md

@@ -43,10 +43,14 @@ rake uninstall  # cronスクリプトをアンインストール
 | --- | --- |
 | access_log_compress | 指定日数が経過したログファイルをgzip圧縮します。 |
 | help | ツール一覧を表示します。 |
+| mastodon_follow | 全ユーザーに指定アカウントを強制フォローさせます。 |
+| mastodon_maintenance | Mastodonのメンテナンスコマンドを実行します。 |
+| mastodon_media_cleanup | Mastodonの古いメディアファイルを削除します。 |
 | mysql_dump | MySQLのダンプファイルを作成します。 |
 | postgresql_dump | PostgreSQLのダンプファイルを作成します。 |
 | postgresql_snapshot | PostgreSQLのZFSスナップショットを作成・管理します。 |
 | reboot_required | システムに再起動が必要かを判定します。 |
+| service_restart | 設定されたサービスを再起動します。 |
 
 ## 設定
 
@@ -91,6 +95,38 @@ rake uninstall  # cronスクリプトをアンインストール
 | days | スナップショット保管日数 | 3 |
 | dsn | PostgreSQL接続文字列 | postgres://postgres@localhost/mastodon |
 
+### mastodon（共通設定）
+
+| キー | 説明 | デフォルト |
+| --- | --- | --- |
+| user | 実行ユーザー | mastodon |
+| rails_env | RAILS_ENV環境変数 | production |
+| dir | Mastodonインストールディレクトリ | /home/mastodon/repos/mastodon |
+
+### service_restart
+
+| キー | 説明 | デフォルト |
+| --- | --- | --- |
+| services | 再起動するサービス名の配列 | [mastodon-sidekiq] |
+
+### mastodon_maintenance
+
+| キー | 説明 | デフォルト |
+| --- | --- | --- |
+| commands | 実行するtootctlサブコマンドの配列 | [cache recount accounts, accounts cull] |
+
+### mastodon_follow
+
+| キー | 説明 | デフォルト |
+| --- | --- | --- |
+| account | 強制フォローするアカウント名 | info |
+
+### mastodon_media_cleanup
+
+| キー | 説明 | デフォルト |
+| --- | --- | --- |
+| commands | 実行するtootctlサブコマンドの配列 | [media remove-orphans, ...] |
+
 ## ライセンス
 
 MIT

app/lib/writers_base/tool/mastodon_follow_tool.rb

@@ -0,0 +1,35 @@
+module WritersBase
+  class MastodonFollowTool < Tool
+    def exec(args = {})
+      result = {success: [], failure: []}
+      logger.info(tool: underscore, account:, message: 'フォロー実行開始')
+      tootctl_command(['account', 'follow', account])
+      result[:success].push(account)
+      return result
+    rescue => e
+      logger.error(tool: underscore, account:, error: e.message.strip)
+      result[:failure].push(account:, error: e.message.strip)
+      return result
+    end
+
+    def description
+      return '全ユーザーに指定アカウントを強制フォローさせます。'
+    end
+
+    private
+
+    def tootctl_command(args)
+      command = CommandLine.new(
+        ['sudo', '-u', mastodon_user, 'bundle', 'exec', 'bin/tootctl', *args],
+      )
+      command.env = {'RAILS_ENV' => mastodon_rails_env}
+      command.dir = mastodon_dir
+      command.exec unless test?
+      return command
+    end
+
+    def mastodon_user = config['/mastodon/user']
+    def mastodon_rails_env = config['/mastodon/rails_env']
+    def mastodon_dir = config['/mastodon/dir']
+  end
+end

app/lib/writers_base/tool/mastodon_maintenance_tool.rb

@@ -0,0 +1,37 @@
+module WritersBase
+  class MastodonMaintenanceTool < Tool
+    def exec(args = {})
+      result = {success: [], failure: []}
+      commands.each do |cmd|
+        tootctl_args = cmd.split(/\s+/)
+        logger.info(tool: underscore, command: cmd, message: '実行開始')
+        tootctl_command(tootctl_args)
+        result[:success].push(cmd)
+      rescue => e
+        logger.error(tool: underscore, command: cmd, error: e.message.strip)
+        result[:failure].push(command: cmd, error: e.message.strip)
+      end
+      return result
+    end
+
+    def description
+      return 'Mastodonのメンテナンスコマンドを実行します。'
+    end
+
+    private
+
+    def tootctl_command(args)
+      command = CommandLine.new(
+        ['sudo', '-u', mastodon_user, 'bundle', 'exec', 'bin/tootctl', *args],
+      )
+      command.env = {'RAILS_ENV' => mastodon_rails_env}
+      command.dir = mastodon_dir
+      command.exec unless test?
+      return command
+    end
+
+    def mastodon_user = config['/mastodon/user']
+    def mastodon_rails_env = config['/mastodon/rails_env']
+    def mastodon_dir = config['/mastodon/dir']
+  end
+end

app/lib/writers_base/tool/mastodon_media_cleanup_tool.rb

@@ -0,0 +1,37 @@
+module WritersBase
+  class MastodonMediaCleanupTool < Tool
+    def exec(args = {})
+      result = {success: [], failure: []}
+      commands.each do |cmd|
+        tootctl_args = cmd.split(/\s+/)
+        logger.info(tool: underscore, command: cmd, message: '実行開始')
+        tootctl_command(tootctl_args)
+        result[:success].push(cmd)
+      rescue => e
+        logger.error(tool: underscore, command: cmd, error: e.message.strip)
+        result[:failure].push(command: cmd, error: e.message.strip)
+      end
+      return result
+    end
+
+    def description
+      return 'Mastodonの古いメディアファイルを削除します。'
+    end
+
+    private
+
+    def tootctl_command(args)
+      command = CommandLine.new(
+        ['sudo', '-u', mastodon_user, 'bundle', 'exec', 'bin/tootctl', *args],
+      )
+      command.env = {'RAILS_ENV' => mastodon_rails_env}
+      command.dir = mastodon_dir
+      command.exec unless test?
+      return command
+    end
+
+    def mastodon_user = config['/mastodon/user']
+    def mastodon_rails_env = config['/mastodon/rails_env']
+    def mastodon_dir = config['/mastodon/dir']
+  end
+end

app/lib/writers_base/tool/service_restart_tool.rb

@@ -0,0 +1,21 @@
+module WritersBase
+  class ServiceRestartTool < Tool
+    def exec(args = {})
+      result = {success: [], failure: []}
+      services.each do |name|
+        logger.info(tool: underscore, service: name, message: '再起動開始')
+        command = CommandLine.new(['service', name, 'restart'])
+        command.exec unless test?
+        result[:success].push(name)
+      rescue => e
+        logger.error(tool: underscore, service: name, error: e.message.strip)
+        result[:failure].push(service: name, error: e.message.strip)
+      end
+      return result
+    end
+
+    def description
+      return '設定されたサービスを再起動します。'
+    end
+  end
+end

config/application.yaml

@@ -19,7 +19,7 @@ package:
     - tkoishi@b-shock.co.jp
   license: MIT
   url: https://github.com/THE-POWERNEWS/writersbase-tools
-  version: 1.3.1
+  version: 1.4.0
 ruby:
   jit: false
   bin: /usr/bin/ruby3.3
@@ -54,3 +54,21 @@ postgresql_snapshot:
   target: zroot/postgres
   days: 3
   dsn: postgres://postgres@localhost/mastodon
+mastodon:
+  user: mastodon
+  rails_env: production
+  dir: /home/mastodon/repos/mastodon
+service_restart:
+  services:
+    - mastodon-sidekiq
+mastodon_maintenance:
+  commands:
+    - cache recount accounts
+    - accounts cull
+mastodon_follow:
+  account: info
+mastodon_media_cleanup:
+  commands:
+    - media remove-orphans
+    - media remove --remote-headers
+    - preview_cards remove -c 1