castcaster 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 473f74bfc9f9b8f427068f501fc5db2dc13071234084647028e13f487152dedd
+  data.tar.gz: 8566513c993848a0160a91f6c548c1c0dcd1cfdd9a162decb5df4c5e5a0a1122
+SHA512:
+  metadata.gz: 7169e96f6ba961b134a96205f43bd023c6ed6a6a1287d57e73e3e78b34fabbb011f19059beb1b27550611a87ac901b51b014a2ef51fdfce99f6d5fd544c21171
+  data.tar.gz: a777f05ee638aa1cb76c74edfcf76c1d8d0aa7b1f8cb405ccd4cfe6c34294c41949144ae47b581d26194b31e02759db0b8a180ac4e99bb8e0da211f49d7dd742

data/bin/castcaster

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+ENV['THOR_SILENCE_DEPRECATION'] = '1'
+
+require_relative '../lib/castcaster'
+
+CastCaster::CLI.start(ARGV)

data/lib/castcaster/channel.rb

@@ -0,0 +1,91 @@
+module CastCaster
+  class Channel
+    STATUSES = %w[stopped live error].freeze
+
+    attr_reader :name, :path
+    attr_accessor :engine, :source, :hls, :transcode, :snapshot, :status
+
+    def initialize(name, attrs = {})
+      @name = name
+      @engine = attrs['engine'] || 'default'
+      @source = attrs['source'] || {}
+      @hls = attrs['hls'] || { 'fragment' => 6, 'playlist_length' => 60 }
+      @transcode = attrs['transcode']
+      @snapshot = attrs['snapshot']
+      @status = attrs['status'] || 'stopped'
+      @path = File.join(self.class.channels_dir, "#{name}.yml")
+    end
+
+    def to_h
+      {
+        'name' => @name,
+        'engine' => @engine,
+        'source' => @source,
+        'hls' => @hls,
+        'transcode' => @transcode,
+        'snapshot' => @snapshot,
+        'status' => @status
+      }.compact
+    end
+
+    def save
+      File.write(@path, YAML.dump(to_h))
+    end
+
+    def destroy
+      File.delete(@path) if File.exist?(@path)
+    end
+
+    def live?
+      @status == 'live'
+    end
+
+    def validate!
+      raise Error, "Channel name required" if @name.nil? || @name.empty?
+      raise Error, "Invalid channel name: #{@name}" unless @name.match?(/\A[a-zA-Z0-9_-]+\z/)
+    end
+
+    class << self
+      def channels_dir
+        File.join(Dir.pwd, 'channels')
+      end
+
+      def all
+        Dir[File.join(channels_dir, '*.yml')].map do |f|
+          from_file(f)
+        end.compact.sort_by(&:name)
+      end
+
+      def find(name)
+        path = File.join(channels_dir, "#{name}.yml")
+        return nil unless File.exist?(path)
+        from_file(path)
+      end
+
+      def create(name, attrs = {})
+        ch = new(name, attrs)
+        ch.validate!
+        raise Error, "Channel '#{name}' already exists" if File.exist?(ch.path)
+        ch.save
+        ch
+      end
+
+      private
+
+      def from_file(path)
+        data = YAML.safe_load(File.read(path))
+        unless data
+          warn "Warning: failed to parse channel file: #{path}"
+          return nil
+        end
+        ch = new(data['name'], data)
+        ch.path = path
+        ch
+      end
+    end
+
+    def path=(p)
+      @path = p
+    end
+  end
+end

data/lib/castcaster/channel_cli.rb

