mirror of
https://github.com/iv-org/invidious
synced 2024-11-27 13:33:28 +01:00
Add sentry (from updated fork)
This commit is contained in:
parent
764fdf42ef
commit
c7c8cc1e85
109
dev/sentry.cr
Normal file
109
dev/sentry.cr
Normal file
@ -0,0 +1,109 @@
|
|||||||
|
module Sentry
|
||||||
|
FILE_TIMESTAMPS = {} of String => String # {file => timestamp}
|
||||||
|
|
||||||
|
class ProcessRunner
|
||||||
|
getter app_process : (Nil | Process) = nil
|
||||||
|
property process_name : String
|
||||||
|
property should_build = true
|
||||||
|
property files = [] of String
|
||||||
|
|
||||||
|
def initialize(
|
||||||
|
@process_name : String,
|
||||||
|
@build_command : String,
|
||||||
|
@run_command : String,
|
||||||
|
@build_args : Array(String) = [] of String,
|
||||||
|
@run_args : Array(String) = [] of String,
|
||||||
|
files = [] of String,
|
||||||
|
should_build = true)
|
||||||
|
@files = files
|
||||||
|
@should_build = should_build
|
||||||
|
@should_kill = false
|
||||||
|
@app_built = false
|
||||||
|
end
|
||||||
|
|
||||||
|
private def build_app_process
|
||||||
|
puts "🤖 compiling #{process_name}..."
|
||||||
|
build_args = @build_args
|
||||||
|
if build_args.size > 0
|
||||||
|
Process.run(@build_command, build_args, shell: true, output: Process::Redirect::Inherit, error: Process::Redirect::Inherit)
|
||||||
|
else
|
||||||
|
Process.run(@build_command, shell: true, output: Process::Redirect::Inherit, error: Process::Redirect::Inherit)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
private def create_app_process
|
||||||
|
app_process = @app_process
|
||||||
|
if app_process.is_a? Process
|
||||||
|
unless app_process.terminated?
|
||||||
|
puts "🤖 killing #{process_name}..."
|
||||||
|
app_process.kill
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
puts "🤖 starting #{process_name}..."
|
||||||
|
run_args = @run_args
|
||||||
|
if run_args.size > 0
|
||||||
|
@app_process = Process.new(@run_command, run_args, output: Process::Redirect::Inherit, error: Process::Redirect::Inherit)
|
||||||
|
else
|
||||||
|
@app_process = Process.new(@run_command, output: Process::Redirect::Inherit, error: Process::Redirect::Inherit)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
private def get_timestamp(file : String)
|
||||||
|
File.stat(file).mtime.to_s("%Y%m%d%H%M%S")
|
||||||
|
end
|
||||||
|
|
||||||
|
# Compiles and starts the application
|
||||||
|
#
|
||||||
|
def start_app
|
||||||
|
return create_app_process unless @should_build
|
||||||
|
build_result = build_app_process()
|
||||||
|
if build_result && build_result.success?
|
||||||
|
@app_built = true
|
||||||
|
create_app_process()
|
||||||
|
elsif !@app_built # if build fails on first time compiling, then exit
|
||||||
|
puts "🤖 Compile time errors detected. SentryBot shutting down..."
|
||||||
|
exit 1
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
# Scans all of the `@files`
|
||||||
|
#
|
||||||
|
def scan_files
|
||||||
|
file_changed = false
|
||||||
|
app_process = @app_process
|
||||||
|
files = @files
|
||||||
|
Dir.glob(files) do |file|
|
||||||
|
timestamp = get_timestamp(file)
|
||||||
|
if FILE_TIMESTAMPS[file]? && FILE_TIMESTAMPS[file] != timestamp
|
||||||
|
FILE_TIMESTAMPS[file] = timestamp
|
||||||
|
file_changed = true
|
||||||
|
puts "🤖 #{file}"
|
||||||
|
elsif FILE_TIMESTAMPS[file]?.nil?
|
||||||
|
puts "🤖 watching file: #{file}"
|
||||||
|
FILE_TIMESTAMPS[file] = timestamp
|
||||||
|
file_changed = true if (app_process && !app_process.terminated?)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
start_app() if (file_changed || app_process.nil?)
|
||||||
|
end
|
||||||
|
|
||||||
|
def run
|
||||||
|
puts "🤖 Your SentryBot is vigilant. beep-boop..."
|
||||||
|
|
||||||
|
loop do
|
||||||
|
if @should_kill
|
||||||
|
puts "🤖 Powering down your SentryBot..."
|
||||||
|
break
|
||||||
|
end
|
||||||
|
scan_files
|
||||||
|
sleep 1
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def kill
|
||||||
|
@should_kill = true
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
99
dev/sentry_cli.cr
Normal file
99
dev/sentry_cli.cr
Normal file
@ -0,0 +1,99 @@
|
|||||||
|
require "option_parser"
|
||||||
|
require "yaml"
|
||||||
|
require "./sentry"
|
||||||
|
|
||||||
|
process_name = nil
|
||||||
|
|
||||||
|
begin
|
||||||
|
shard_yml = YAML.parse File.read("shard.yml")
|
||||||
|
name = shard_yml["name"]?
|
||||||
|
process_name = name.as_s if name
|
||||||
|
rescue e
|
||||||
|
end
|
||||||
|
|
||||||
|
build_args = [] of String
|
||||||
|
build_command = "crystal build ./src/#{process_name}.cr"
|
||||||
|
run_args = [] of String
|
||||||
|
run_command = "./#{process_name}"
|
||||||
|
files = ["./src/**/*.cr", "./src/**/*.ecr"]
|
||||||
|
files_cleared = false
|
||||||
|
show_help = false
|
||||||
|
should_build = true
|
||||||
|
|
||||||
|
OptionParser.parse! do |parser|
|
||||||
|
parser.banner = "Usage: ./sentry [options]"
|
||||||
|
parser.on(
|
||||||
|
"-n NAME",
|
||||||
|
"--name=NAME",
|
||||||
|
"Sets the name of the app process (current name: #{process_name})") { |name| process_name = name }
|
||||||
|
parser.on(
|
||||||
|
"-b COMMAND",
|
||||||
|
"--build=COMMAND",
|
||||||
|
"Overrides the default build command") { |command| build_command = command }
|
||||||
|
parser.on(
|
||||||
|
"--build-args=ARGS",
|
||||||
|
"Specifies arguments for the build command") do |args|
|
||||||
|
args_arr = args.strip.split(" ")
|
||||||
|
build_args = args_arr if args_arr.size > 0
|
||||||
|
end
|
||||||
|
parser.on(
|
||||||
|
"--no-build",
|
||||||
|
"Skips the build step") { should_build = false }
|
||||||
|
parser.on(
|
||||||
|
"-r COMMAND",
|
||||||
|
"--run=COMMAND",
|
||||||
|
"Overrides the default run command") { |command| run_command = command }
|
||||||
|
parser.on(
|
||||||
|
"--run-args=ARGS",
|
||||||
|
"Specifies arguments for the run command") do |args|
|
||||||
|
args_arr = args.strip.split(" ")
|
||||||
|
run_args = args_arr if args_arr.size > 0
|
||||||
|
end
|
||||||
|
parser.on(
|
||||||
|
"-w FILE",
|
||||||
|
"--watch=FILE",
|
||||||
|
"Overrides default files and appends to list of watched files") do |file|
|
||||||
|
unless files_cleared
|
||||||
|
files.clear
|
||||||
|
files_cleared = true
|
||||||
|
end
|
||||||
|
files << file
|
||||||
|
end
|
||||||
|
parser.on(
|
||||||
|
"-i",
|
||||||
|
"--info",
|
||||||
|
"Shows the values for build/run commands, build/run args, and watched files") do
|
||||||
|
puts "
|
||||||
|
name: #{process_name}
|
||||||
|
build: #{build_command}
|
||||||
|
build args: #{build_args}
|
||||||
|
run: #{run_command}
|
||||||
|
run args: #{run_args}
|
||||||
|
files: #{files}
|
||||||
|
"
|
||||||
|
end
|
||||||
|
parser.on(
|
||||||
|
"-h",
|
||||||
|
"--help",
|
||||||
|
"Show this help") do
|
||||||
|
puts parser
|
||||||
|
exit 0
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
if process_name
|
||||||
|
process_runner = Sentry::ProcessRunner.new(
|
||||||
|
process_name: process_name.as(String),
|
||||||
|
build_command: build_command,
|
||||||
|
run_command: run_command,
|
||||||
|
build_args: build_args,
|
||||||
|
run_args: run_args,
|
||||||
|
should_build: should_build,
|
||||||
|
files: files
|
||||||
|
)
|
||||||
|
|
||||||
|
process_runner.run
|
||||||
|
else
|
||||||
|
puts "🤖 Sentry error: 'name' not given and not found in shard.yml"
|
||||||
|
exit 1
|
||||||
|
end
|
Loading…
Reference in New Issue
Block a user