lib/jekyll_auth/jekyll_site.rb in jekyll-auth-2.1.0 vs lib/jekyll_auth/jekyll_site.rb in jekyll-auth-2.1.1

@@ -1,13 +1,15 @@
+# frozen_string_literal: true
+
 class JekyllAuth
   class JekyllSite < Sinatra::Base
     register Sinatra::Index
-    set :public_folder, File.expand_path("_site", Dir.pwd)
+    set :public_folder, File.expand_path(JekyllAuth.destination, Dir.pwd)
     use_static_index "index.html"
 
     not_found do
       status 404
-      four_oh_four = File.expand_path("_site/404.html", Dir.pwd)
+      four_oh_four = File.expand_path(settings.public_folder + "/404.html", Dir.pwd)
       File.read(four_oh_four) if File.exist?(four_oh_four)
     end
   end
 end