main.c
author Kristian H?gsberg <krh@redhat.com>
Mon Apr 07 12:38:21 2008 -0400 (2008-04-07)
changeset 205 bb1a6b6578a9
parent 204 60badcb3d8ea
child 206 75d4b6a55593
permissions -rw-r--r--
Silly spelling error.
krh@43
     1
#include <stdlib.h>
krh@43
     2
#include <stddef.h>
krh@43
     3
#include <stdio.h>
krh@186
     4
#include <stdint.h>
krh@43
     5
#include <string.h>
krh@72
     6
#include <sys/stat.h>
krh@43
     7
#include <unistd.h>
krh@197
     8
#include <fcntl.h>
krh@75
     9
#include <dirent.h>
krh@71
    10
#include <curl/curl.h>
krh@94
    11
#include <fnmatch.h>
krh@43
    12
#include "razor.h"
krh@151
    13
#include "razor-internal.h"
krh@43
    14
krh@152
    15
static const char system_repo_filename[] = "system.repo";
krh@171
    16
static const char next_repo_filename[] = "system-next.repo";
krh@152
    17
static const char rawhide_repo_filename[] = "rawhide.repo";
krh@152
    18
static const char updated_repo_filename[] = "system-updated.repo";
krh@152
    19
static const char razor_root_path[] = "/var/lib/razor";
krh@152
    20
static const char root[] = "install";
krh@152
    21
static const char *repo_filename = system_repo_filename;
krh@43
    22
krh@43
    23
static int
krh@43
    24
command_list(int argc, const char *argv[])
krh@43
    25
{
krh@43
    26
	struct razor_set *set;
krh@92
    27
	struct razor_package_iterator *pi;
krh@92
    28
	struct razor_package *package;
krh@192
    29
	const char *pattern, *name, *version, *arch;
krh@187
    30
	int only_names = 0, i = 0;
krh@43
    31
krh@193
    32
	if (i < argc && strcmp(argv[i], "--only-names") == 0) {
krh@187
    33
		only_names = 1;
krh@187
    34
		i++;
krh@187
    35
	}
krh@187
    36
krh@187
    37
	pattern = argv[i];
krh@43
    38
	set = razor_set_open(repo_filename);
krh@92
    39
	pi = razor_package_iterator_create(set);
krh@192
    40
	while (razor_package_iterator_next(pi, &package,
krh@192
    41
					   &name, &version, &arch)) {
krh@92
    42
		if (pattern && fnmatch(pattern, name, 0) != 0)
krh@92
    43
			continue;
krh@92
    44
krh@187
    45
		if (only_names)
krh@187
    46
			printf("%s\n", name);
krh@187
    47
		else
krh@192
    48
			printf("%s-%s.%s\n", name, version, arch);
krh@92
    49
	}
krh@92
    50
	razor_package_iterator_destroy(pi);
krh@92
    51
	razor_set_destroy(set);
krh@92
    52
krh@92
    53
	return 0;
krh@92
    54
}
krh@92
    55
krh@92
    56
static int
krh@92
    57
list_properties(const char *package_name,
krh@92
    58
		enum razor_property_type required_type)
krh@92
    59
{
krh@92
    60
	struct razor_set *set;
krh@92
    61
	struct razor_property *property;
krh@92
    62
	struct razor_package *package;
krh@92
    63
	struct razor_property_iterator *pi;
krh@92
    64
	const char *name, *version;
krh@92
    65
	enum razor_property_type type;
danw@109
    66
	enum razor_version_relation relation;
krh@92
    67
krh@92
    68
	set = razor_set_open(repo_filename);
krh@92
    69
	if (package_name)
krh@92
    70
		package = razor_set_get_package(set, package_name);
krh@92
    71
	else
krh@92
    72
		package = NULL;
krh@92
    73
krh@92
    74
	pi = razor_property_iterator_create(set, package);
krh@92
    75
	while (razor_property_iterator_next(pi, &property,
danw@109
    76
					    &name, &relation, &version,
danw@109
    77
					    &type)) {
krh@92
    78
		if (type != required_type)
krh@92
    79
			continue;
krh@92
    80
		if (version[0] == '\0')
krh@92
    81
			printf("%s\n", name);
krh@92
    82
		else
danw@137
    83
			printf("%s %s %s\n", name,
danw@137
    84
			       razor_version_relations[relation], version);
krh@92
    85
	}
krh@92
    86
	razor_property_iterator_destroy(pi);
krh@92
    87
krh@43
    88
	razor_set_destroy(set);
krh@43
    89
krh@43
    90
	return 0;
krh@43
    91
}
krh@43
    92
