]> pd.if.org Git - zpackage/blob - zpm-merge
fix possible memory leak in uncompress
[zpackage] / zpm-merge
1 #!/bin/sh
2
3 # merge
4 # collects info from package file and copies it into another package file
5
6 # merge [ -d ZPMDB ] [-f $pkgfile] pkg
7 # -F include file content
8 # -S don't include script content
9
10 die() {
11         printf 'zpm-merge:' 1>&2
12         printf ' %s' "$@" 1>&2
13         printf '\n' 1>&2
14         exit 1
15 }
16
17 verbose=0
18 mergefiles=0
19 mergescripts=1
20 mergeall=1
21 update=0
22 remove_older=0
23 only_if_newer=0
24
25 target=${ZPMDB:=/var/lib/zpm/local.db}
26
27 # TODO option to only update if new
28 # new is hash is different or build time is later
29 # TODO option to merge all packages found in pkgfile
30 while getopts :f:vd:FSs:auon opt; do
31         case $opt in
32                 f) pkgfile="$OPTARG" ;;
33                 v) verbose=1 ;;
34                 d) target="$OPTARG" ;;
35                 F) mergefiles=1 ;;
36                 S) mergescripts=0 ;;
37                 s) newstatus="$OPTARG" ;;
38                 a) mergeall=1 ;;
39                 u) update=1 ;;
40                 o) remove_older=1 ;;
41                 n) only_if_newer=1 ;;
42                 *) echo 'zpm-merge unknown option' $OPTARG; exit 1 ;;
43         esac
44 done
45 shift $((OPTIND - 1))
46
47 # TODO test target
48 if [ ! -e "$target" ]; then
49         die "target file $target does not exist"
50 fi
51
52 zpm test -v "$target" || exit 1
53
54 if [ $# -eq 0 ]; then
55         mergeall=1
56 else
57         mergeall=0
58 fi
59
60 # TODO if pkgfile is specified, allow multiple packages as args
61 if [ $mergeall -eq 0 ]; then
62         pkgid=$1
63         shift
64
65         if [ -z "$pkgid" ]; then
66                 die "must specify pkgid"
67         fi
68
69         eval "$(zpm parse -E $pkgid)"
70
71         if [ -z "$pkgfile" ]; then
72                 pkgfile=$ZPM_PACKAGE_FILE
73         fi
74
75         # calculate package id, pkgfile, etc
76         # cases R = full package id, F = specified package file
77
78         # immediate error
79         # --- 000 error, must specify something
80         if [ -z "$release" ] && [ -z "$pkgfile" ]; then
81                 die must specify package file or complete package id
82         fi
83
84         # try to get from package file
85         if [ -z "$release" ]; then
86                 pkgid=$(zpm findpkg -f $pkgfile $pkgid)
87                 if [ -z "$pkgid" ]; then
88                         die cannot find package id
89                 fi
90                 eval "$(zpm parse -E $pkgid)"
91         fi
92
93         # --F 001 error, wouldn't know which pkgid to create, could derive from file?
94         if [ -z "$release" ]; then
95                 die must specify complete package id
96         fi
97
98         # set file from pkgid
99         # -R- 010 set file from pkgid, create in file, error if no file
100         if [ -z "$pkgfile" ]; then
101                 pkgfile="$pkgid.zpm"
102         fi
103
104         if [ $verbose -gt 0 ]; then
105                 echo merging $pkgfile $pkgid into $target
106         fi
107         pkglist=$pkgid
108 else
109         pkglist=$(zpm list -f "$pkgfile")
110 fi
111
112 zpm test -v "$pkgfile" || exit 1
113
114 if [ -n "$newstatus" ]; then
115         newstatus=$(zpm quote "$newstatus")
116 fi
117
118 mergeone() {
119         pkgid=$1
120         package=$(zpm parse -n "$pkgid")
121         where="where printf('%s-%s-%s', P.package, P.version, P.release) = '$pkgid'"
122         if [ $update -eq 1 ]; then
123                 printf "delete from packages_pkgid where pkgid = '%s';\n" "$pkgid"
124         fi
125         if [ $remove_older -eq 1 ]; then
126                 printf "delete from packages_pkgid where package = '$package' and pkgid < '%s' collate vercmp;\n" "$pkgid"
127         fi
128         printf "insert or rollback into packages select * from remote.packages P %s;\n" "$where"
129         printf "insert or rollback into packagefiles select * from remote.packagefiles P %s;\n" "$where"
130         printf "insert or rollback into scripts select * from remote.scripts P %s;\n" "$where"
131         if [ $mergefiles -eq 1 ]; then
132                 printf "insert into files select F.* from remote.files F\n"
133                 printf "inner join remote.packagefiles_pkgid P on P.hash = F.hash %s\n" "$where"
134                 printf "on conflict (hash) do nothing;\n";
135         fi
136         # scripts
137         if [ $mergescripts -eq 1 ]; then
138                 printf "insert into files select F.* from remote.files F\n"
139                 printf "inner join remote.scripts_pkgid P on P.hash = F.hash where P.pkgid = '%s'\n" "$pkgid"
140                 printf "on conflict (hash) do nothing;\n";
141         fi
142         if [ -n "$newstatus" ]; then
143                 newstatus=$(zpm quote "$newstatus")
144                 printf "update packages as P set status = '$newstatus' %s;\n" "$where"
145         fi
146 }
147
148 # check for already merged packages
149 if [ $update -eq 0 ]; then
150         efail=0
151
152         for pkg in $pkglist; do
153                 merged=$(zpm shell "$target" "select 1 from packages_pkgid where pkgid = '$pkgid'")
154                 if [ -n "$merged" ]; then
155                         warn "$pkgid already exists in $target";
156                         efail=1
157                 fi
158         done
159         if [ $efail -eq 1 ]; then
160                 die "aborting merge"
161         fi
162 fi
163
164 check_newer() {
165         pkg=$1
166         package=$(zpm parse -n "$pkg")
167         newer=$(zpm shell "$target" "select '$pkg' > (select max(pkgid) collate vercmp from packages_pkgid where package = '$package') from packages_pkgid where package = '$package'")
168         if [ "$newer" = '1' ] || [ -z "$newer" ]; then
169                 return 0;
170         fi
171         return 1
172 }
173
174 mlist="$pkglist"
175 if [ $only_if_newer -eq 1 ]; then
176         mlist=
177         for pkgid in $pkglist; do
178                 if check_newer "$pkgid" ; then
179                         mlist="$mlist $pkgid"
180                 fi
181         done
182         if [ -z "$mlist" ]; then
183                 exit 0
184         fi
185 fi
186
187 # TODO file tags and package tags
188 {
189         printf ".bail on\n"
190         printf "attach '%s' as remote;\n" "$pkgfile"
191         printf "begin;\n"
192
193         for pkgid in $pkglist; do
194                 mergeone "$pkgid"
195         done
196
197         printf "insert or ignore into elfneeded select * from remote.elfneeded;\n"
198         printf "insert or ignore into elflibraries select * from remote.elflibraries;\n"
199
200         printf "commit;\n"
201 } | zpm shell $target
202
203 if [ $verbose -ne 0 ]; then
204         printf "%s\n" $pkglist
205 fi