check if the link exists, redirects home if it doesn't

2023-10-07 13:38:10 -07:00 · 2023-10-07 13:38:10 -07:00 · f8499a79c2
parent b97ee3f3bb
commit f8499a79c2
1 changed files with 6 additions and 2 deletions
--- 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