include rackup

use puma for now

.gitignore

@@ -1 +1,5 @@
 *.sw*
+.bundle
+*.db
+.env.rb
+coverage

Gemfile

@@ -2,7 +2,6 @@
 
 source "https://rubygems.org"
 
-gem "falcon", "~> 0.42.3"
 
 gem "roda", "~> 3.72"
 
@@ -14,7 +13,7 @@ gem "sqlite3", "~> 1.6"
 
 gem "erubi", "~> 1.12"
 
-group :development do
+group :test do
 
   gem "cucumber", "~> 9.0"
 
@@ -24,4 +23,10 @@ group :development do
 
   gem "selenium-webdriver", "~> 4.13"
 
+  gem "simplecov"
+
 end
+
+gem "puma", "~> 6.4"
+
+gem "rackup", "~> 2.1"

Gemfile.lock

@@ -3,30 +3,7 @@ GEM
   specs:
     addressable (2.8.5)
       public_suffix (>= 2.0.2, < 6.0)
-    async (2.6.4)
-      console (~> 1.10)
-      fiber-annotation
-      io-event (~> 1.1)
-      timers (~> 4.1)
-    async-container (0.16.12)
-      async
-      async-io
-    async-http (0.60.2)
-      async (>= 1.25)
-      async-io (>= 1.28)
-      async-pool (>= 0.2)
-      protocol-http (~> 0.24.0)
-      protocol-http1 (~> 0.15.0)
-      protocol-http2 (~> 0.15.0)
-      traces (>= 0.10.0)
-    async-http-cache (0.4.3)
-      async-http (~> 0.56)
-    async-io (1.36.0)
-      async
-    async-pool (0.4.0)
-      async (>= 1.25)
     bigdecimal (3.1.4)
-    build-environment (1.13.0)
     builder (3.2.4)
     capybara (3.39.2)
       addressable
@@ -37,9 +14,6 @@ GEM
       rack-test (>= 0.6.3)
       regexp_parser (>= 1.5, < 3.0)
       xpath (~> 3.2)
-    console (1.23.2)
-      fiber-annotation
-      fiber-local
     cucumber (9.0.2)
       builder (~> 3.2, >= 3.2.4)
       cucumber-ci-environment (~> 9.2, >= 9.2.0)
@@ -65,52 +39,27 @@ GEM
     cucumber-messages (21.0.1)
     cucumber-tag-expressions (4.1.0)
     diff-lcs (1.5.0)
+    docile (1.4.0)
     erubi (1.12.0)
-    falcon (0.42.3)
-      async
-      async-container (~> 0.16.0)
-      async-http (~> 0.57)
-      async-http-cache (~> 0.4.0)
-      async-io (~> 1.22)
-      build-environment (~> 1.13)
-      bundler
-      localhost (~> 1.1)
-      openssl (~> 3.0)
-      process-metrics (~> 0.2.0)
-      protocol-rack (~> 0.1)
-      samovar (~> 2.1)
     ffi (1.16.2)
-    fiber-annotation (0.2.0)
-    fiber-local (1.0.0)
-    io-event (1.3.2)
-    localhost (1.1.10)
-    mapping (1.1.1)
     matrix (0.4.2)
     mini_mime (1.1.5)
     mini_portile2 (2.8.4)
     multi_test (1.1.0)
+    nio4r (2.5.9)
     nokogiri (1.15.4)
       mini_portile2 (~> 2.8.2)
       racc (~> 1.4)
-    openssl (3.2.0)
-    process-metrics (0.2.1)
-      console (~> 1.8)
-      samovar (~> 2.1)
-    protocol-hpack (1.4.2)
-    protocol-http (0.24.7)
-    protocol-http1 (0.15.1)
-      protocol-http (~> 0.22)
-    protocol-http2 (0.15.1)
-      protocol-hpack (~> 1.4)
-      protocol-http (~> 0.18)
-    protocol-rack (0.2.6)
-      protocol-http (~> 0.23)
-      rack (>= 1.0)
     public_suffix (5.0.3)
+    puma (6.4.0)
+      nio4r (~> 2.0)
     racc (1.7.1)
     rack (3.0.8)
     rack-test (2.1.0)
       rack (>= 1.3)
+    rackup (2.1.0)
+      rack (>= 3)
+      webrick (~> 1.8)
     regexp_parser (2.8.1)
     rexml (3.2.6)
     roda (3.72.0)
