X-Git-Url: https://git.danieliu.xyz/?p=pb.git;a=blobdiff_plain;f=pb;h=3bc32e98e517587a0eb7181828a06fce90a4993c;hp=0f87d74386756b9c953377bf09d10b245216c124;hb=45bbbdd54e52f76ac3d9c1a03b807f9337d0a559;hpb=2bd0312dbe43e71cb47b06f851b1168054a325de

diff --git a/pb b/pb
index 0f87d74..3bc32e9 100755
--- a/pb
+++ b/pb
@@ -14,19 +14,16 @@ index_template="index_entry.html"
 rolling_template="rolling_entry.html"
 rss_template="rss_entry.html"
 
-start_token="<!-- BLOG START -->"
-end_token="<!-- BLOG END -->"
-
 [ ! -z "$EDITOR" ] && EDITOR="vim"
 
 init() {
-    read -p "Initialize blog? [y/n] " ask
+    read -p "Initialize blog here? [y/n] " ask
     [ "$ask" != "y" ] && exit 0
 
     mkdir -p "$data_dir/drafts" "$data_dir/published" "$data_dir/html" "$data_dir/templates" 
 
-    echo '<p id="{{TITLE}}">{{TITLE}}</p>' >> "$data_dir/templates/$index_template"
-    echo -e '<div id="{{TITLE}}">\n<h2>{{TITLE}}</h2>\n<p>{{DATE}}</p></div>' >> "$data_dir/templates/$rolling_template"
+    echo '<p>{{TITLE}}</p>' >> "$data_dir/templates/$index_template"
+    echo -e '<div>\n<h2>{{TITLE}}</h2>\n<p>{{DATE}}</p>\n<p>{{BODY}}</p>\n</div>' >> "$data_dir/templates/$rolling_template"
     echo -e '<item>\n<title>{{TITLE}}</title>\n<link></link>\n<description><\description>\n<\item>' \
         >> "$data_dir/templates/$rss_template"
 
@@ -36,82 +33,80 @@ init() {
 refresh() {
 
     read -p "Are you sure you want to refresh? [y/n] " ask
-    [ "$ask" != "y" ] && exit 0
+    [ "$ask" != "y" ] && echo "Aborting..." && exit 0
 
     # delete everything between tokens (remove dupe code)
-    sed -i "/$start_token/,/$end_token/{/$start_token/!{/$end_token/!d}}" "$blog_index_file"
-    sed -i "/$start_token/,/$end_token/{/$start_token/!{/$end_token/!d}}" "$rolling_file"
-    sed -i "/$start_token/,/$end_token/{/$start_token/!{/$end_token/!d}}" "$rss_file"
+    echo -e "$blog_index_file\n$rolling_file\n$rss_file" | xargs sed -i "/<!-- BLOG START -->/,/<!-- BLOG END -->/{/<!-- BLOG START -->/!{/<!-- BLOG END -->/!d}}"
 
     # deletes all html files and republishes all published files
-    echo "Refreshed."
+    echo "Successfully refreshed."
 }
 
 new() {
     [ -z "$1" ] && echo "Please give your blog post a name (you should put it inside quotations)" && exit 1 
-
-    # sanitize input
-    sanitized=`echo -n "$1" | sed -e 's/[^A-Za-z0-9 _-]//g'| sed -e 's/ /-/g'`
-
-    # open in editor
+    sanitized=`echo -n "$1" | sed -e 's/[^A-Za-z0-9 _-]//g'`
+    [ -f "$data_dir/drafts/$sanitized.draft.html" ] && echo "Blog of that name already exists." && exit 1
     $EDITOR "$data_dir/drafts/$sanitized.draft.html"
 }
 
 sub() {
     cat - |\
-        sed "s/{{TITLE}}/$1/g;
-            s/{{DATE}}/`date +'%a, %b %d %H:%M'`/g;
-            s/{{URL}}/$website_url\\/$1/g" |\
+        sed "1i <!-- ID:$1 START -->" |\
+        sed  "\$a <!-- ID:$1 END -->" |\
+        sed "s|{{TITLE}}|$1|g;
+            s|{{DATE}}|`date +'%a, %b %d %H:%M'`|g;
+            s|{{URL}}|$website_url/$1|g" |\
         sed "/{{BODY}}/r $data_dir/drafts/$1" |\
         sed "/{{BODY}}/d" 
 }
 
+# $1 is directory
+choose() { # working on abstraction
+    options=`ls -1 "$1" | sed 's/\.draft\.html$//;s/\.html$//'` 
+    [ -z "$options" ] && echo "No drafts to publish" && exit 0
+    echo "$options" | nl
+    read -p 'Choose an entry by number > ' choice
+    chosen=`ls -1 "$1" | sed -n "$choice p"`
+    [ -z "$chosen" ] && echo "Invalid choice" && exit 1
+}
+
 publish() {
     
-    drafts=`ls -1 "$data_dir/drafts" | sed -e 's/\.draft\.html$//'`
-    [ -z "$drafts" ] && echo "No drafts to publish" && exit 0
-
-    echo "Select which draft to publish"
-    echo "$drafts" | nl 
-
-    read -p '> ' choice
-    to_publish=`ls -1 "$data_dir/drafts/" | sed -n "$choice p"`
-    to_publish=${to_publish%.draft.html}
-    [ -z "$to_publish" ] && echo "Invalid choice" && exit 1
+    choose "$data_dir/drafts"
+    to_publish=${chosen%.draft.html}
 
     cat $blog_template | sub "$to_publish" \
        > "$data_dir/html/$to_publish.html" 
 
-    # Add new entry to blog index (do something about indent??)
-    sed -i "/$start_token/ a\
-        <!-- ID:$to_publish START -->\n`cat "$data_dir/templates/$index_template" | sub "$to_publish"`\n<!-- ID:$to_publish END -->" "$blog_index_file"
+    temp_index="$(mktemp)" # probably bad idea
+    cat "$data_dir/templates/$index_template" | sub "$to_publish" >> $temp_index
+    temp_rolling="$(mktemp)" 
+    cat "$data_dir/templates/$rolling_template" | sub "$to_publish" >> $temp_rolling
+    temp_rss="$(mktemp)" 
+    cat "$data_dir/templates/$rss_template" | sub "$to_publish" >> $temp_rss
 
-    sed -i "/$start_token/ a\
-        <!-- ID:$to_publish START -->\n`cat "$data_dir/templates/$rolling_template" | sub "$to_publish"`\n<!-- ID:$to_publish END -->" "$rolling_file"
+    # Add new entry to blog index (do something about indent??)
+    sed -i "/<!-- BLOG START -->/r $temp_index" "$blog_index_file"
+    sed -i "/<!-- BLOG START -->/r $temp_rolling" "$rolling_file"
+    sed -i "/<!-- BLOG START -->/r $temp_rss" "$rss_file"
 
     mv "$data_dir/drafts/$to_publish.draft.html" "$data_dir/published/"
 
+    echo "Successfully published $to_publish"
+
 }
 
 delete() {
-    published=`ls -1 "$data_dir/published" | sed -e 's/\.draft\.html$//'`
-    [ -z "$published" ] && echo "No posts to delete" && exit 0
-
-    echo "Select which post to delete"
-    echo "$published" | nl 
-
-    read -p '> ' choice
-    to_delete=`ls -1 "$data_dir/published/" | sed -n "$choice p"`
-    to_delete=${to_delete%.draft.html}
-    [ -z "$to_delete" ] && echo "Invalid choice" && exit 1
+    choose "$data_dir/published"
+    to_delete=${chosen%.draft.html}
 
     mv "$data_dir/published/$to_delete.draft.html" "$data_dir/drafts/" &&\
         rm "$data_dir/html/$to_delete.html"
 
-    # remove entry from blog index 
-    sed -i "/<!-- ID:$to_delete START -->/,/<!-- ID:$to_delete END -->/ d" "$blog_index_file"
-    sed -i "/<!-- ID:$to_delete START -->/,/<!-- ID:$to_delete END -->/ d" "$rolling_file"
+    # remove entries from files 
+    echo -e "$blog_index_file\n$rolling_file\n$rss_file" | xargs sed -i "/<!-- ID:$to_delete START -->/,/<!-- ID:$to_delete END -->/ d"
 
+    echo "Successfully deleted $to_delete"
 }
 
 # check to see if all required files are present
@@ -121,10 +116,9 @@ delete() {
 
 
 # check if blog dir exists
-[ ! -d $data_dir ]  && init
+[ ! -d $data_dir ]  && init && exit
 
 case $1 in
-    i|init) init;;
     n|new) new "$2";;
     p|publish) publish;;
     d|delete) delete;;