diff --git a/app/controllers/clients_controller.rb b/app/controllers/clients_controller.rb index c7ea95e..1fdfaf7 100644 --- a/app/controllers/clients_controller.rb +++ b/app/controllers/clients_controller.rb @@ -1,158 +1,166 @@ class ClientsController < ApplicationController before_action :require_logged_in before_action :fetch_client, only: [:show, :jobs, :logs, :stats, :users, :restore, :run_restore, :restore_selected, :remove_user] before_action :fetch_logs, only: [:logs] + before_action :require_non_blocked_client, only: [:restore, :restore_selected, :run_restore] # GET /clients # POST /clients def index @client_ids = Client.for_user(current_user.id).pluck(:ClientId) @clients = Client.where(ClientId: @client_ids).includes(:jobs) @hosts = current_user.hosts.not_baculized fetch_jobs_info get_charts end # GET /clients/1 def show @schedules = @client.host.job_templates.map(&:schedule) @filesets = @client.host.job_templates.map(&:fileset) end # GET /clients/1/jobs def jobs @jobs = @client.recent_jobs.page(params[:page]) end # GET /clients/1/logs def logs; end # GET /clients/1/stats # POST /clients/1/stats def stats get_charts end # GET /clients/1/users def users @users = @client.host.users if @client.manually_inserted? @invitation = @client.host.invitations.new excluded_ids = @users.pluck(:id) + @client.host.invitations.pluck(:user_id) @available_users = User.institutional.where.not(id: excluded_ids).pluck(:username, :id) end end # DELETE /clients/1/user def remove_user user = @client.host.users.find(params[:user_id]) if @client.host.users.delete(user) flash[:success] = if @client.manually_inserted? 'User successfully removed' else 'User must be removed from the VM\'s list form your VM provider too (ViMa or Okeanos).' end else flash[:alert] = 'User not removed, something went wrong' end redirect_to users_client_path(@client) end # GET /clients/1/restore def restore return if @client.is_backed_up? flash[:error] = 'Can not issue a restore for this client' redirect_to client_path(@client) end # POST /clients/1/run_restore def run_restore @location = params[:restore_location].blank? ? '/tmp/bacula_restore' : params[:restore_location] fileset = params[:fileset] restore_point = fetch_restore_point if params[:commit] == 'Restore All Files' if @location.nil? || fileset.nil? || !@client.host.restore(fileset, @location, restore_point) flash[:error] = 'Something went wrong, try again later' else flash[:success] = "Restore job issued successfully, files will be soon available in #{@location}" end render js: "window.location = '#{client_path(@client)}'" else session[:job_ids] = @client.get_job_ids(fileset, restore_point) Bvfs.new(@client, session[:job_ids]).update_cache render 'select_files' end end # POST /clients/1/restore_selected def restore_selected Bvfs.new(@client, session[:job_ids]).restore_selected_files(params[:files], params[:location]) session.delete(:job_ids) flash[:success] = "Restore job issued successfully, files will be soon available in #{params[:location]}" redirect_to client_path(@client) end # GET /clients/1/tree?id=1 def tree @client = Client.for_user(current_user.id).find(params[:client_id]) bvfs = Bvfs.new(@client, session[:job_ids]) pathid = params[:id].to_i if pathid.nonzero? bvfs.fetch_dirs(pathid) else bvfs.fetch_dirs end tree = bvfs.extract_dir_id_and_name.map do |id, name| { id: id, text: name, state: { checkbox_disabled: true }, children: true } end if pathid.nonzero? bvfs.fetch_files(pathid) bvfs.extract_file_id_and_name.each do |id, name| tree << { id: id, text: name, type: 'file' } end end render json: tree end private + def require_non_blocked_client + if @client.host.blocked? + flash[:error] = 'Client disabled by admins' + redirect_to clients_path + end + end + def fetch_client @client = Client.for_user(current_user.id).find(params[:id]) @client_ids = [@client.id] end def fetch_jobs_info @stats = JobStats.new(@client_ids) end def get_charts @job_status = ChartGenerator.job_statuses(@client_ids, days_ago) @job_stats = ChartGenerator.job_stats(@client_ids, days_ago - 1) end def fetch_restore_point if params['restore_time(4i)'].blank? || params['restore_time(5i)'].blank? || params[:restore_date].blank? return nil end restore_point = "#{params[:restore_date]} #{params['restore_time(4i)']}:#{params['restore_time(5i)']}:00" begin DateTime.strptime(restore_point, '%Y-%m-%d %H:%M:%S') rescue return nil end restore_point end end diff --git a/app/controllers/jobs_controller.rb b/app/controllers/jobs_controller.rb index ae6ea48..b492527 100644 --- a/app/controllers/jobs_controller.rb +++ b/app/controllers/jobs_controller.rb @@ -1,89 +1,97 @@ class JobsController < ApplicationController before_action :require_logged_in before_action :fetch_host, only: [:new, :edit, :show, :create, :update, :destroy, :toggle_enable, :backup_now] before_action :fetch_job, only: [:show, :edit, :update, :destroy, :toggle_enable, :backup_now] + before_action :require_non_blocked_client # GET /jobs def new @job = @host.job_templates.new @schedule = @host.schedules.find(params[:schedule_id]) if params[:schedule_id] end # POST /jobs def create @job = @host.job_templates.new(fetch_params) if @job.save flash[:success] = 'Job created successfully' redirect_to host_path(@host) else render :new end end # GET /jobs/1 def show; end # GET /jobs/1/edit def edit @fileset = @host.filesets.find(params[:fileset_id]) if params[:fileset_id] @schedule = @host.schedules.find(params[:schedule_id]) if params[:schedule_id] end # PUT /jobs/1 def update if @job.update_attributes(fetch_params) flash[:success] = 'Job updated' redirect_to host_job_path(@host, @job) else render :edit end end # DELETE /jobs/1 def destroy if @job.destroy flash[:success] = 'Job deleted' @host.recalculate redirect_to host_path(@host) else flash[:error] = 'Job not deleted' redirect_to host_job_path(@host, @job) end end # PATCH /hosts/1/jobs/1/enable def toggle_enable @job.enabled = !@job.enabled @job.save flash[:success] = @job.enabled? ? 'Job enabled' : 'Job disabled' redirect_to host_path(@host, anchor: :tab_jobs) end # POST /hosts/1/jobs/1/backup_now def backup_now if @job.backup_now flash[:success] = 'Backup directive was sent to bacula. Backup will be taken in a while' else flash[:error] = 'Backup was not sent, try again later' end redirect_to client_path(@host.client) end private def fetch_job @job = @host.job_templates.find(params[:id]) end def fetch_host @host = current_user.hosts.find(params[:host_id]) end + def require_non_blocked_client + if @host.blocked? + flash[:error] = 'Client disabled by admins' + redirect_to clients_path + end + end + def fetch_params params.require(:job_template). permit(:name, :fileset_id, :schedule_id, :client_before_run_file, :client_after_run_file) end end