@@ -129,22 +78,24 @@ GEM
       rspec-support (~> 3.12.0)
     rspec-support (3.12.1)
     rubyzip (2.3.2)
-    samovar (2.2.0)
-      console (~> 1.0)
-      mapping (~> 1.0)
     selenium-webdriver (4.13.1)
       rexml (~> 3.2, >= 3.2.5)
       rubyzip (>= 1.2.2, < 3.0)
       websocket (~> 1.0)
     sequel (5.73.0)
       bigdecimal
+    simplecov (0.22.0)
+      docile (~> 1.1)
+      simplecov-html (~> 0.11)
+      simplecov_json_formatter (~> 0.1)
+    simplecov-html (0.12.3)
+    simplecov_json_formatter (0.1.4)
     sqlite3 (1.6.6)
       mini_portile2 (~> 2.8.0)
     sys-uname (1.2.3)
       ffi (~> 1.1)
     tilt (2.3.0)
-    timers (4.3.5)
+    webrick (1.8.1)
-    traces (0.11.1)
     websocket (1.2.10)
     xpath (3.2.0)
       nokogiri (~> 1.8)
@@ -156,11 +107,13 @@ DEPENDENCIES
   capybara (~> 3.39)
   cucumber (~> 9.0)
   erubi (~> 1.12)
-  falcon (~> 0.42.3)
+  puma (~> 6.4)
+  rackup (~> 2.1)
   roda (~> 3.72)
   rspec (~> 3.12)
   selenium-webdriver (~> 4.13)
   sequel (~> 5.72)
+  simplecov
   sqlite3 (~> 1.6)
   tilt (~> 2.2)
 

Jenkinsfile

@@ -0,0 +1,61 @@
+pipeline {
+  agent any
+  
+  environment {
+    APP_SESSION_SECRET = ''
+    DB_NAME = 'url_shortener.db'
+  }
+  stages {
+    stage('Init') {
+      steps {
+        sh 'rbenv local 3.2.2'
+        script {
+	  env.APP_SESSION_SECRET = sh(script: 'ruby secret.rb', returnStdout: true)
+        }
+        sh 'echo "ENV[\\\"APP_SESSION_SECRET\\\"] ||= $(ruby secret.rb)" > .env.rb'
+        sh 'echo "ENV[\\\"DB_NAME\\\"] ||= \\\"${DB_NAME}\\\"" >> .env.rb'
+        sh 'cat .env.rb'
+      }
+    }
+    stage('Build dependencies') {
+      steps {
+        sh 'bundle install'
+        sh 'sequel -m db/migrations sqlite://db/${DB_NAME}'
+      }
+    }
+    stage('Run tests') {
+      steps {
+        catchError(buildResult: 'UNSTABLE', stageResult: 'FAILURE') {
+	  sh 'cucumber features --format html --out cucumber.html'
+        }
+        catchError(buildResult: 'UNSTABLE', stageResult: 'FAILURE') {
+          sh 'rspec spec --format html --out spec.html'
+        }
+      }
+    }
+    stage('Report results') {
+      steps {
+        archive(includes: 'pkg/*.gem')
+        publishHTML (target: [
+          allowMissing: false,
+          alwaysLinkToLastBuild: false,
+          keepAll: true,
+          reportDir: '.',
+          reportFiles: 'cucumber.html, spec.html, coverage/index.html',
+          reportName: 'Test Results',
+          reportTitles: 'Cucumber Results, RSpec Results, Test Coverage'])
+      }
+    }
+  }
+  post {
+    success {
+      mattermostSend channel: 'git-messages', color: 'good', message: "[${JOB_NAME}](${JOB_URL}) [#${BUILD_NUMBER}](${BUILD_URL}) ([Gitea](${GIT_URL}))", text: 'Build Finished Successfully'
+    }
+    unstable {
+      mattermostSend channel: 'git-messages', color: 'warning', message: "[${JOB_NAME}](${JOB_URL}) [#${BUILD_NUMBER}](${BUILD_URL}) ([Gitea](${GIT_URL}))", text: "Build Unstable"
+    }
+    failure {
+      mattermostSend channel: 'git-messages', color: 'danger', message: "[${JOB_NAME}](${JOB_URL}) [#${BUILD_NUMBER}](${BUILD_URL}) ([Gitea](${GIT_URL}))", text: "Build Failed"
+    }
+  }
+}

README.md

@@ -5,18 +5,33 @@ Roda, Falcon, Sequel, and SQLite
 
 the point of this project is to quickly build something and work on continuous deployment while making small refinements to the functional pieces.
 