krh@43
    93
static int
krh@43
    94
command_list_requires(int argc, const char *argv[])
krh@43
    95
{
krh@92
    96
	return list_properties(argv[0], RAZOR_PROPERTY_REQUIRES);
krh@43
    97
}
krh@43
    98
krh@43
    99
static int
krh@43
   100
command_list_provides(int argc, const char *argv[])
krh@43
   101
{
krh@92
   102
	return list_properties(argv[0], RAZOR_PROPERTY_PROVIDES);
krh@43
   103
}
krh@43
   104
krh@43
   105
static int
krh@67
   106
command_list_obsoletes(int argc, const char *argv[])
krh@67
   107
{
krh@92
   108
	return list_properties(argv[0], RAZOR_PROPERTY_OBSOLETES);
krh@67
   109
}
krh@67
   110
krh@67
   111
static int
krh@67
   112
command_list_conflicts(int argc, const char *argv[])
krh@67
   113
{
krh@92
   114
	return list_properties(argv[0], RAZOR_PROPERTY_CONFLICTS);
krh@67
   115
}
krh@67
   116
krh@67
   117
static int
krh@48
   118
command_list_files(int argc, const char *argv[])
krh@48
   119
{
krh@48
   120
	struct razor_set *set;
krh@48
   121
krh@48
   122
	set = razor_set_open(repo_filename);
krh@48
   123
	if (set == NULL)
krh@48
   124
		return 1;
krh@49
   125
	razor_set_list_files(set, argv[0]);
krh@48
   126
	razor_set_destroy(set);
krh@48
   127
krh@48
   128
	return 0;
krh@48
   129
}
krh@48
   130
krh@48
   131
static int
krh@52
   132
command_list_file_packages(int argc, const char *argv[])
krh@52
   133
{
krh@52
   134
	struct razor_set *set;
krh@102
   135
	struct razor_package_iterator *pi;
krh@102
   136
	struct razor_package *package;
krh@192
   137
	const char *name, *version, *arch;
krh@52
   138
krh@52
   139
	set = razor_set_open(repo_filename);
krh@52
   140
	if (set == NULL)
krh@52
   141
		return 1;
krh@102
   142
krh@102
   143
	pi = razor_package_iterator_create_for_file(set, argv[0]);
krh@192
   144
	while (razor_package_iterator_next(pi, &package,
krh@192
   145
					   &name, &version, &arch))
krh@102
   146
		printf("%s-%s\n", name, version);
krh@102
   147
	razor_package_iterator_destroy(pi);
krh@102
   148
krh@52
   149
	razor_set_destroy(set);
krh@52
   150
krh@52
   151
	return 0;
krh@52
   152
}
krh@52
   153
krh@56
   154
static int
krh@56
   155
command_list_package_files(int argc, const char *argv[])
krh@56
   156
{
krh@56
   157
	struct razor_set *set;
krh@56
   158
krh@56
   159
	set = razor_set_open(repo_filename);
krh@56
   160
	if (set == NULL)
krh@56
   161
		return 1;
krh@56
   162
	razor_set_list_package_files(set, argv[0]);
krh@56
   163
	razor_set_destroy(set);
krh@56
   164
krh@56
   165
	return 0;
krh@56
   166
}
krh@52
   167
krh@101
   168
static void
krh@101
   169
list_packages_for_property(struct razor_set *set,
krh@101
   170
			   struct razor_property *property)
krh@101
   171
{
krh@101
   172
	struct razor_package_iterator *pi;
krh@101
   173
	struct razor_package *package;
krh@192
   174
	const char *name, *version, *arch;
krh@101
   175
krh@101
   176
	pi = razor_package_iterator_create_for_property(set, property);
krh@192
   177
	while (razor_package_iterator_next(pi, &package,
krh@192
   178
					   &name, &version, &arch))
krh@192
   179
		printf("%s-%s.%s\n", name, version, arch);
krh@101
   180
	razor_package_iterator_destroy(pi);
krh@101
   181
}
krh@101
   182
krh@52
   183
static int
krh@100
   184
list_property_packages(const char *ref_name,
krh@100
   185
		       const char *ref_version,
krh@100
   186
		       enum razor_property_type ref_type)