@@ -0,0 +1,105 @@
+module CastCaster
+  class ChannelCLI < Thor
+    desc 'list', 'List all channels'
+    def list
+      channels = Channel.all
+      if channels.empty?
+        say 'No channels. Use `castcaster channel add --name NAME` to create one.'
+        return
+      end
+
+      shell = Thor::Shell::Basic.new
+      rows = channels.map do |ch|
+        [ch.name, ch.engine, ch.live? ? 'live' : 'stopped', ch.source.fetch('url', '-')]
+      end
+      shell.print_table(rows, headings: %w[Name Engine Status Source])
+    end
+
+    desc 'add', 'Add a new channel'
+    option :name, type: :string, required: true, desc: 'Channel name (a-z, 0-9, -, _)'
+    option :engine, type: :string, desc: 'Engine override'
+    option :source, type: :string, desc: 'Stream source URL'
+    option :source_type, type: :string, default: 'test', desc: 'Source type (test / rtmp_pull / rtmp_push / http_pull / srt)'
+    option :transcode, type: :string, desc: 'Transcode profiles (e.g. 720p,480p,360p)'
+    option :bitrate, type: :string, desc: 'Target bitrate (e.g. 1500k)'
+    option :snapshot, type: :numeric, desc: 'Snapshot interval in seconds'
+    def add
+      attrs = {
+        'engine' => options[:engine],
+        'source' => { 'type' => options[:source_type], 'url' => options[:source] }.compact,
+        'status' => 'stopped'
+      }
+      attrs['transcode'] = { 'profiles' => options[:transcode], 'bitrate' => options[:bitrate] }.compact if options[:transcode] || options[:bitrate]
+      attrs['snapshot'] = options[:snapshot] if options[:snapshot]
+      Channel.create(options[:name], attrs)
+      say_status :ok, "channel '#{options[:name]}' created"
+      say_status :info, "Run `docker compose restart nginx` to apply"
+    rescue Error => e
+      say_status :error, e.message
+      exit 1
+    end
+
+    desc 'rm NAME', 'Remove a channel'
+    def rm(name)
+      ch = Channel.find(name)
+      unless ch
+        say_status :error, "channel '#{name}' not found"
+        exit 1
+      end
+      ch.destroy
+      say_status :ok, "channel '#{name}' removed"
+      say_status :info, "Run `docker compose restart nginx` to apply"
+    end
+
+    desc 'update NAME', 'Update channel configuration'
+    option :source_type, type: :string, desc: 'Source type (test / rtmp_pull / rtmp_push / http_pull / srt)'
+    option :source, type: :string, desc: 'Stream source URL'
+    option :transcode, type: :string, desc: 'Transcode profiles (e.g. 720p,480p,360p)'
+    option :bitrate, type: :string, desc: 'Target bitrate (e.g. 1500k)'
+    option :snapshot, type: :numeric, desc: 'Snapshot interval in seconds (0 to disable)'
+    def update(name)
+      ch = Channel.find(name)
+      unless ch
+        say_status :error, "channel '#{name}' not found"
+        exit 1
+      end
+      re_read = YAML.safe_load(File.read(ch.path)) || {}
+      re_read['source'] ||= {}
+      re_read['source']['type'] = options[:source_type] if options[:source_type]
+      re_read['source']['url'] = options[:source] if options[:source]
+      if options[:transcode] || options[:bitrate]
+        re_read['transcode'] = { 'profiles' => options[:transcode], 'bitrate' => options[:bitrate] }.compact
+      end
+      if options.key?(:snapshot)
+        re_read['snapshot'] = options[:snapshot] > 0 ? options[:snapshot] : nil
+      end
+      File.write(ch.path, YAML.dump(re_read))
+      say_status :ok, "channel '#{name}' updated"
+      say_status :info, "Run `docker compose restart nginx` to apply"
+    end
+
+    desc 'start NAME', 'Mark a channel as live'
+    def start(name)
+      ch = Channel.find(name)
+      unless ch
+        say_status :error, "channel '#{name}' not found"
+        exit 1
+      end
+      ch.status = 'live'
+      ch.save
+      say_status :ok, "channel '#{name}' marked live"
+    end
+
+    desc 'stop NAME', 'Mark a channel as stopped'
+    def stop(name)
+      ch = Channel.find(name)
+      unless ch
+        say_status :error, "channel '#{name}' not found"
+        exit 1
+      end
+      ch.status = 'stopped'
+      ch.save
+      say_status :ok, "channel '#{name}' marked stopped"
+    end
+  end
+end

data/lib/castcaster/cli.rb