+the outside pieces of software that this project relies on are sqlite3 and pkgconf. 
+
+
 
 first you have to install the dependencies:
 
 `bundle install`
 
+
 if you want the development group included run this first:
 
-`bundle config set --local with 'development'
+`bundle config set --local with 'development'`
+
+
+then create a .env.rb file in the root directory that contains the following ENV attributes:
+
+```
+ENV["APP_SESSION_SECRET"] = {output of a random 64 byte secret}
+
+ENV["DB_NAME"] = {db file name}
+```
+
 
 after the dependencies are installed, you have to create the db
 
-`sequel -m db/migrations sqlite://db/url_shortener.db`
+`sequel -m db/migrations sqlite://db/{DB_NAME}`
+
 
 to start the application with Falcon:
 

app.rb

@@ -2,30 +2,102 @@ require 'roda'
 require 'securerandom'
 require 'json'
 require 'sequel'
+require 'open-uri'
 
 class App < Roda
-  DB = Sequel.sqlite('db/url_shortener.db')
+  plugin :sessions, secret: ENV.delete('APP_SESSION_SECRET')
-  links = DB[:links]
   plugin :render, escape: true
+  plugin :flash
+  plugin :json_parser
+  plugin :request_headers
+
+  DB = Sequel.sqlite("db/#{ENV['DB_NAME']}")
+  links = DB[:links]
+
   route do |r|
     r.root do
+      @message = flash['message'] || "Enter a URL"
       view :home
     end
 
     r.get String do | url_code |
-      link = links.filter(:code => url_code).first[:url]
+      link = links.filter(:code => url_code)
-      r.redirect link
+      r.redirect link.first[:url] unless link.first.nil?
+      @message = "Link #{url_code} doesn't exist"
+      response.status = 404
+      view :home
     end
 
     r.post "create" do
       url = r.params['url']
-      if url.empty? then r.redirect '/' end
+      if url.nil? or url.empty? 
-      if nil == links.filter(:url => url).first
+        flash['message'] = "Please enter a valid URL";
+        r.redirect '/'
+      end
+
+      begin
+        OpenURI.open_uri(url)
+      rescue URI::BadURIError
+        flash['message'] = "Invalid URL"
+        r.redirect '/'
+      rescue OpenURI::HTTPError
+        flash['message'] = "URL not found"
+        r.redirect '/'
+      rescue SocketError => e
+        flash['message'] = "URL does not resolve"
+        r.redirect '/'
+      end
+
+      if links.filter(:url => url).first.nil?
+        code = SecureRandom.urlsafe_base64 4
+        links.insert(url: url, code: code) 
+        @message = "Link created"
+      end
+      code = links.filter(:url => url).first[:code]
+      @message ||= "Link exists"
+      @new_link = 'http://' + request.env['HTTP_HOST'] + '/' + code
+      view :create   
+    end
+    r.on "links" do
+      r.post do
+        if 'application/json' != r.headers['CONTENT_TYPE']
+          response.status = 400
+          return {message: "not a valid json request"}.to_json
+        end
+
+        url = r.params['url']
+        if url.nil?
+          response.status = 400
+          return {message: "missing url parameter"}.to_json
+        end
+
+        if url.empty? 
+          response.status = 400
+          return {message: "invalid url parameter"}.to_json
+        end
+
+        begin
+          OpenURI.open_uri(url)
+        rescue URI::BadURIError
+          response.status = 400
+          return {message: "invalid url parameter"}.to_json
+        rescue OpenURI::HTTPError
+          response.status = 400
+          return {message: "url not found"}.to_json
+        rescue SocketError => e
+          response.status = 400
+          return {message: "url does not resolve"}.to_json
+        end
+
+        if links.filter(:url => url).first.nil?
           code = SecureRandom.urlsafe_base64 4
           links.insert(url: url, code: code) 
         end
-      @new_link = 'http://' + request.env['HTTP_HOST'] + '/' + links.filter(:url => url).first[:code]
+
-      view :create   
+        code = links.filter(:url => url).first[:code]
+        @new_link = 'http://' + request.env['HTTP_HOST'] + '/' + code
+        return {url: url, code: code, link: @new_link}.to_json
+      end
     end
   end
 end

config.ru

@@ -1,3 +1,4 @@
+require './.env'
 require './app'
 
 run App.app

features/homepage.feature