krh@43
   187
{
krh@43
   188
	struct razor_set *set;
krh@100
   189
	struct razor_property *property;
krh@100
   190
	struct razor_property_iterator *pi;
krh@100
   191
	const char *name, *version;
krh@100
   192
	enum razor_property_type type;
danw@109
   193
	enum razor_version_relation relation;
krh@100
   194
krh@100
   195
	if (ref_name == NULL)
krh@100
   196
		return 0;
krh@43
   197
krh@43
   198
	set = razor_set_open(repo_filename);
krh@100
   199
	if (set == NULL)
krh@100
   200
		return 1;
krh@100
   201
krh@100
   202
	pi = razor_property_iterator_create(set, NULL);
krh@100
   203
	while (razor_property_iterator_next(pi, &property,
danw@109
   204
					    &name, &relation, &version,
danw@109
   205
					    &type)) {
krh@100
   206
		if (strcmp(ref_name, name) != 0)
krh@100
   207
			continue;
danw@109
   208
		if (ref_version && relation == RAZOR_VERSION_EQUAL &&
danw@109
   209
		    strcmp(ref_version, version) != 0)
krh@100
   210
			continue;
krh@100
   211
		if (ref_type != type)
krh@100
   212
			continue;
krh@100
   213
krh@101
   214
		list_packages_for_property(set, property);
krh@100
   215
	}
krh@100
   216
	razor_property_iterator_destroy(pi);
krh@43
   217
krh@43
   218
	return 0;
krh@43
   219
}
krh@43
   220
krh@43
   221
static int
krh@100
   222
command_what_requires(int argc, const char *argv[])
krh@100
   223
{
krh@100
   224
	return list_property_packages(argv[0], argv[1],
krh@100
   225
				      RAZOR_PROPERTY_REQUIRES);
krh@100
   226
}
krh@100
   227
krh@100
   228
static int
krh@43
   229
command_what_provides(int argc, const char *argv[])
krh@43
   230
{
krh@100
   231
	return list_property_packages(argv[0], argv[1],
krh@100
   232
				      RAZOR_PROPERTY_PROVIDES);
krh@43
   233
}
krh@43
   234
krh@73
   235
static int
krh@73
   236
show_progress(void *clientp,
krh@73
   237
	      double dltotal, double dlnow, double ultotal, double ulnow)
krh@73
   238
{
krh@73
   239
	const char *file = clientp;
krh@73
   240
krh@73
   241
	if (!dlnow < dltotal)
krh@73
   242
		fprintf(stderr, "\rdownloading %s, %dkB/%dkB",
krh@73
   243
			file, (int) dlnow / 1024, (int) dltotal / 1024);
krh@73
   244
krh@73
   245
	return 0;
krh@73
   246
}
krh@73
   247
krh@73
   248
static int
krh@194
   249
download_if_missing(const char *url, const char *file)
krh@73
   250
{
krh@194
   251
	CURL *curl;
krh@73
   252
	struct stat buf;
krh@73
   253
	char error[256];
krh@73
   254
	FILE *fp;
krh@73
   255
	CURLcode res;
krh@204
   256
	long response;
krh@73
   257
krh@194
   258
	curl = curl_easy_init();
krh@194
   259
	if (curl == NULL)
krh@194
   260
		return 1;
krh@194
   261
krh@73
   262
	curl_easy_setopt(curl, CURLOPT_ERRORBUFFER, error);
krh@73
   263
	curl_easy_setopt(curl, CURLOPT_NOPROGRESS, 0);
krh@73
   264
	curl_easy_setopt(curl, CURLOPT_PROGRESSFUNCTION, show_progress);
krh@73
   265
	curl_easy_setopt(curl, CURLOPT_PROGRESSDATA, file);
krh@73
   266
krh@73
   267
	if (stat(file, &buf) < 0) {
krh@73
   268
		fp = fopen(file, "w");
krh@73
   269
		curl_easy_setopt(curl, CURLOPT_WRITEDATA, fp);
krh@187
   270
		curl_easy_setopt(curl, CURLOPT_URL, url);
krh@73
   271
		res = curl_easy_perform(curl);
krh@73
   272
		fclose(fp);
krh@73
   273
		if (res != CURLE_OK) {
krh@73
   274
			fprintf(stderr, "curl error: %s\n", error);
krh@73
   275
			unlink(file);
krh@73
   276
			return -1;
krh@73
   277
		}
krh@204
   278
		res = curl_easy_getinfo(curl,
krh@204
   279
					CURLINFO_RESPONSE_CODE, &response);
krh@204
   280
		if (res != CURLE_OK) {
krh@204
   281
			fprintf(stderr, "curl error: %s\n", error);
krh@204
   282
                        unlink(file);
krh@204
   283
                        return -1;
krh@204
   284
		}
krh@204
   285
		if (response != 200) {
krh@204
   286
			fprintf(stderr, " - failed %ld\n", response);
krh@204
   287
                        unlink(file);
krh@204
   288
                        return -1;
krh@204
   289
		}
krh@188
   290
		fprintf(stderr, "\n");
krh@73
   291
	}
krh@73
   292
krh@194
   293
	curl_easy_cleanup(curl);
krh@194
   294
krh@73
   295
	return 0;
krh@73
   296
}
krh@73
   297
