bit of polish
[pb.git] / pb
1 #!/bin/sh
2 # pinosaur's blog script
3
4 data_dir="blog"
5 website_url="https://www.youtube.com/watch?v=oHg5SJYRHA0/" 
6 blog_index_file="blogindex.html"
7 rolling_file="rolling.html"
8 rss_file="rss.xml"
9 blog_template="template.html"
10 index_template="index_entry.html"
11 rolling_template="rolling_entry.html"
12 rss_template="rss_entry.html"
13
14 [ ! -z "$EDITOR" ] && EDITOR="vim"
15
16 init() {
17     read -p "Initialize blog here? [y/n] " ask
18     [ "$ask" != "y" ] && echo "Cancelled init" && exit 0
19
20     mkdir -p "$data_dir/drafts" "$data_dir/published" "$data_dir/html" "$data_dir/templates" "$data_dir/backups"
21     echo '<p>{{TITLE}}</p>' >> "$data_dir/templates/$index_template"
22     echo -e '<div>\n<h2>{{TITLE}}</h2>\n<p>{{DATE}}</p>\n<p>{{BODY}}</p>\n</div>' >> "$data_dir/templates/$rolling_template"
23     echo -e '<item>\n<title>{{TITLE}}</title>\n<link></link>\n<description><\description>\n<\item>' \
24         >> "$data_dir/templates/$rss_template"
25
26     echo "Successfully created blog files."
27 }
28
29 backup() {
30     backup_name=`mktemp --tmpdir="$data_dir/backups" -d "backup_$(date +'%b-%d')_XXX"`
31     echo "Creating backup, will be placed in $backup_name"
32     cp -r "$data_dir/drafts/" "$backup_name"
33     cp -r "$data_dir/published/" "$backup_name"
34 }
35
36 new() {
37     [ -z "$1" ] && echo "Please give your blog post a name (you should put it inside quotations)" && exit 1 
38     sanitized=`echo -n "$1" | sed -e 's/[^A-Za-z0-9 _-]//g'`
39     [ -f "$data_dir/drafts/$sanitized.draft.html" ] && echo "Blog of that name already exists." && exit 1
40     $EDITOR "$data_dir/drafts/$sanitized.draft.html"
41 }
42
43 sub() {
44     cat - |\
45         sed "1i <!-- ID:$1 START -->" |\
46         sed  "\$a <!-- ID:$1 END -->" |\
47         sed "s|{{TITLE}}|$1|g;
48             s|{{DATE}}|`date +'%a, %b %d %H:%M'`|g;
49             s|{{URL}}|$website_url/$1|g" |\
50         sed "/{{BODY}}/r $data_dir/drafts/$1" |\
51         sed "/{{BODY}}/d" 
52 }
53
54 # $1 is directory
55 choose() { 
56     options=`ls -1 "$1" | sed 's/\.draft\.html$//;s/\.html$//'` 
57     [ -z "$options" ] && echo "No drafts to publish" && exit 0
58     echo "$options" | nl
59     read -p 'Choose an entry by number > ' choice
60     chosen=`ls -1 "$1" | sed -n "$choice p"`
61     [ -z "$chosen" ] && echo "Invalid choice" && exit 1
62 }
63
64 publish() {
65     choose "$data_dir/drafts"
66     to_publish=${chosen%.draft.html}
67
68     cat $blog_template | sub "$to_publish" > "$data_dir/html/$to_publish.html" 
69
70     # make this part less horrendous
71     temp_index="$(mktemp)" 
72     cat "$data_dir/templates/$index_template" | sub "$to_publish" >> $temp_index
73     temp_rolling="$(mktemp)" 
74     cat "$data_dir/templates/$rolling_template" | sub "$to_publish" >> $temp_rolling
75     temp_rss="$(mktemp)" 
76     cat "$data_dir/templates/$rss_template" | sub "$to_publish" >> $temp_rss
77
78     # Add new entry to blog index (do something about indent??)
79     sed -i "/<!-- BLOG START -->/r $temp_index" "$blog_index_file"
80     sed -i "/<!-- BLOG START -->/r $temp_rolling" "$rolling_file"
81     sed -i "/<!-- BLOG START -->/r $temp_rss" "$rss_file"
82
83     mv "$data_dir/drafts/$to_publish.draft.html" "$data_dir/published/"
84     echo "Successfully published $to_publish"
85 }
86
87 delete() {
88     choose "$data_dir/published"
89     to_delete=${chosen%.draft.html}
90
91     mv "$data_dir/published/$to_delete.draft.html" "$data_dir/drafts/" &&\
92         rm "$data_dir/html/$to_delete.html"
93
94     # remove entries from files 
95     echo -e "$blog_index_file\n$rolling_file\n$rss_file" | xargs sed -i "/<!-- ID:$to_delete START -->/,/<!-- ID:$to_delete END -->/ d"
96
97     echo "Successfully deleted $to_delete"
98 }
99
100 # check to see if all required files are present
101 [ -f $blog_index_file ] && [ -f $rolling_file ] && [ -f $blog_template ] && [ -f $rss_file ] || { echo "You are missing a file, please check that you have $blog_index_file, $template_file, $rolling_file and $rss_file in your home directory" && exit 1; }
102
103 # check if blog dir exists
104 [ ! -d $data_dir ]  && init && exit 0
105
106 case $1 in
107     n|new) new "$2";;
108     p|publish) publish;;
109     d|delete) delete;;
110     b|backup) backup;;
111     h|*) echo -e "=-=-=-=-=-=-= Pb =-=-=-=-=-=-=\nAvailable commands:\nn - new blog post\np - publish existing blog post\nd - deletes published post\nb - creates a backup";;
112 esac