@@ -50,7 +50,7 @@ run_upgrade (){
50
50
else
51
51
echo " Upgrading versions: $1 " ;
52
52
fi
53
- arr=$@ ;
53
+ arr=( " $@ " ) ;
54
54
for i in ${1// ;/ }
55
55
do
56
56
if [[ $2 == " fs" ]]
@@ -63,7 +63,7 @@ run_upgrade (){
63
63
continue
64
64
fi
65
65
fi
66
- bash " $DIR /../upgrade/$i /upgrade_fs.sh" ;
66
+ # bash "$DIR/../upgrade/$i/upgrade_fs.sh";
67
67
else
68
68
echo " No filesystem upgrade script provided for $i " ;
69
69
fi
@@ -101,6 +101,11 @@ countly_upgrade (){
101
101
arr=(" $@ " );
102
102
if [[ " ${arr[*]} " == * " -y " * ]]; then
103
103
y=" -y" ;
104
+ for arg do
105
+ shift
106
+ [ " $arg " = " -y" ] && continue
107
+ set -- " $@ " " $arg "
108
+ done
104
109
fi
105
110
countly_root ;
106
111
if [ $# -eq 0 ]
@@ -183,8 +188,8 @@ countly_upgrade (){
183
188
(cd " $DIR /../.." ;
184
189
tar xaf countly-enterprise-edition* .tar.gz --strip=1 countly; )
185
190
186
- EE_PLUGINS=$( cat " $DIR /../../plugins/plugins.ee.json" | sed ' s/\"//g ' | sed ' s/\[//g' | sed ' s/\]//g' )
187
- CE_PLUGINS=$( cat " $DIR /../../plugins/plugins.ce.json" | sed ' s/\"//g ' | sed ' s/\[//g' | sed ' s/\]//g' )
191
+ EE_PLUGINS=$( sed ' s/\"//g ' " $DIR /../../plugins/plugins.ee.json" | sed ' s/\[//g' | sed ' s/\]//g' )
192
+ CE_PLUGINS=$( sed ' s/\"//g ' " $DIR /../../plugins/plugins.ce.json" | sed ' s/\[//g' | sed ' s/\]//g' )
188
193
PLUGINS_DIFF=$( echo " ${EE_PLUGINS} , ${CE_PLUGINS} " | tr ' ,' ' \n' | sort | uniq -u)
189
194
echo " Enabling plugins..."
190
195
for plugin in $PLUGINS_DIFF ; do
@@ -297,8 +302,13 @@ countly_backupdb (){
297
302
(mkdir -p " $1 " ;
298
303
cd " $1 " ;
299
304
echo " Backing up mongodb..." ;
300
- mongodump $( node $DIR /scripts/db.conf.js countly) > /dev/null;
301
- mongodump $( node $DIR /scripts/db.conf.js countly_drill) > /dev/null;
305
+ shift
306
+ # allow passing custom flags
307
+ connection=( $( node " $DIR /scripts/db.conf.js" ) " ${@ } " );
308
+ mongodump " ${connection[@]} " --db countly > /dev/null;
309
+ mongodump " ${connection[@]} " --db countly_drill > /dev/null;
310
+ mongodump " ${connection[@]} " --db countly_fs > /dev/null;
311
+ mongodump " ${connection[@]} " --db countly_out > /dev/null;
302
312
)
303
313
}
304
314
@@ -333,13 +343,13 @@ countly_save (){
333
343
if [ -f " $1 " ]
334
344
then
335
345
match=false
336
- files=$( ls " $2 " | wc -l )
346
+ files=$( find " $2 " -maxdepth 1 -type f -printf x | wc -c )
337
347
338
- if [ $files -gt 0 ]
348
+ if [ " $files " -gt 0 ]
339
349
then
340
350
for d in $2 /* ; do
341
- diff=$( diff $1 $d | wc -l)
342
- if [ $diff == 0 ]
351
+ diff=$( diff " $1 " " $d " | wc -l)
352
+ if [ " $diff " == 0 ]
343
353
then
344
354
match=true
345
355
break
@@ -348,7 +358,7 @@ countly_save (){
348
358
fi
349
359
350
360
files=$(( files+ 1 ))
351
- filebasename=$( basename $1 )
361
+ filebasename=$( basename " $1 " )
352
362
if [ " $match " == false ]
353
363
then
354
364
cp -a " $1 " " $2 /${filebasename} .backup.${files} "
@@ -401,7 +411,7 @@ countly_restorefiles (){
401
411
fi
402
412
403
413
for d in files/plugins/* ; do
404
- PLUGIN=$( basename $d ) ;
414
+ PLUGIN=$( basename " $d " ) ;
405
415
if [ -f " $d /config.js" ]; then
406
416
mkdir -p " $DIR /../../plugins/$PLUGIN " ;
407
417
cp " $d /config.js" " $DIR /../../plugins/$PLUGIN /config.js" ;
@@ -429,18 +439,35 @@ countly_restoredb (){
429
439
echo " Please provide path" ;
430
440
return 0;
431
441
fi
442
+ shift
443
+ # allow passing custom flags
444
+ connection=( $( node " $DIR /scripts/db.conf.js" ) " ${@ } " );
432
445
if [ -d " $1 /dump/countly" ]; then
433
446
echo " Restoring countly database..." ;
434
- mongorestore $( node $DIR /scripts/db.conf.js countly) --batchSize=10 " $1 /dump/countly" > /dev/null;
447
+ mongorestore " ${connection[@]} " --db countly --batchSize=10 " $1 /dump/countly" > /dev/null;
435
448
else
436
449
echo " No countly database dump to restore from" ;
437
450
fi
438
451
if [ -d " $1 /dump/countly_drill" ]; then
439
452
echo " Restoring countly_drill database..." ;
440
- mongorestore $( node $DIR /scripts/db.conf.js countly_drill) --batchSize=10 " $1 /dump/countly_drill" > /dev/null;
453
+ mongorestore " ${connection[@]} " --db countly_drill --batchSize=10 " $1 /dump/countly_drill" > /dev/null;
441
454
else
442
455
echo " No countly_drill database dump to restore from" ;
443
456
fi
457
+
458
+ if [ -d " $1 /dump/countly_fs" ]; then
459
+ echo " Restoring countly_fs database..." ;
460
+ mongorestore " ${connection[@]} " --db countly_fs --batchSize=10 " $1 /dump/countly_fs" > /dev/null;
461
+ else
462
+ echo " No countly_fs database dump to restore from" ;
463
+ fi
464
+
465
+ if [ -d " $1 /dump/countly_out" ]; then
466
+ echo " Restoring countly_out database..." ;
467
+ mongorestore " ${connection[@]} " --db countly_out --batchSize=10 " $1 /dump/countly_out" > /dev/null;
468
+ else
469
+ echo " No countly_out database dump to restore from" ;
470
+ fi
444
471
}
445
472
446
473
countly_restore (){
@@ -465,9 +492,9 @@ source "$DIR/enabled/countly.sh"
465
492
# process command
466
493
NAME=" $1 " ;
467
494
SCRIPT=" $2 " ;
468
- if [ -n " $( type -t countly_$1 ) " ] && [ " $( type -t countly_$1 ) " = function ]; then
495
+ if [ -n " $( type -t " countly_$1 " ) " ] && [ " $( type -t " countly_$1 " ) " = function ]; then
469
496
shift ;
470
- countly_${NAME} " $@ " ;
497
+ " countly_${NAME} " " $@ " ;
471
498
elif [ -f " $DIR /scripts/$NAME .sh" ]; then
472
499
shift ;
473
500
bash " $DIR /scripts/$NAME .sh" " $@ " ;
0 commit comments