include last csv column
[ink-generator.git] / generator.sh
1 #!/bin/bash
2
3 #   Generator - a Inkscape extension to generate end-use files from a model
4 #   Copyright (C) 2008  Aurélio A. Heckert
5 #
6 #   Added CSV parser from
7 #   http://fixunix.com/unix/83523-how-do-i-read-split-derive-csv-using-bash-script.html
8 #   by Chris F.A. Johnson <cfaj@freeshell.org> http://cfaj.freeshell.org
9 #
10 #   This program is free software: you can redistribute it and/or modify
11 #   it under the terms of the GNU General Public License as published by
12 #   the Free Software Foundation, either version 3 of the License, or
13 #   (at your option) any later version.
14 #
15 #   This program is distributed in the hope that it will be useful,
16 #   but WITHOUT ANY WARRANTY; without even the implied warranty of
17 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18 #   GNU General Public License for more details.
19
20 #   Version 0.4
21
22 # Collect all args and create local variables:
23 for arg in "$@"; do
24   if ( echo -n $arg | grep -q '^--.*=' ); then
25     key="$( echo -n "$arg" |
26             sed 's/^--\([^=]*\)=.*$/\1/; s/[^a-zA-Z0-9]/_/' )"
27     val="$( echo -n "$arg" |
28             sed 's/^--\([^=]*\)=\(.*\)$/\2/' )"
29     eval "$key=\"$val\""
30   else
31     svg_file="$arg"
32   fi
33 done
34
35 if ! zenity --version 2>/dev/null; then
36   function zenity() {
37     echo -e "\n$@\n" >&2
38   }
39   zenity "You must to install Zenity to have a better interaction
40 with this script, but it will work anyway."
41 fi
42
43 if [ "${data_file:0:1}" != "/" ]; then
44   data_file="$(pwd)/$data_file"
45 fi
46
47 if ! test -f "$data_file"; then
48   zenity --error --title="User Information Error" \
49          --text="The CSV file \"$data_file\" was not found.
50
51 Please, give the right name, or give the full path for the CSV file."
52   exit 1
53 fi
54
55 #function sep_line {
56 #  sed 's/^\([^"]*\("[^"]*"[^"]*\)*"[^"]*\),/\1#COMMAHACK#/g' |
57 #  sed 's/,/\n/g' | sed "s/\"\"/\"/g; s/^['\"]\|['\"]$//g"
58 #}
59
60 function csv_split() { ## USAGE: csv_split CSV_RECORD
61         local record=${1%"${CR}"}
62         local right
63         local vnum=0
64         unset values
65         while [ -n "$record" ]
66         do
67
68                 case $record in
69                         \"*) right=${record#*\",}
70                                 value=${record%%\",*}
71                                 values[$vnum]=${value#\"}
72                                 ;;
73                 *) values[$vnum]=${record%%,*}
74                         right=${record#*,}
75                         ;;
76                 esac
77
78                 case $record in
79                         *,*) record=${right} ;;
80                         *) record=${record#\"}
81                                 values[$vnum]=${record%\"}
82                                 break;;
83                 esac
84
85                 echo ${values[$vnum]}
86
87                 vnum=$(( $vnum + 1 ))
88
89         done
90
91         echo $record
92 }
93
94 # Set column names:
95 eval "$(
96   col=0
97   csv_split `head --lines=1 "$data_file"` | 
98   while read name; do
99     let col++
100     if [ "$var_type" == "name" ]; then
101       echo col_name[$col]=$( echo $name | sed "s/[][ \$'\"]/_/g" )
102     else
103       echo col_name[$col]=$col
104     fi
105     echo tot_col=$col
106   done
107 )"
108
109 # '
110
111 eval "$(
112   echo "$extra_vars" | sed 's/|/\n/g' |
113   while read extra; do
114     key="$( echo "$extra" | sed 's/^.*=>\(.*\)$/\1/g' )"
115     for i in $( seq $tot_col ); do
116       if [ "${col_name[$i]}" = "$key" ]; then
117         echo extracol[$i]="'$( echo "$extra" | sed 's/^\(.*\)=>.*$/\1/g' )'"
118       fi
119     done
120   done
121 )"
122
123 if [ "$preview" = "true" ]; then
124   # debug:
125   txt=''
126   for i in $( seq $tot_col ); do
127     txt="$txt\n%VAR_${col_name[$i]}%"
128   done
129   for i in $( seq $tot_col ); do
130     [ "${extracol[$i]}" != "" ] && txt="$txt\n${extracol[$i]}"
131   done
132   zenity --info --title="Generator Variables" \
133          --text="The replaceable text, based on your configuration and on the CSV are:\n$txt"
134 fi
135
136 eval "output=\"$output\""
137
138 [ "$( dirname "$output" )" != "" ] && mkdir --parents "$( dirname "$output" )"
139
140 [ "$format" = "" ] && format=PDF
141 format=$( echo $format | tr a-z A-Z )
142
143 if ! ( echo "$output" | grep -qi "$format\$" ); then
144   if zenity --question --text="
145 Your output pattern has a file extension diferent from the export format.
146
147 Did you want to add the file extension?"; then
148     output="$output.$( echo $format | tr A-Z a-z )"
149   fi
150 fi
151
152 tmp_svg=$( mktemp )
153 tmp_png=$( mktemp )
154 ink_error=$( mktemp )
155
156 my_pid=$$
157 function the_end {
158   rm $tmp_svg $tmp_png $ink_error
159   kill $my_pid
160 }
161
162 function ink-generate {
163   f="$( echo "$1" | sed 's/^[^=]*=\(.*\)$/\1/' )"
164   rm "$f" 2>/dev/null
165   ( inkscape --without-gui \
166              "$1" --export-dpi="$dpi" \
167              $tmp_svg 2>&1 ) > $ink_error
168   if ! test -f "$f"; then
169     zenity --error --title="Inkscape Converting Error" \
170            --text="$(cat $ink_error |
171                      sed 's/&/&amp;/g; s/</&lt;/g; s/>/&gt;/g' )"
172     the_end
173     exit 1
174   fi
175 }
176
177 function show_preview {
178   case $format in
179     SVG)
180       ( inkview "$1" || inkscape "$1" ) 2>/dev/null ||
181       echo 'There is no visualizator for SVG' >&2 ;;
182     PDF)
183       ( evince "$1" || kpdf "$1" || xpdf "$1" || gs "$1" ) 2>/dev/null ||
184       echo 'There is no visualizator for PDF' >&2 ;;
185     PS)
186       ( evince "$1" || gs "$1" ) 2>/dev/null ||
187       echo 'There is no visualizator for PS'  >&2 ;;
188     EPS)
189       ( evince "$1" || gs "$1" ) 2>/dev/null ||
190       echo 'There is no visualizator for EPS' >&2 ;;
191     PNG)
192       ( eog "$1" || kview "$1" || display "$1" ) 2>/dev/null ||
193       echo 'There is no visualizator for PNG' >&2 ;;
194     JPG)
195       ( eog "$1" || kview "$1" || display "$1" ) 2>/dev/null ||
196       echo 'There is no visualizator for JPG' >&2 ;;
197   esac
198   echo 100
199   the_end
200 }
201
202 tot_lines=$( wc --lines "$data_file" | sed 's/^\([0-9]\+\).*/\1/' )
203 if [ "$var_type" == "name" ]; then
204   let tot_lines--
205 fi
206
207 cur_line=0
208
209 cat "$data_file" | (
210   [ "$var_type" == "name" ] && read cut_frist_line
211   while read line; do
212     col=0
213     replace="$(
214       csv_split "$line" |
215       while read val; do
216         let col++
217         echo -n "s/%VAR_${col_name[$col]}%/$(
218                  echo "$val" | sed "s/\//\\\\\//g; s/'/\´/g"
219                  )/g; " | sed 's/#COMMAHACK#/,/g'
220         if [ "${extracol[$col]}" != "" ]; then
221           echo -n "s/${extracol[$col]}/$(
222                    echo "$val" | sed "s/\//\\\\\//g; s/'/\´/g"
223                    )/g; " | sed 's/#COMMAHACK#/,/g'
224         fi
225       done
226     )"
227     eval "sed '$replace' '$svg_file' > $tmp_svg"
228     out_file="$( echo "$output" | sed "$replace" )"
229     #echo "Gerando $out_file ..."
230     case $format in
231       SVG)
232         cp "$tmp_svg" "$out_file" ;;
233       PDF)
234         ink-generate --export-pdf="$out_file" ;;
235       PS)
236         ink-generate --export-ps="$out_file" ;;
237       EPS)
238         ink-generate --export-eps="$out_file" ;;
239       PNG)
240         ink-generate --export-png="$out_file" ;;
241       JPG)
242         ink-generate --export-png="$tmp_png"
243         if ! ( echo "$output" | grep -qi '.jpe\?g$' ); then
244           output="$output.jpg"
245         fi
246         convert $tmp_png "$out_file" ;;
247     esac
248     let cur_line++
249     echo $(( ( $cur_line * 100 ) / $tot_lines ))
250     [ "$preview" = "true" ] && show_preview "$out_file" && exit 0
251   done |
252   zenity --progress --title="Generator" \
253          --text="Generating..." --auto-close --width=400
254 )
255
256 [ "$preview" = "true" ] && exit 0
257
258 the_end