krh@71
   298
#define REPO_URL "http://download.fedora.redhat.com" \
krh@187
   299
	"/pub/fedora/linux/development/i386/os"
krh@71
   300
krh@43
   301
static int
krh@43
   302
command_import_yum(int argc, const char *argv[])
krh@43
   303
{
krh@43
   304
	struct razor_set *set;
krh@71
   305
krh@194
   306
	if (download_if_missing(REPO_URL "/repodata/primary.xml.gz",
krh@187
   307
				"primary.xml.gz") < 0)
krh@73
   308
		return -1;
krh@194
   309
	if (download_if_missing(REPO_URL "/repodata/filelists.xml.gz",
krh@187
   310
				"filelists.xml.gz") < 0)
krh@73
   311
		return -1;
krh@43
   312
krh@70
   313
	set = razor_set_create_from_yum();
krh@43
   314
	if (set == NULL)
krh@43
   315
		return 1;
krh@43
   316
	razor_set_write(set, rawhide_repo_filename);
krh@43
   317
	razor_set_destroy(set);
krh@43
   318
	printf("wrote %s\n", rawhide_repo_filename);
krh@43
   319
krh@43
   320
	return 0;
krh@43
   321
}
krh@43
   322
krh@43
   323
static int
krh@43
   324
command_import_rpmdb(int argc, const char *argv[])
krh@43
   325
{
krh@43
   326
	struct razor_set *set;
krh@43
   327
krh@43
   328
	set = razor_set_create_from_rpmdb();
krh@43
   329
	if (set == NULL)
krh@43
   330
		return 1;
krh@43
   331
	razor_set_write(set, repo_filename);
krh@43
   332
	razor_set_destroy(set);
krh@43
   333
	printf("wrote %s\n", repo_filename);
krh@43
   334
krh@43
   335
	return 0;
krh@43
   336
}
krh@43
   337
krh@43
   338
static int
krh@43
   339
command_validate(int argc, const char *argv[])
krh@43
   340
{
krh@43
   341
	struct razor_set *set;
krh@43
   342
krh@43
   343
	set = razor_set_open(repo_filename);
krh@43
   344
	if (set == NULL)
krh@43
   345
		return 1;
krh@43
   346
	razor_set_list_unsatisfied(set);
krh@43
   347
	razor_set_destroy(set);
krh@43
   348
krh@43
   349
	return 0;
krh@43
   350
}
krh@43
   351
krh@43
   352
static int
krh@43
   353
command_update(int argc, const char *argv[])
krh@43
   354
{
krh@43
   355
	struct razor_set *set, *upstream;
danw@137
   356
	struct razor_transaction *trans;
krh@190
   357
	int errors;
krh@43
   358
krh@43
   359
	set = razor_set_open(repo_filename);
krh@43
   360
	upstream = razor_set_open(rawhide_repo_filename);
krh@43
   361
	if (set == NULL || upstream == NULL)
krh@43
   362
		return 1;
danw@137
   363
	trans = razor_transaction_create(set, upstream, argc, argv, 0, NULL);
krh@190
   364
	errors = razor_transaction_describe(trans);
krh@190
   365
	if (errors)
danw@137
   366
		return 1;
danw@137
   367
krh@196
   368
	set = razor_transaction_finish(trans);
krh@44
   369
	razor_set_write(set, updated_repo_filename);
krh@43
   370
	razor_set_destroy(set);
krh@43
   371
	razor_set_destroy(upstream);
krh@43
   372
	printf("wrote system-updated.repo\n");
krh@43
   373
krh@43
   374
	return 0;
krh@43
   375
}
krh@43
   376
danw@129
   377
static int
danw@129
   378