@@ -7,16 +7,6 @@ Feature: Homepage
   Scenario: Homepage Loads with a form
     Given I visit the "/" page
     Then I should see text "URL Shortener"
+    And I should see the message "Enter a URL"
     And I should see a form field "url"
     And I should see a "Submit" button
-
-  Scenario: Submitting the form without entering a URL
-    Given I visit the "/" page
-    When I click the "Submit" button
-    Then I should be on "/" page
-
-  Scenario: Submitting the form with a correct URL
-    Given I visit the "/" page
-    When I type "http://google.com/" in the "url" field
-    And I click the "Submit" button
-    Then I should be on "/create" page

features/step_definitions/steps.rb

@@ -1,9 +1,24 @@
+# BEFORE
+Before('@db-test') do
+  @links = Sequel.sqlite("db/#{ENV['DB_NAME']}")[:links]
+end
+
+
 # GIVEN
 
 Given('I visit the {string} page') do |string|
   visit string
 end
 
+Given('A link already exists with the url {string}') do |string|
+  @links.insert(url: string, code: "aaaaaa")
+end
+
+Given('A link already exists with the url {string} and code {string}') do |url, code|
+  @links.insert(url: url, code: code)
+end
+
+
 # WHEN
 
 When('I click the {string} button') do |string|
@@ -14,6 +29,11 @@ When('I type {string} in the {string} field') do |text, field|
   fill_in field, with: text
 end
 
+When('I visit the {string} location') do |string|
+  visit string
+end
+
+
 # THEN
 
 Then('I should see text {string}') do |string|
@@ -31,3 +51,24 @@ end
 Then('I should be on {string} page') do |string|
   page.should have_current_path string
 end
+
+Then('I should see the message {string}') do |message|
+  page.should have_selector '#message', text: message
+end
+
+Then('I should be redirected to {string}') do |string|
+  actual = URI.parse(current_url).to_s
+  location = actual.index(string)
+  location.should equal(0)
+end
+
+Then('The status code should be {int}') do |code|
+  page.status_code.should eq(code)
+end
+
+
+# AFTER
+
+After('@db-test') do
+  @links.delete
+end

features/submit.feature

@@ -0,0 +1,52 @@
+# features/submit.feature
+
+Feature: Submit
+
+  Submitting URL's on the homepage
+
+  Scenario: Submitting the form without entering a URL
+    Given I visit the "/" page
+    When I click the "Submit" button
+    Then I should be on "/" page
+    And I should see the message "Please enter a valid URL"
+
+  @db-test
+  Scenario: Submitting the form with a correct URL
+    Given I visit the "/" page
+    When I type "http://google.com" in the "url" field
+    And I click the "Submit" button
+    Then I should be on "/create" page
+    And I should see the message "Link created"
+
+  @db-test
+  Scenario: Submitting the form with an existing URL
+    Given I visit the "/" page
+    And A link already exists with the url "http://google.com"
+    When I type "http://google.com" in the "url" field
+    And I click the "Submit" button
+    Then I should be on "/create" page
+    And I should see the message "Link exists"
+
+  @db-test
+  Scenario: Submitting the form with a URL that is 404
+    Given I visit the "/" page
+    When I type "http://google.com/example" in the "url" field
+    And I click the "Submit" button
+    Then I should be on "/" page
+    And I should see the message "URL not found"
+
+  @db-test
+  Scenario: Submitting the form with a URL that does not resolve
+    Given I visit the "/" page
+    When I type "http://bad.tld" in the "url" field
+    And I click the "Submit" button
+    Then I should be on "/" page
+    And I should see the message "URL does not resolve"
+
+  @db-test
+  Scenario: Submitting the form with an invalid URL
+    Given I visit the "/" page
+    When I type "not-an-url" in the "url" field
+    And I click the "Submit" button
+    Then I should be on "/" page
+    And I should see the message "Invalid URL"

features/support/env.rb

@@ -1,11 +1,23 @@
+require 'simplecov'
+SimpleCov.start
+
+require_relative '../../.env'
+ENV["DB_NAME"] = "test_#{ENV["DB_NAME"]}"
 require_relative '../../app'
 require 'rubygems'
 require 'roda'
+require 'sequel'
 require 'capybara'
 require 'capybara/dsl'
 require 'rspec'
 
+# DB initialization
+Sequel.extension :migration
+Sequel.sqlite("db/#{ENV['DB_NAME']}") do |db|
+  Sequel::Migrator.apply(db, "db/migrations")
+end
 
+# attach app to Capybara
 Capybara.app = App
 
 include Capybara::DSL

