Sigs

Arie · Aug 18, 2024 · 85ef737 · 85ef737
1 parent 87239cd
commit 85ef737
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/app/models/server.rb b/app/models/server.rb
@@ -144,22 +144,22 @@ def enable_plugins
     write_configuration(sourcemod_file, sourcemod_body)
   end
 
-  sig { params(user: User).returns(String) }
+  sig { params(user: User).returns(T.any(String, T::Boolean)) }
   def add_sourcemod_admin(user)
     write_configuration(sourcemod_admin_file, sourcemod_admin_body(user))
   end
 
-  sig { params(reservation: Reservation).returns(String) }
+  sig { params(reservation: Reservation).returns(T.any(String, T::Boolean)) }
   def add_sourcemod_servers(reservation)
     write_configuration(sourcemod_servers_file, sourcemod_servers_body(reservation))
   end
 
-  sig { params(reservation: Reservation).returns(String) }
+  sig { params(reservation: Reservation).returns(T.any(String, T::Boolean)) }
   def add_motd(reservation)
     write_configuration(motd_file, motd_body(reservation))
   end
 
-  sig { returns(T.nilable(T::Boolean)) }
+  sig { returns(T.nilable(T.any(T::Boolean, String))) }
   def disable_plugins
     delete_from_server([sourcemod_file, sourcemod_admin_file])
   end
@@ -390,12 +390,12 @@ def rcon_auth(rcon = current_rcon)
     nil
   end
 
-  sig { params(message: String).returns(T.nilable(String)) }
+  sig { params(message: String).returns(T.nilable(T.any(String, ActiveSupport::Multibyte::Chars))) }
   def rcon_say(message)
     rcon_exec("say #{message}")
   end
 
-  sig { params(command: String, allow_blocked: T::Boolean).returns(T.nilable(String)) }
+  sig { params(command: String, allow_blocked: T::Boolean).returns(T.nilable(T.any(String, ActiveSupport::Multibyte::Chars))) }
   def rcon_exec(command, allow_blocked: false)
     return nil if blocked_command?(command) && !allow_blocked
 

diff --git a/app/models/ssh_server.rb b/app/models/ssh_server.rb
@@ -73,7 +73,7 @@ def copy_to_server(files, destination)
     system("#{scp_command} #{files.map(&:shellescape).join(' ')} #{ip}:#{destination}")
   end
 
-  sig { params(files: [String], destination: String).returns(T.nilable(T::Boolean)) }
+  sig { params(files: T::Array[String], destination: String).returns(T.nilable(T::Boolean)) }
   def copy_from_server(files, destination)
     logger.debug "SCP GET, FILES: #{files.join(', ')} DESTINATION: #{destination}"
     system("#{scp_command} #{ip}:\"#{files.map(&:shellescape).join(' ')}\" #{destination}")