Fix field addition when using rst.

[pelican-mode.git] / pelican-mode.el

diff --git a/pelican-mode.el b/pelican-mode.el
index f82be8a..fe4eea0 100644 (file)
--- a/pelican-mode.el
+++ b/pelican-mode.el
@@ -49,12 +49,16 @@
             (t (error "Unsupported major mode %S" major-mode)))
     ""))
 
+(defun pelican-rst-title (title)
+  "Create a ReSt version of TITLE."
+  (concat title "\n" (make-string (string-width title) ?#) "\n\n"))
+
 (defun pelican-title (title)
   "Format a TITLE for the current document, according to major mode."
   (cond ((derived-mode-p 'markdown-mode)
          (pelican-field "title" title))
         ((derived-mode-p 'rst-mode)
-         (concat title "\n" (make-string (string-width title) ?#) "\n"))
+         (pelican-rst-title title))
         (t (error "Unsupported major mode %S" major-mode))))
 
 (defun pelican-header (title date status category tags slug)
@@ -99,13 +103,28 @@
 
 (defun pelican-set-field (field value)
   "Set FIELD to VALUE."
+  (interactive "sField: \nsValue: ")
   (save-excursion
     (goto-char 0)
     (if (re-search-forward (concat "^" (pelican-field field ".+*")) nil t)
         (replace-match (pelican-field field value))
+      (re-search-forward "#")
+      (forward-line 2)
       (re-search-forward "^$")
       (replace-match (pelican-field field value)))))
 
+(defun pelican-set-title (title)
+  "Set the title to TITLE."
+  (interactive "sTitle: ")
+  (if (pelican-is-markdown)
+      (pelican-set-field "title" title)
+    (save-excursion
+      (goto-char 0)
+      (let ((header (pelican-rst-title title)))
+        (if (looking-at ".*\n#+\n+")
+            (replace-match header)
+          (insert header))))))
+
 (defun pelican-update-date ()
   "Update a Pelican date header."
   (interactive)
@@ -208,7 +227,6 @@ to show buffer size and position in mode-line."
   :keymap pelican-keymap
   :group 'pelican)
 
-
 ;;;###autoload
 (add-hook 'markdown-mode-hook 'pelican-enable-if-site)