forked from microsoft/kernel-memory
-
Notifications
You must be signed in to change notification settings - Fork 0
/
upload-file.sh
executable file
·112 lines (91 loc) · 2.58 KB
/
upload-file.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#!/usr/bin/env bash
# This script uploads a file to Kernel Memory web service
# from the command line, using curl
set -e
trap 'exitScript' ERR
# Show manual
help() {
cat <<-_EOF_
Help for Bash script
Usage:
./upload-file.sh -s <url> -f <file path> [-p <index name>] [-i <id>] [-t <tag1> -t <tag2> -t <tag3> (...)]
-s web service URL (required) Kernel Memory web service URL.
-f file path (required) Path to the document to upload.
-p index (optional) Index where to store memories.
-i document ID (optional) Unique identifier for the document uploaded.
-t "key:value" (optional) Key-Value tag. Multiple tags and values per tag can be set.
-h Print this help content.
Example:
./upload-file.sh -s http://127.0.0.1:9001 -f myFile.pdf -u me -t "type:notes" -t "type:test" -i "bash test"
For more information visit https://github.com/microsoft/kernel-memory
_EOF_
}
# Read command line parameters
readParameters() {
while [ "$1" != "" ]; do
case $1 in
-s)
shift
SERVICE_URL=$1
;;
-f)
shift
FILENAME=$1
;;
-p)
shift
INDEXNAME=$1
;;
-i)
shift
DOCUMENT_ID=$1
;;
-t)
shift
TAGS="$TAGS $1"
;;
*)
help
exitScript
;;
esac
shift
done
}
validateParameters() {
if [ -z "$SERVICE_URL" ]; then
echo "Please specify the web service URL"
exit 1
fi
if [ -z "$FILENAME" ]; then
echo "Please specify a file to upload"
exit 3
fi
if [ -d "$FILENAME" ]; then
echo "$FILENAME is a directory."
exit 3
fi
if [ ! -f "$FILENAME" ]; then
echo "$FILENAME does not exist."
exit 3
fi
}
# Remove variables and functions from the environment, in case the script was sourced
cleanupEnv() {
unset SERVICE_URL FILENAME INDEXNAME DOCUMENT_ID TAGS TAG CMD
unset -f help readParameters validateParameters cleanupEnv exitScript
}
# Clean exit for sourced scripts
exitScript() {
cleanupEnv
kill -SIGINT $$
}
readParameters "$@"
validateParameters
# Prepare curl command
CMD="curl -v -F 'file1=@\"${FILENAME}\"' -F 'index=\"${INDEXNAME}\"' -F 'documentId=\"${DOCUMENT_ID}\"' -F 'tags=\"${TAGS}\"'"
# Add URL
CMD="$CMD $SERVICE_URL/upload"
# Send HTTP request using curl
set -x
eval $CMD