@@ -0,0 +1,104 @@
+require 'thor'
+
+module CastCaster
+  class CLI < Thor
+    package_name 'castcaster'
+
+    map %w[--version -v] => :version
+
+    desc 'init', 'Generate CastCaster config and project structure'
+    option :engine, type: :string, aliases: '-e', desc: 'Streaming engine (nginx-rtmp)'
+    def init
+      config = Config.new
+      config.ensure_project_dirs
+      token = config.ensure_token
+
+      cfg = Config::DEFAULTS.dup
+      cfg['engine'] = options[:engine] if options[:engine]
+      cfg['project_dir'] = Dir.pwd
+      config.save(cfg)
+
+      say_status :created, "castcaster.yml"
+      say_status :created, "castcaster.token"
+      say_status :created, "channels/"
+      say_status :created, "hls/"
+      say ''
+      say "  CastCaster v#{VERSION} initialized"
+      say "  Engine: #{cfg['engine']}"
+      say ''
+      say "  Edit channels/, then run:"
+      say "    docker compose up -d"
+    end
+
+    desc 'config', 'Preview generated nginx config'
+    option :engine, type: :string, aliases: '-e', desc: 'Override engine'
+    def config
+      cfg = Config.new.load
+      engine_name = options[:engine] || cfg['engine']
+      engine = Engines.create(engine_name, cfg)
+      say engine.config_preview
+    end
+
+    desc 'deploy', 'Generate compose.yml'
+    option :engine, type: :string, aliases: '-e', desc: 'Override engine'
+    option :mode, type: :string, aliases: '-m', default: 'compose', desc: 'compose | swarm | k8s'
+    option :with_traefik, type: :boolean, desc: 'Include Traefik'
+    def deploy
+      cfg = Config.new.load
+      cfg['enable_traefik'] = true if options[:with_traefik]
+      engine = create_engine
+      channels = Channel.all
+      case options[:mode]
+      when 'swarm'
+        dpl = Deploy::Swarm.new(engine, channels, cfg)
+      when 'k8s'
+        dpl = Deploy::K8s.new(engine, channels, cfg)
+      else
+        dpl = Deploy::Compose.new(engine, channels)
+        dpl.with_traefik = true if options[:with_traefik]
+      end
+      path = dpl.write
+      say_status :created, path
+    end
+
+    IMAGES = {
+      'nginx' => 'ghcr.io/lax/castcaster-nginx',
+      'ffmpeg' => 'ghcr.io/lax/castcaster-ffmpeg',
+      'webui' => 'ghcr.io/lax/castcaster-webui'
+    }.freeze
+
+    desc 'doctor', 'Check system environment'
+    def doctor
+      checks = {
+        'Ruby'          => RUBY_VERSION,
+        'Docker'        => system('docker --version >/dev/null 2>&1') ? `docker --version`.strip : 'not found',
+        'Docker Compose'=> system('docker compose version >/dev/null 2>&1') ? `docker compose version`.strip : 'not found',
+        'config.yml'    => File.exist?('castcaster.yml') ? 'ok' : 'not found (run init)'
+      }
+      IMAGES.each do |name, tag|
+        if system("docker image inspect #{tag} >/dev/null 2>&1")
+          checks[name] = 'available'
+        else
+          checks[name] = "docker pull #{tag}"
+        end
+      end
+      checks.each { |name, status| say "#{name.ljust(14)} #{status}" }
+    end
+
+    desc 'channel SUBCOMMAND ...', 'Manage channels'
+    subcommand 'channel', ChannelCLI
+
+    desc 'version', 'Show version'
+    def version
+      say VERSION
+    end
+
+    private
+
+    def create_engine
+      cfg = Config.new.load
+      engine_name = options[:engine] || cfg['engine']
+      Engines.create(engine_name, cfg)
+    end
+  end
+end

data/lib/castcaster/config.rb

