diff --git a/app/controllers/main_controller.rb b/app/controllers/main_controller.rb --- a/app/controllers/main_controller.rb +++ b/app/controllers/main_controller.rb @@ -39,7 +39,7 @@ redirect_to :action => 'list' end - def get_source + def source submission = Submission.find(params[:id]) if submission.user_id == session[:user_id] fname = submission.problem.name + '.' + submission.language.ext @@ -52,6 +52,16 @@ end end + def compiler_msg + @submission = Submission.find(params[:id]) + if @submission.user_id == session[:user_id] + render :action => 'compiler_msg', :layout => 'empty' + else + flash[:notice] = 'Error viewing source' + redirect_to :action => 'list' + end + end + def submission @user = User.find(session[:user_id]) @problems = Problem.find_available_problems