command_remove(int argc, const char *argv[])
danw@129
   379
{
danw@129
   380
	struct razor_set *set;
danw@137
   381
	struct razor_transaction *trans;
krh@190
   382
	int errors;
danw@129
   383
danw@129
   384
	set = razor_set_open(repo_filename);
danw@129
   385
	if (set == NULL)
danw@129
   386
		return 1;
danw@137
   387
	trans = razor_transaction_create(set, NULL, 0, NULL, argc, argv);
krh@190
   388
	errors = razor_transaction_describe(trans);
krh@190
   389
	if (errors)
danw@137
   390
		return 1;
danw@137
   391
krh@196
   392
	set = razor_transaction_finish(trans);
danw@129
   393
	razor_set_write(set, updated_repo_filename);
danw@129
   394
	razor_set_destroy(set);
danw@129
   395
	printf("wrote system-updated.repo\n");
danw@129
   396
danw@129
   397
	return 0;
danw@129
   398
}
danw@129
   399
krh@44
   400
static void
krh@44
   401
print_diff(const char *name,
krh@192
   402
	   const char *old_version, const char *new_version, const char *arch,
krh@192
   403
	   void *data)
krh@44
   404
{
krh@44
   405
	if (old_version)
krh@44
   406
		printf("removing %s %s\n", name, old_version);
krh@44
   407
	else
krh@44
   408
		printf("install %s %s\n", name, new_version);
krh@44
   409
}
krh@44
   410
krh@44
   411
static int
krh@44
   412
command_diff(int argc, const char *argv[])
krh@44
   413
{
krh@44
   414
	struct razor_set *set, *updated;
krh@44
   415
krh@44
   416
	set = razor_set_open(repo_filename);
krh@44
   417
	updated = razor_set_open(updated_repo_filename);
krh@44
   418
	if (set == NULL || updated == NULL)
krh@44
   419
		return 1;
krh@44
   420
krh@44
   421
	razor_set_diff(set, updated, print_diff, NULL);	
krh@44
   422
krh@44
   423
	razor_set_destroy(set);
krh@44
   424
	razor_set_destroy(updated);
krh@44
   425
krh@44
   426
	return 0;
krh@44
   427
}
krh@44
   428
krh@74
   429
static int
krh@75
   430
command_import_rpms(int argc, const char *argv[])
krh@74
   431
{
krh@75
   432
	DIR *dir;
krh@75
   433
	struct dirent *de;
krh@75
   434
	struct razor_importer *importer;
krh@75
   435
	struct razor_set *set;
krh@77
   436
	struct razor_rpm *rpm;
krh@75
   437
	int len;
krh@75
   438
	char filename[256];
krh@75
   439
	const char *dirname = argv[0];
krh@75
   440
krh@75
   441
	if (dirname == NULL) {
krh@75
   442
		fprintf(stderr, "usage: razor import-rpms DIR\n");
krh@75
   443
		return -1;
krh@75
   444
	}
krh@75
   445
krh@75
   446
	dir = opendir(dirname);
krh@75
   447
	if (dir == NULL) {
krh@75
   448
		fprintf(stderr, "couldn't read dir %s\n", dirname);
krh@75
   449
		return -1;
krh@75
   450
	}
krh@75
   451
krh@75
   452
	importer = razor_importer_new();
krh@75
   453
krh@75
   454
	while (de = readdir(dir), de != NULL) {
krh@75
   455
		len = strlen(de->d_name);
krh@75
   456
		if (len < 5 || strcmp(de->d_name + len - 4, ".rpm") != 0)
krh@75
   457
		    continue;
krh@75
   458
		snprintf(filename, sizeof filename,
krh@75
   459
			 "%s/%s", dirname, de->d_name);
krh@77
   460
		rpm = razor_rpm_open(filename);
krh@77
   461
		if (rpm == NULL) {
krh@77
   462
			fprintf(stderr,
krh@77
   463
				"failed to open rpm \"%s\"\n", filename);
krh@77
   464
			continue;
krh@77
   465
		}
krh@77
   466
		if (razor_importer_add_rpm(importer, rpm)) {
krh@75
   467
			fprintf(stderr, "couldn't import %s\n", filename);
krh@75
   468
			break;
krh@75
   469
		}
krh@77
   470
		razor_rpm_close(rpm);
krh@75
   471
	}
krh@75
   472
krh@75
   473
	if (de != NULL) {
krh@75
   474
		razor_importer_destroy(importer);
krh@75
   475
		return -1;
krh@75
   476
	}
krh@75
   477
krh@75
   478
	set = razor_importer_finish(importer);
krh@75
   479
krh@75
   480
	razor_set_write(set, repo_filename);
krh@75
   481
	razor_set_destroy(set);
krh@75
   482
	printf("wrote %s\n", repo_filename);
krh@74
   483
krh@74
   484
	return 0;
krh@74
   485
}
krh@74
   486