features/visit.feature

@@ -0,0 +1,19 @@
+# features/visit.feature
+
+Feature: Visit
+
+Accessing a URL from it's link code
+
+  @db-test
+  Scenario: Accessing the URL based on the shortcode
+    Given A link already exists with the url "https://google.com" and code "aaaaaa"
+    When I visit the "/aaaaaa" location
+    Then I should be redirected to "https://google.com"
+
+  Scenario: Accessing a nonexistent URL code
+    Given I visit the "/aaaaaa" location
+    Then I should be on "/aaaaaa" page
+    And The status code should be 404
+    And I should see the message "Link aaaaaa doesn't exist"
+    And I should see a form field "url"
+    And I should see a "Submit" button

secret.rb

@@ -0,0 +1,2 @@
+require 'securerandom'
+puts SecureRandom.base64(64).inspect()

spec/create_spec.rb

@@ -0,0 +1,101 @@
+require 'simplecov'
+SimpleCov.start
+
+require_relative '../.env'
+ENV["DB_NAME"] = "test_#{ENV["DB_NAME"]}"
+require_relative '../app'
+require 'rubygems'
+require 'roda'
+require 'sequel'
+require 'rspec'
+require 'rack/test'
+
+# DB initialization
+Sequel.extension :migration
+Sequel.sqlite("db/#{ENV['DB_NAME']}") do |db|
+  Sequel::Migrator.apply(db, "db/migrations")
+end
+
+def app
+  App
+end
+
+describe "Submit API request to create new link" do
+  include Rack::Test::Methods
+  before :each do
+    @links = Sequel.sqlite("db/#{ENV['DB_NAME']}")[:links]
+  end
+  after :each do
+    @links.delete
+  end
+  it "should return link data in json format when a valid url is submitted" do
+    data = {
+      url: 'http://google.com'
+    }
+    post('/links', data.to_json, "CONTENT_TYPE" => "application/json")
+    expect(last_response).to be_ok
+    response_json = JSON.parse(last_response.body)
+    expect(response_json['url']).to eq(data[:url])
+    expect(response_json['code']).not_to eq(nil)
+    expect(response_json['link']).to include(response_json['code'])
+  end
+  
+  it "should return with a 400 status and 'invalid url parameter' message when an empty url is submitted" do
+    data = {
+      url: ''
+    }
+    post('/links', data.to_json, "CONTENT_TYPE" => "application/json")
+    expect(last_response.status).to eq(400)
+    response_json = JSON.parse(last_response.body)
+    expect(response_json['message']).to eq('invalid url parameter')
+  end
+
+  it "should return with a 400 status and 'missing url parameter' message when an empty url is submitted" do
+    data = {
+    }
+    post('/links', data.to_json, "CONTENT_TYPE" => "application/json")
+    expect(last_response.status).to eq(400)
+    response_json = JSON.parse(last_response.body)
+    expect(response_json['message']).to eq('missing url parameter')
+  end
+
+  it "should return with a 400 status and 'invalid url parameter' message when an invalid url is submitted" do
+    data = {
+      url: 'not-an-url'
+    }
+    post('/links', data.to_json, "CONTENT_TYPE" => "application/json")
+    expect(last_response.status).to eq(400)
+    response_json = JSON.parse(last_response.body)
+    expect(response_json['message']).to eq('invalid url parameter')
+  end
+
+  it "should return with a 400 status and 'url not found' message when a 404 url is submitted" do
+    data = {
+      url: 'http://google.com/example'
+    }
+    post('/links', data.to_json, "CONTENT_TYPE" => "application/json")
+    expect(last_response.status).to eq(400)
+    response_json = JSON.parse(last_response.body)
+    expect(response_json['message']).to eq('url not found')
+  end
+
+  it "should return with a 400 status and 'url not found' message when a URL with no DNS is submitted" do
+    data = {
+      url: 'http://bad.tld'
+    }
+    post('/links', data.to_json, "CONTENT_TYPE" => "application/json")
+    expect(last_response.status).to eq(400)
+    response_json = JSON.parse(last_response.body)
+    expect(response_json['message']).to eq('url does not resolve')
+  end
+
+  it "should return with a 400 status and 'not a valid json request' message when a request is made with the wrong content type header" do
+    data = {
+      url: 'http://google.com'
+    }
+    post('/links', data.to_json)
+    expect(last_response.status).to eq(400)
+    response_json = JSON.parse(last_response.body)
+    expect(response_json['message']).to eq('not a valid json request')
+  end
+end