@@ -0,0 +1,60 @@
+module CastCaster
+  class Config
+    DEFAULTS = {
+      'engine'       => 'nginx-rtmp',
+      'hls_fragment' => 6,
+      'hls_window'   => 60,
+      'domain'       => '',
+      'acme_email'   => 'admin@example.com',
+      'enable_traefik' => false
+    }.freeze
+
+    def initialize(path = nil)
+      @path = path || default_config_path
+    end
+
+    def load
+      return DEFAULTS.dup unless File.exist?(@path)
+      cfg = YAML.safe_load(File.read(@path)) || {}
+      DEFAULTS.merge(cfg)
+    rescue => e
+      raise Error, "Config corrupted: #{e.message}"
+    end
+
+    def save(data)
+      FileUtils.mkdir_p(File.dirname(@path))
+      File.write(@path, YAML.dump(data))
+    end
+
+    def ensure_project_dirs
+      project_dir = Dir.pwd
+      FileUtils.mkdir_p(File.join(project_dir, 'channels'))
+      FileUtils.mkdir_p(File.join(project_dir, 'hls'))
+      FileUtils.mkdir_p(File.join(project_dir, 'nginx'))
+      FileUtils.mkdir_p(File.join(project_dir, 'logs'))
+      FileUtils.mkdir_p(File.join(project_dir, 'data'))
+      project_dir
+    end
+
+    def ensure_token
+      path = File.join(Dir.pwd, 'castcaster.token')
+      return File.read(path).strip if File.exist?(path)
+      token = SecureRandom.hex(32)
+      File.write(path, token)
+      File.chmod(0600, path)
+      token
+    end
+
+    def load_token
+      path = File.join(Dir.pwd, 'castcaster.token')
+      return nil unless File.exist?(path)
+      File.read(path).strip
+    end
+
+    private
+
+    def default_config_path
+      File.join(Dir.pwd, 'castcaster.yml')
+    end
+  end
+end

data/lib/castcaster/deploy/compose.rb

@@ -0,0 +1,76 @@
+module CastCaster
+  module Deploy
+    class Compose
+      include FFmpegServices
+      attr_writer :with_traefik
+
+      def initialize(engine, channels)
+        @engine = engine
+        @channels = channels
+        @cfg = Config.new.load
+        @with_traefik = false
+      end
+
+      def write
+        File.write(compose_file, render)
+        compose_file
+      end
+
+      def compose_file
+        File.join(project_dir, 'compose.yml')
+      end
+
+      private
+
+      def project_dir
+        @cfg.fetch('project_dir', Dir.pwd)
+      end
+
+      def render
+        enable_traefik = @with_traefik || @cfg.fetch('enable_traefik', false)
+        nginx_ports = ['1935:1935']
+        nginx_ports << "#{@cfg.fetch('web_port', 8080)}:8080" unless enable_traefik
+
+        all = {
+          'nginx' => {
+            'image' => 'ghcr.io/lax/castcaster-nginx',
+            'restart' => 'unless-stopped',
+            'ports' => nginx_ports,
+            'volumes' => [
+              "./nginx/nginx.conf:/etc/nginx/nginx.conf:ro",
+              "./hls:/var/lib/castcaster/hls"
+            ]
+          },
+          'webui' => {
+            'image' => 'ghcr.io/lax/castcaster-webui',
+            'restart' => 'unless-stopped',
+            'volumes' => ["./channels:/channels:rw", "./hls:/var/lib/castcaster/hls:ro"],
+            'depends_on' => ['nginx'],
+            'environment' => {
+              'HLS_DIR' => '/var/lib/castcaster/hls',
+              'CHANNELS_DIR' => '/channels'
+            }
+          }
+        }
+
+        # FFmpeg services (relative paths for host-side compose)
+        all.merge!(build_ffmpeg_services(@channels, './hls'))
+
+        if enable_traefik
+          traefik = Deploy::Traefik.new(@cfg)
+          all.merge!(traefik.service_definition)
+          FileUtils.mkdir_p(File.join(project_dir, 'traefik'))
+          File.write(File.join(project_dir, 'traefik', 'dynamic.yml'), traefik.send(:dynamic_config))
+        end
+
+        tpl = File.read(File.expand_path('../../../templates/deploy/docker-compose.yml.erb', __dir__))
+        erb = ERB.new(tpl, trim_mode: '-')
+        ctx = TemplateContext.new(services: all)
+        erb.result(ctx.get_binding)
+      end
+
+
+
+    end
+  end
+end

data/lib/castcaster/deploy/ffmpeg_services.rb

