From 9c318a05400766f5927a65c689e007210bcce8fd Mon Sep 17 00:00:00 2001
From: njp <n.pearman@gmail.com>
Date: Wed, 20 Dec 2017 12:59:41 +0100
Subject: [PATCH] Uses master post and URL to set @url and @name.

---
 lib/jekyll/amp_generate.rb | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/lib/jekyll/amp_generate.rb b/lib/jekyll/amp_generate.rb
index e69513e..5217732 100644
--- a/lib/jekyll/amp_generate.rb
+++ b/lib/jekyll/amp_generate.rb
@@ -1,13 +1,16 @@
 module Jekyll
   # Defines the base class of AMP posts
   class AmpPost < Page
-    def initialize(site, base, dir, post)
+    def initialize(site, base, dir, post, amp_root_dir)
       @site = site
       @base = base
       @dir = dir
       # Needed for posts with permalink
-      @url = dir
-      @name = 'index.html'
+      @url ||= URL.new({
+        :template     => File.join(amp_root_dir, post.url_template),
+        :placeholders => post.url_placeholders
+      }).to_s
+      @name = post.basename
       self.process(@name)
       self.read_yaml(File.join(base, '_layouts'), 'amp.html')
       self.content               = post.content
@@ -33,7 +36,7 @@ def generate(site)
       dir = site.config['ampdir'] || 'amp'
       site.posts.docs.each do |post|
         next if post.data['skip_amp'] == true
-        site.pages << AmpPost.new(site, site.source, File.join(dir, post.id), post)
+        site.pages << AmpPost.new(site, site.source, File.join(dir, post.id), post, dir)
       end
     end
   end