krh@194
   487
static void
krh@194
   488
download_package(const char *name,
krh@194
   489
		 const char *old_version,
krh@194
   490
		 const char *new_version,
krh@194
   491
		 const char *arch,
krh@194
   492
		 void *data)
krh@152
   493
{
krh@194
   494
	char file[PATH_MAX], url[256];
krh@201
   495
	const char *v;
krh@204
   496
	int *errors = data;
krh@194
   497
krh@194
   498
	if (old_version)
krh@194
   499
		return;
krh@194
   500
krh@201
   501
	/* Skip epoch */
krh@201
   502
	v = strchr(new_version, ':');
krh@201
   503
	if (v != NULL)
krh@201
   504
		v = v + 1;
krh@201
   505
	else
krh@201
   506
		v = new_version;
krh@201
   507
krh@194
   508
	snprintf(url, sizeof url,
krh@201
   509
		 REPO_URL "/Packages/%s-%s.%s.rpm", name, v, arch);
krh@194
   510
	snprintf(file, sizeof file,
krh@201
   511
		 "rpms/%s-%s.%s.rpm", name, v, arch);
krh@194
   512
	if (download_if_missing(url, file) < 0)
krh@204
   513
		(*errors)++;
krh@194
   514
}
krh@194
   515
krh@194
   516
static void
krh@194
   517
install_package(const char *name,
krh@194
   518
		const char *old_version,
krh@194
   519
		const char *new_version,
krh@194
   520
		const char *arch,
krh@194
   521
		void *data)
krh@194
   522
{
krh@201
   523
	const char *v, *root = data;
krh@194
   524
	char file[PATH_MAX];
krh@152
   525
	struct razor_rpm *rpm;
krh@152
   526
krh@194
   527
	if (old_version) {
krh@194
   528
		printf("removing %s %s not handled\n", name, old_version);
krh@194
   529
		return;
krh@152
   530
	}
krh@152
   531
krh@201
   532
	/* Skip epoch */
krh@201
   533
	v = strchr(new_version, ':');
krh@201
   534
	if (v != NULL)
krh@201
   535
		v = v + 1;
krh@201
   536
	else
krh@201
   537
		v = new_version;
krh@201
   538
krh@201
   539
	printf("install %s %s\n", name, v);
krh@201
   540
	snprintf(file, sizeof file, "rpms/%s-%s.%s.rpm", name, v, arch);
krh@152
   541
krh@194
   542
 	rpm = razor_rpm_open(file);
krh@194
   543
	if (rpm == NULL) {
krh@194
   544
		fprintf(stderr, "failed to open rpm %s\n", file);
krh@194
   545
		return;
krh@194
   546
	}
krh@194
   547
	if (razor_rpm_install(rpm, root) < 0) {
krh@194
   548
		fprintf(stderr,
krh@194
   549
			"failed to install rpm %s\n", file);
krh@194
   550
		return;
krh@194
   551
	}
krh@194
   552
	razor_rpm_close(rpm);
krh@152
   553
}
krh@151
   554
krh@77
   555
static int
krh@77
   556