@@ -0,0 +1,73 @@
+module CastCaster
+  module Deploy
+    module FFmpegServices
+      def build_ffmpeg_services(channels, hls_dir)
+        base = 'rtmp://nginx:1935/live'
+        services = {}
+        channels.each do |ch|
+          src = ch.source
+          next unless src && src['type']
+          next if src['type'] == 'rtmp_push'
+
+          name = ch.name
+          label = { 'castcaster/channel' => name }
+          common = {
+            'image' => 'ghcr.io/lax/castcaster-ffmpeg',
+            'restart' => 'unless-stopped',
+            'volumes' => ["#{hls_dir}:/var/lib/castcaster/hls"],
+            'depends_on' => ['nginx'],
+            'labels' => label,
+            'environment' => {
+              'CHANNEL_NAME' => name,
+              'RTMP_BASE' => base
+            }
+          }
+
+          case src['type']
+          when 'test'
+            services["castcaster-test-#{name}"] = common.merge(
+              'command' => ["-c", "teststream.sh"],
+              'labels' => label.merge('castcaster/task' => 'test')
+            )
+          else
+            next unless src['url']
+            bitrate = (ch.transcode || {}).fetch('bitrate', FFmpegProfiles::RELAY_DEFAULTS['bitrate'])
+            profiles = (ch.transcode || {}).fetch('profiles', nil)
+
+            if profiles
+              services["castcaster-adaptive-#{name}"] = common.merge(
+                'environment' => common['environment'].merge(
+                  'SOURCE_URL' => src['url'],
+                  'PROFILES' => profiles
+                ),
+                'command' => ["-c", "adaptive.sh"],
+                'labels' => label.merge('castcaster/task' => 'adaptive')
+              )
+            else
+              services["castcaster-relay-#{name}"] = common.merge(
+                'environment' => common['environment'].merge(
+                  'SOURCE_URL' => src['url'],
+                  'BITRATE' => bitrate
+                ),
+                'command' => ["-c", "relay.sh"],
+                'labels' => label.merge('castcaster/task' => 'relay')
+              )
+            end
+
+            if ch.snapshot
+              services["castcaster-snapshot-#{name}"] = common.merge(
+                'environment' => common['environment'].merge(
+                  'INTERVAL' => ch.snapshot.to_s,
+                  'SNAPSHOTS_DIR' => '/var/lib/castcaster/snapshots'
+                ),
+                'command' => ["-c", "snapshot.sh"],
+                'labels' => label.merge('castcaster/task' => 'snapshot')
+              )
+            end
+          end
+        end
+        services
+      end
+    end
+  end
+end

data/lib/castcaster/deploy/k8s.rb