views/create.erb

@@ -1,2 +1 @@
-<h2><a href="/" style="color: #555555;">Home</a></h2>
+<h3 style="overflow-wrap: break-word; word-wrap: break-word; margin-top: 3rem; line-height: 2rem;"><a href="<%= @new_link %>" target="_blank"><%= @new_link %></a></h3>
-<h3 style="overflow-wrap: break-word; word-wrap: break-word;"><a href="<%= @new_link %>" target="_blank" style="color: #555555;"><%= @new_link %></a></h3>

views/home.erb

@@ -1,9 +1,13 @@
 <form method="POST" action="/create">
+  <div style="margin-bottom: 1rem;">
     <input type="url"
       name="url"
       pattern="https?://.+"
       placeholder="http://www.example.com/"
-    style="padding: 0.5rem;"
+      autocomplete="off"
       required />
-  <button type="submit" style="padding: 0.5rem;">Submit</button>
+  </div>
+  <div>
+    <button type="submit">Submit</button>
+  </div>
 </form>

views/layout.erb

@@ -1,11 +1,107 @@
 <html>
   <head>
     <title><%= @page_title || "URL Shortener" %></title>
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <link rel="icon" href="data:,">
+    <style>
+      :root {
+        --white: #FEFEFE;
+        --shadow: #333;
+        --black: #555;
+        --darker-blue: #809DFF;
+        --dark-blue: #99b1ff;
+        --medium-blue: #B3C6FF;
+        --light-blue: #CCD8FF;
+        --lightest-blue: #E6EBFF;
+      }
+
+      body {
+        background: var(--white);
+        background-image: linear-gradient(to bottom right, var(--lightest-blue), var(--white), var(--white), var(--white), var(--lightest-blue));
+        font-family: Helvetica, sans-serif;
+        color: var(--black);
+        width: 100%;
+        height: 100%;
+        margin: 0;
+        padding: 5rem 0 0;
+        border: 0;
+        box-sizing: border-box;
+      }
+
+      main {
+        width: 30rem;
+        max-width: calc(100% - 1rem);
+        box-sizing: border-box;
+        border: 5px solid var(--medium-blue);
+        border-radius: 20px;
+        margin: auto;
+        padding: 1rem;
+        text-align: center;
+        min-height: 260px;
+        box-shadow: 3px 2px 10px var(--shadow);
+        background: var(--white);
+        overflow: hidden;
+      }
+
+      input, button {
+        width: 80%;
+      }
+
+      input {
+        padding: 1rem;
+        border-radius: 5px;
+        border: 3px solid var(--dark-blue);
+        outline: none;
+        transition: all .25s ease-in-out;
+        -moz-transition: all .25s ease-in-out;
+        -webkit-transition: all .25s ease-in-out;
+      }
+
+      input:focus, input:hover {
+        border: 3px solid var(--darker-blue);
+      }
+
+      button {
+        color: var(--white);
+        padding: 1rem;
+        border-radius: 5px;
+        border: 2px solid var(--darker-blue);
+        background: var(--dark-blue);
+        font-weight: bold;
+        transition: background .25s ease-in-out;
+        -moz-transition: background .25s ease-in-out;
+        -webkit-transition: background .25s ease-in-out;
+      }
+
+      button:hover {
+        background: var(--darker-blue);
+      }
+
+      button:active {
+        background: var(--light-blue);
+        color: var(--black);
+      }
+
+      a {
+        color: var(--black);
+        text-decoration: none;
+        border-bottom: 4px solid var(--white);
+        transition: all .25s ease-in-out;
+        -moz-transition: all .25s ease-in-out;
+        -webkit-transition: all .25s ease-in-out;
+      }
+
+      a:hover, a:active, a:focus {
+        color: var(--darker-blue);
+        border-bottom: 4px solid var(--darker-blue);
+      }
+
+    </style>
   </head>
-  <body style="background: #FEFEFE; font-family: Helvetica, sans-serif; color: #555555;">
+  <body style="">
-    <main style="max-width: 30rem; min-width: 18rem; border: 5px solid #999999; border-radius: 20px; margin: auto; padding: 1rem; margin-top: 5rem; text-align: center; min-height: 175px;">
+    <main>
-      <h1>URL Shortener</h1>
+      <h1><a href="/">URL Shortener</a></h1>
+      <h4 id="message"><%== @message %></h4>
       <%== yield %>
     </main>
   </body>