187 lines
7.6 KiB
Bash
Executable File
187 lines
7.6 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
__isint='^[0-9]+$'
|
|
__isvalidstr='^[a-z0-9]+$'
|
|
__isvalidaddr='^[a-z]+$|^((25[0-5]|2[0-4]?[0-9]|1[0-9]?[0-9]|[3-9][0-9]|[0-9])\.){3}(25[0-5]|2[0-4]?[0-9]|1[0-9]?[0-9]|[3-9][0-9]|[0-9])$'
|
|
__isJSONfile='^[~_\-.\/a-zA-Z0-9]+(\.json)$'
|
|
|
|
create_template() {
|
|
curl -H "Content-Type: application/json" -X PUT "$1:$2/_index_template/papers_t" -d @"$3"
|
|
}
|
|
|
|
#=============================================================================================================================================================================
|
|
|
|
create_index() {
|
|
curl -X PUT "$1:$2/papers-$3?pretty"
|
|
}
|
|
|
|
#=============================================================================================================================================================================
|
|
|
|
upload_files() {
|
|
_file="$4"
|
|
_dir="$5"
|
|
|
|
if [[ -n "$_file" ]]; then
|
|
curl -X POST "$1:$2/papers-$3/_doc/" \
|
|
-H "Content-Type: application/json" \
|
|
-d @"$_file"
|
|
elif [[ -n "$_dir" ]]; then
|
|
for file in "$_dir"/*; do
|
|
|
|
curl -X POST "$1:$2/papers-$3/_doc/" \
|
|
-H "Content-Type: application/json" \
|
|
-d @"$file"
|
|
done
|
|
fi
|
|
}
|
|
|
|
#=============================================================================================================================================================================
|
|
|
|
__usage="
|
|
Usage: $(basename "$0")
|
|
#-------------------------------------------------------------------------#
|
|
| *MODES* |
|
|
| |
|
|
| Note: 2 modes cannot be specified in one call. |
|
|
| |
|
|
| -c, --create-only | Skip template initialization and only create |
|
|
| | specified index. Result index name will be |
|
|
| | 'papers-{specified name}' |
|
|
| |
|
|
| -u, --updload | Uploads the specified file(s) to specified index |
|
|
|-------------------------------------------------------------------------|
|
|
| *OPTIONS* |
|
|
| |
|
|
| -h | Help information |
|
|
| |
|
|
| -a | Specifies the address |
|
|
| |
|
|
| -p, --port | Specifies the port |
|
|
| |
|
|
| -i, --index-name | Specifies the index name: |
|
|
| | Must be lowercase, cannot include [\/*?\"<>| ,#:], |
|
|
| | cannot start with [.-_+], cannot be \".\" or \"..\" |
|
|
| | cannot be longer than 255 bytes (note: multi-byte |
|
|
| | characters will count towards the limit faster) |
|
|
| | Result index name will be 'papers-{specified name}' |
|
|
| |
|
|
| -f, --file | Specify a JSON file that either a config or a |
|
|
| | document |
|
|
| |
|
|
| -d, --dir | Specify a directory containing documents to be |
|
|
| | uploaded to the index |
|
|
| |
|
|
#-------------------------------------------------------------------------#
|
|
"
|
|
#=============================================================================================================================================================================
|
|
|
|
CTRL=0
|
|
|
|
if [[ "$1" == "-h" ]]; then
|
|
echo "$__usage"
|
|
elif [[ -z "$1" ]]; then
|
|
echo "Use -h or --help for usage information."
|
|
else
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
-p | --port)
|
|
if [[ -n "$2" && $2 =~ $__isint && "$2" -ge 1 && "$2" -le 65535 ]]; then
|
|
PORT="$2"
|
|
shift
|
|
else
|
|
echo "Invalid port number!"
|
|
fi
|
|
;;
|
|
-a | --address)
|
|
if [[ -n "$2" ]]; then
|
|
IP="$2"
|
|
shift
|
|
else
|
|
echo "Address is not specified!"
|
|
fi
|
|
;;
|
|
-i | --index-name)
|
|
if [[ -n "$2" && $2 =~ $__isvalidstr ]]; then
|
|
IND="$2"
|
|
shift
|
|
else
|
|
echo "Index name is not specified!"
|
|
fi
|
|
;;
|
|
-f | --file)
|
|
if [[ true ]]; then
|
|
FPATH="$2"
|
|
shift
|
|
else
|
|
echo "Invalid file name!"
|
|
fi
|
|
;;
|
|
-d | --dir)
|
|
if [[ -n "$2" && -d "$2" ]]; then
|
|
DIRPATH="$2"
|
|
shift
|
|
fi
|
|
;;
|
|
-c | --create-only)
|
|
if [[ CTRL -ne 0 ]]; then echo "Incorrect use of modes. Use -h or --help.";
|
|
exit;
|
|
fi
|
|
CTRL=2
|
|
;;
|
|
-u | --upload)
|
|
if [[ CTRL -ne 0 ]]; then echo "Incorrect use of modes. Use -h or --help.";
|
|
exit;
|
|
fi
|
|
CTRL=3
|
|
;;
|
|
-*)
|
|
echo "Option '$1' is not supported"
|
|
exit
|
|
;;
|
|
*)
|
|
if [[ $1 =~ $__isvalidaddr ]]; then
|
|
IP="$1"
|
|
elif [[ $1 =~ $__isint && "$1" -ge 1 && "$1" -le 65535 ]]; then
|
|
PORT="$1"
|
|
elif [[ $1 =~ $__isvalidstr ]]; then
|
|
IND="$1"
|
|
elif [[ $1 =~ $__isJSONfile ]]; then
|
|
FPATH="$1"
|
|
else
|
|
echo "Invalid argument!"
|
|
exit;
|
|
fi
|
|
;;
|
|
esac
|
|
shift
|
|
done
|
|
|
|
echo "Specified: $IP:$PORT | Index name: $IND"
|
|
if [[ -z "$IP" || -z "$PORT" || -z "$IND" ]]; then
|
|
echo "Failed to identify target!"
|
|
exit
|
|
fi
|
|
|
|
case $CTRL in
|
|
0) # Default behaviour - full initialization (template creation and index creation)
|
|
if [[ -z "$FPATH" ]]; then
|
|
echo "Template config-file was not specified."
|
|
fi
|
|
create_template "$IP" "$PORT" "$FPATH"
|
|
echo -e "\nElasticsearch index template created"
|
|
create_index "$IP" "$PORT" "$IND"
|
|
echo -e "\nElasticsearch index (papers-$IND) created"
|
|
exit
|
|
;;
|
|
2) # Create index, skip creating the template
|
|
create_index "$IP" "$PORT" "$IND"
|
|
echo -e "\nElasticsearch index (papers-$IND) created"
|
|
exit
|
|
;;
|
|
3) # Uploads the specified file(s) to specified index
|
|
upload_files "$IP" "$PORT" "$IND" "$FPATH" "$DIRPATH"
|
|
echo -e "\nFinished uploading to index (papers-$IND)!"
|
|
exit
|
|
;;
|
|
esac
|
|
fi |