@@ -0,0 +1,91 @@
+module CastCaster
+  module Deploy
+    class K8s
+      def initialize(engine, channels, cfg)
+        @engine = engine
+        @channels = channels
+        @cfg = cfg
+        @traefik = Deploy::Traefik.new(@cfg) if @cfg.fetch('enable_traefik', false)
+      end
+
+      def write
+        FileUtils.mkdir_p(deploy_dir)
+        write_file('deployment.yaml', render_deployment)
+        write_file('service.yaml', render_service)
+        write_file('configmap.yaml', render_configmap)
+        write_file('pvc.yaml', render_pvc)
+        write_file('ingress.yaml', render_ingress) if @traefik
+        deploy_dir
+      end
+
+      def deploy_dir
+        File.join(@cfg.fetch('project_dir', Dir.pwd), 'k8s')
+      end
+
+      private
+
+      def write_file(name, content)
+        path = File.join(deploy_dir, name)
+        File.write(path, content)
+      end
+
+      def render_deployment
+        tpl = File.read(File.expand_path('../../../templates/deploy/k8s/deployment.yaml.erb', __dir__))
+        erb = ERB.new(tpl, trim_mode: '-')
+        ctx = TemplateContext.new(
+          app: 'castcaster',
+          engine_image: 'ghcr.io/lax/castcaster-nginx',
+          webui_image: 'ghcr.io/lax/castcaster-webui',
+          ffmpeg_image: 'ghcr.io/lax/castcaster-ffmpeg',
+          engine_name: @engine.name,
+          channels_dir: File.join(@cfg.fetch('project_dir', Dir.pwd), 'channels'),
+          hls_dir: @cfg.fetch('hls_dir', '/var/lib/castcaster/hls'),
+          web_port: @cfg.fetch('web_port', 9527),
+          webui_port: @cfg.fetch('webui_port', 9528)
+        )
+        erb.result(ctx.get_binding)
+      end
+
+      def render_service
+        tpl = File.read(File.expand_path('../../../templates/deploy/k8s/service.yaml.erb', __dir__))
+        erb = ERB.new(tpl, trim_mode: '-')
+        ctx = TemplateContext.new(
+          app: 'castcaster',
+          web_port: @cfg.fetch('web_port', 9527),
+          webui_port: @cfg.fetch('webui_port', 9528)
+        )
+        erb.result(ctx.get_binding)
+      end
+
+      def render_configmap
+        tpl = File.read(File.expand_path('../../../templates/deploy/k8s/configmap.yaml.erb', __dir__))
+        erb = ERB.new(tpl, trim_mode: '-')
+        ctx = TemplateContext.new(
+          app: 'castcaster',
+          nginx_conf_path: File.join(@cfg.fetch('project_dir', Dir.pwd), 'nginx', 'nginx.conf'),
+          channels_dir: File.join(@cfg.fetch('project_dir', Dir.pwd), 'channels')
+        )
+        erb.result(ctx.get_binding)
+      end
+
+      def render_pvc
+        tpl = File.read(File.expand_path('../../../templates/deploy/k8s/pvc.yaml.erb', __dir__))
+        erb = ERB.new(tpl, trim_mode: '-')
+        ctx = TemplateContext.new(app: 'castcaster')
+        erb.result(ctx.get_binding)
+      end
+
+      def render_ingress
+        domain = @cfg.fetch('domain', 'stream.example.com')
+        tpl = File.read(File.expand_path('../../../templates/deploy/k8s/ingress.yaml.erb', __dir__))
+        erb = ERB.new(tpl, trim_mode: '-')
+        ctx = TemplateContext.new(
+          app: 'castcaster',
+          domain: domain,
+          acme_email: @cfg.fetch('acme_email', 'admin@example.com')
+        )
+        erb.result(ctx.get_binding)
+      end
+    end
+  end
+end

data/lib/castcaster/deploy/swarm.rb

@@ -0,0 +1,57 @@
+module CastCaster
+  module Deploy
+    class Swarm
+      include FFmpegServices
+
+      def initialize(engine, channels, cfg)
+        @engine = engine
+        @channels = channels
+        @cfg = cfg
+      end
+
+      def write
+        FileUtils.mkdir_p(deploy_dir)
+        File.write(compose_file, render)
+        compose_file
+      end
+
+      def compose_file
+        File.join(deploy_dir, 'docker-stack.yml')
+      end
+
+      def deploy_dir
+        @cfg.fetch('project_dir', Dir.pwd)
+      end
+
+      private
+
+      def render
+        all = {
+          'nginx' => {
+            'image' => 'ghcr.io/lax/castcaster-nginx',
+            'restart' => 'unless-stopped',
+            'ports' => ['1935:1935', '8080:8080'],
+            'volumes' => ["./nginx/nginx.conf:/etc/nginx/nginx.conf:ro", "./hls:/var/lib/castcaster/hls"]
+          },
+          'webui' => {
+            'image' => 'ghcr.io/lax/castcaster-webui',
+            'restart' => 'unless-stopped',
+            'volumes' => ["./channels:/channels:rw"],
+            'environment' => {'HLS_DIR' => '/var/lib/castcaster/hls', 'CHANNELS_DIR' => '/channels'}
+          }
+        }
+
+        all.merge!(build_ffmpeg_services(@channels, './hls'))
+
+        tpl = File.read(File.expand_path('../../../templates/deploy/docker-stack.yml.erb', __dir__))
+        erb = ERB.new(tpl, trim_mode: '-')
+        ctx = TemplateContext.new(services: all, project: 'castcaster', traefik_public: @cfg.fetch('enable_traefik', false))
+        erb.result(ctx.get_binding)
+      end
+
+      def project_dir
+        @cfg.fetch('project_dir', Dir.pwd)
+      end
+    end
+  end
+end