command_install(int argc, const char *argv[])
krh@77
   557
{
krh@171
   558
	struct razor_set *system, *upstream, *next;
krh@152
   559
	struct razor_transaction *trans;
krh@194
   560
	char path[PATH_MAX], new_path[PATH_MAX];
krh@197
   561
	int errors, fd;
krh@197
   562
krh@197
   563
	/* Create the new next repo file up front to ensure exclusive
krh@197
   564
	 * access. */
krh@197
   565
	snprintf(new_path, sizeof new_path,
krh@197
   566
		 "%s%s/%s", root, razor_root_path, next_repo_filename);
krh@197
   567
	fd = open(new_path, O_CREAT | O_WRONLY | O_TRUNC | O_EXCL, 0666);
krh@197
   568
	if (fd < 0) {
krh@197
   569
		fprintf(stderr, "failed to get lock file, "
krh@197
   570
			"maybe previous operation crashed?\n");
krh@197
   571
krh@197
   572
		/* FIXME: Use fcntl advisory locking to figure out
krh@197
   573
		 * whether previous operation crashed or is still in
krh@197
   574
		 * progress. */
krh@197
   575
krh@197
   576
		return -1;
krh@197
   577
	}
krh@151
   578
krh@194
   579
	upstream = razor_set_open(rawhide_repo_filename);
krh@152
   580
	snprintf(path, sizeof path,
krh@152
   581
		 "%s%s/%s", root, razor_root_path, system_repo_filename);
krh@152
   582
	system = razor_set_open(path);
krh@197
   583
	if (system == NULL || upstream == NULL) {
krh@197
   584
		unlink(new_path);
krh@194
   585
		return 1;
krh@197
   586
	}
krh@152
   587
	trans = razor_transaction_create(system, upstream,
krh@194
   588
					 argc, argv, 0, NULL);
krh@190
   589
	errors = razor_transaction_describe(trans);
krh@197
   590
	if (errors) {
krh@197
   591
		unlink(new_path);
krh@152
   592
		return 1;
krh@197
   593
	}
krh@152
   594
krh@196
   595
	next = razor_transaction_finish(trans);
krh@152
   596
krh@197
   597
	razor_set_write_to_fd(next, fd);
krh@194
   598
	printf("wrote %s\n", new_path);
krh@194
   599
krh@204
   600
	razor_set_diff(system, next, download_package, &errors);
krh@204
   601
	if (errors > 0) {
krh@204
   602
		fprintf(stderr, "failed to download %d packages\n", errors);
krh@204
   603
		unlink(new_path);
krh@204
   604
                return 1;
krh@204
   605
        }
krh@194
   606
krh@200
   607
	/* FIXME: We need to figure out the right install order here,
krh@200
   608
	 * so the post and pre scripts can run. */
krh@194
   609
	razor_set_diff(system, next, install_package, (void *) root);
krh@171
   610
krh@171
   611
	razor_set_destroy(next);
krh@171
   612
	razor_set_destroy(system);
krh@171
   613
	razor_set_destroy(upstream);
krh@171
   614
krh@171
   615
	/* Make it so. */
krh@171
   616
	rename(new_path, path);
krh@171
   617
	printf("renamed %s to %s\n", new_path, path);
krh@171
   618
krh@151
   619
	return 0;
krh@151
   620
}
krh@151
   621
krh@151
   622
static int
krh@151
   623
command_init(int argc, const char *argv[])
krh@151
   624
{
krh@77
   625
	struct stat buf;
krh@151
   626
	struct razor_set *set;
krh@151
   627
	char path[PATH_MAX];
krh@77
   628
krh@77
   629
	if (stat(root, &buf) < 0) {
krh@77
   630
		if (mkdir(root, 0777) < 0) {
krh@77
   631
			fprintf(stderr,
krh@77
   632
				"could not create install root \"%s\"\n",
krh@77
   633
				root);
krh@77
   634
			return -1;
krh@77
   635
		}
krh@77
   636
		fprintf(stderr, "created install root \"%s\"\n", root);
krh@77
   637
	} else if (!S_ISDIR(buf.st_mode)) {
krh@77
   638
		fprintf(stderr,
krh@77
   639
			"install root \"%s\" exists, but is not a directory\n",
krh@77
   640
			root);
krh@77
   641
		return -1;
krh@77
   642
	}
krh@77
   643
krh@151
   644
	if (razor_create_dir(root, razor_root_path) < 0) {
krh@151
   645
		fprintf(stderr, "could not create %s%s\n",
krh@151
   646
			root, razor_root_path);
krh@77
   647
		return -1;
krh@77
   648
	}
krh@151
   649
krh@151
   650
	set = razor_set_create();
krh@151
   651
	snprintf(path, sizeof path, "%s%s/%s",
krh@152
   652
		 root, razor_root_path, system_repo_filename);
krh@151
   653
	if (razor_set_write(set, path) < 0) {
krh@151
   654
		fprintf(stderr, "could not write initial package set\n");
krh@77
   655
		return -1;
krh@77
   656
	}
krh@151
   657
	razor_set_destroy(set);
krh@77
   658
krh@77
   659
	return 0;
krh@77
   660
}
krh@77
   661
krh@187
   662
static int
krh@187
   663
