diff --git a/app.rb b/app.rb index 54554b6..44dae4c 100644 --- a/app.rb +++ b/app.rb @@ -18,8 +18,12 @@ class App < Roda end r.get String do | url_code | - link = links.filter(:code => url_code).first[:url] - r.redirect link + link = links.filter(:code => url_code) + if link.first.nil? + flash['message'] = "Link #{url_code} doesn't exist" + r.redirect '/' + end + r.redirect link.first[:url] end r.post "create" do