command_download(int argc, const char *argv[])
krh@187
   664
{
krh@187
   665
	struct razor_set *set;
krh@187
   666
	struct razor_package_iterator *pi;
krh@187
   667
	struct razor_package *package;
krh@192
   668
	const char *pattern = argv[0], *name, *version, *arch;
krh@187
   669
	char url[256], file[256];
krh@203
   670
	int matches = 0;
krh@187
   671
krh@187
   672
	set = razor_set_open(rawhide_repo_filename);
krh@187
   673
	pi = razor_package_iterator_create(set);
krh@192
   674
	while (razor_package_iterator_next(pi, &package,
krh@192
   675
					   &name, &version, &arch)) {
krh@187
   676
		if (pattern && fnmatch(pattern, name, 0) != 0)
krh@187
   677
			continue;
krh@187
   678
krh@203
   679
		matches++;
krh@187
   680
		snprintf(url, sizeof url,
krh@203
   681
			 REPO_URL "/Packages/%s-%s.%s.rpm",
krh@203
   682
			 name, version, arch);
krh@187
   683
		snprintf(file, sizeof file,
krh@203
   684
			 "rpms/%s-%s.%s.rpm", name, version, arch);
krh@204
   685
		download_if_missing(url, file);
krh@187
   686
	}
krh@187
   687
	razor_package_iterator_destroy(pi);
krh@187
   688
	razor_set_destroy(set);
krh@187
   689
krh@203
   690
	if (matches == 0)
krh@203
   691
		fprintf(stderr, "no packages matched \"%s\"\n", pattern);
krh@203
   692
	else if (matches == 1)
krh@203
   693
		fprintf(stderr, "downloaded 1 package\n");
krh@203
   694
	else
krh@205
   695
		fprintf(stderr, "downloaded %d packages\n", matches);
krh@203
   696
krh@187
   697
	return 0;
krh@187
   698
}
krh@187
   699
krh@43
   700
static struct {
krh@43
   701
	const char *name;
krh@43
   702
	const char *description;
krh@43
   703
	int (*func)(int argc, const char *argv[]);
krh@43
   704
} razor_commands[] = {
krh@43
   705
	{ "list", "list all packages", command_list },
krh@67
   706
	{ "list-requires", "list all requires for the given package", command_list_requires },
krh@75
   707
	{ "list-provides", "list all provides for the given package", command_list_provides },
krh@75
   708
	{ "list-obsoletes", "list all obsoletes for the given package", command_list_obsoletes },
krh@75
   709
	{ "list-conflicts", "list all conflicts for the given package", command_list_conflicts },
krh@48
   710
	{ "list-files", "list files for package set", command_list_files },
krh@52
   711
	{ "list-file-packages", "list packages owning file", command_list_file_packages },
krh@56
   712
	{ "list-package-files", "list files in package", command_list_package_files },
krh@43
   713
	{ "what-requires", "list the packages that have the given requires", command_what_requires },
krh@43
   714
	{ "what-provides", "list the packages that have the given provides", command_what_provides },
krh@75
   715
	{ "import-yum", "import yum metadata files", command_import_yum },
krh@43
   716
	{ "import-rpmdb", "import the system rpm database", command_import_rpmdb },
krh@75
   717
	{ "import-rpms", "import rpms from the given directory", command_import_rpms },
krh@43
   718
	{ "validate", "validate a package set", command_validate },
krh@44
   719
	{ "update", "update all or specified packages", command_update },
danw@129
   720
	{ "remove", "remove specified packages", command_remove },
krh@77
   721
	{ "diff", "show diff between two package sets", command_diff },
krh@151
   722
	{ "install", "install rpm", command_install },
krh@187
   723
	{ "init", "init razor root", command_init },
krh@187
   724
	{ "download", "download packages", command_download }
krh@43
   725
};
krh@43
   726
krh@43
   727
static int
krh@43
   728
usage(void)
krh@43
   729
{
krh@43
   730
	int i;
krh@43
   731
krh@43
   732
	printf("usage:\n");
krh@43
   733
	for (i = 0; i < ARRAY_SIZE(razor_commands); i++)
krh@43
   734
		printf("  %-20s%s\n",
krh@43
   735
		       razor_commands[i].name, razor_commands[i].description);
krh@43
   736
krh@43
   737
	return 1;
krh@43
   738
}
krh@43
   739
krh@43
   740
int
krh@43
   741
main(int argc, const char *argv[])
krh@43
   742
{
krh@43
   743
	char *repo;
krh@43
   744
	int i;
krh@43
   745
krh@43
   746
	repo = getenv("RAZOR_REPO");
krh@43
   747
	if (repo != NULL)
krh@43
   748
		repo_filename = repo;
krh@43
   749
krh@43
   750
	if (argc < 2)
krh@43
   751
		return usage();
krh@43
   752
krh@43
   753
	for (i = 0; i < ARRAY_SIZE(razor_commands); i++)
krh@43
   754
		if (strcmp(razor_commands[i].name, argv[1]) == 0)
krh@43
   755
			return razor_commands[i].func(argc - 2, argv + 2);
krh@43
   756
krh@43
   757
	return usage();
krh@43
   758
}