327 lines
7.3 KiB
C
327 lines
7.3 KiB
C
/* vim: set sw=3 ts=3 sts=3 expandtab: */
|
|
#include "ecdb.h"
|
|
|
|
int ecdb_source_init(Ecdb_Source *src);
|
|
void ecdb_source_add_directory_recursive(Ecdb_Source *parent);
|
|
|
|
Ecdb_Source *
|
|
ecdb_source_new(void)
|
|
{
|
|
Ecdb_Source *src;
|
|
|
|
src = calloc(1, sizeof(Ecdb_Source));
|
|
if (!src)
|
|
{
|
|
return NULL;
|
|
}
|
|
if (!ecdb_source_init(src))
|
|
{
|
|
FREE(src);
|
|
return NULL;
|
|
}
|
|
|
|
return src;
|
|
}
|
|
|
|
int
|
|
ecdb_source_init(Ecdb_Source *src)
|
|
{
|
|
src->dir = FALSE;
|
|
src->num_children = 0;
|
|
src->size = 0;
|
|
src->children = calloc(1, sizeof(Ecdb_Source));
|
|
if (!src->children)
|
|
{
|
|
return FALSE;
|
|
}
|
|
src->children[src->num_children] = NULL;
|
|
return TRUE;
|
|
}
|
|
|
|
void
|
|
ecdb_source_destroy(Ecdb_Source *src)
|
|
{
|
|
int i;
|
|
Ecdb_Source *child;
|
|
|
|
if (!src)
|
|
return;
|
|
|
|
/* free the non-recursive stuff */
|
|
if (src->dst) eina_stringshare_del(src->dst);
|
|
if (src->node) iso_node_unref(src->node);
|
|
|
|
for (i = 0; src->children[i]; i++)
|
|
{
|
|
child = src->children[i];
|
|
ecdb_source_destroy(child);
|
|
}
|
|
FREE(src->children);
|
|
FREE(src);
|
|
}
|
|
|
|
void
|
|
ecdb_source_data_set(Ecdb_Source *src, const char *dst)
|
|
{
|
|
if ((!src) || (!ecore_file_exists(dst)))
|
|
{
|
|
return;
|
|
}
|
|
|
|
src->dst = eina_stringshare_add(dst);
|
|
src->size = ecore_file_size(dst);
|
|
|
|
/* Add the files recursively here */
|
|
if (ecore_file_is_dir(src->dst))
|
|
{
|
|
src->dir = !!TRUE;
|
|
ecdb_source_add_directory_recursive(src);
|
|
}
|
|
}
|
|
|
|
void
|
|
ecdb_source_add_directory_recursive(Ecdb_Source *parent)
|
|
{
|
|
Ecore_List *files;
|
|
Ecdb_Source *child;
|
|
char path[PATH_MAX];
|
|
char *src;
|
|
|
|
files = ecore_file_ls(parent->dst);
|
|
|
|
while ((src = ecore_list_first_remove(files)))
|
|
{
|
|
child = ecdb_source_new();
|
|
snprintf(path, PATH_MAX, "%s/%s", parent->dst, src);
|
|
ecdb_source_data_set(child, path);
|
|
ecdb_source_child_append(parent, child);
|
|
FREE(src);
|
|
}
|
|
ecore_list_destroy(files);
|
|
}
|
|
|
|
void
|
|
ecdb_source_child_append(Ecdb_Source *src, Ecdb_Source *child)
|
|
{
|
|
long long orig, diff;
|
|
Ecdb_Source *p;
|
|
|
|
if (src == child)
|
|
{
|
|
printf("Trying to make a parent of itself!\n");
|
|
return;
|
|
}
|
|
|
|
orig = src->size;
|
|
src->num_children++;
|
|
src->children = realloc(src->children, sizeof(Ecdb_Source) *
|
|
(src->num_children + 1));
|
|
src->children[src->num_children - 1] = child;
|
|
src->children[src->num_children] = NULL;
|
|
src->size += child->size;
|
|
child->parent = src;
|
|
|
|
diff = src->size - orig;
|
|
p = src;
|
|
while (p->parent)
|
|
{
|
|
p = p->parent;
|
|
p->size += diff;
|
|
}
|
|
}
|
|
|
|
/* Basically here we can remove all occurences (who knows why we'd get
|
|
* multiple, but whatever), or just the first. For now remove the first, and
|
|
* see how that goes
|
|
*/
|
|
void
|
|
ecdb_source_child_remove(Ecdb_Source *src, Ecdb_Source *child)
|
|
{
|
|
Ecdb_Source **temp;
|
|
Ecdb_Source *s;
|
|
int i, cidx, f;
|
|
|
|
if (src == child)
|
|
{
|
|
printf("Trying to remove oneself\n");
|
|
return;
|
|
}
|
|
|
|
temp = calloc(src->num_children, sizeof(Ecdb_Source));
|
|
temp[src->num_children - 1] = NULL;
|
|
|
|
cidx = f = 0;
|
|
for (i = 0; src->children[i]; i++)
|
|
{
|
|
if ((src->children[i] == child) && (!f))
|
|
{
|
|
f++;
|
|
continue;
|
|
}
|
|
temp[cidx] = src->children[i];
|
|
cidx++;
|
|
}
|
|
|
|
FREE(src->children);
|
|
src->children = temp;
|
|
src->num_children--;
|
|
|
|
// Propogate the size changes back up to the root
|
|
s = src;
|
|
do {
|
|
s->size -= child->size;
|
|
|
|
if (s->parent)
|
|
s = s->parent;
|
|
else
|
|
break;
|
|
} while (1);
|
|
|
|
child->parent = NULL;
|
|
}
|
|
|
|
void
|
|
ecdb_source_add_children_rec(Ecdb_Source *parent, IsoImage *image)
|
|
{
|
|
IsoDir *cd = NULL;
|
|
Ecdb_Source *cs;
|
|
int i;
|
|
|
|
if ((!parent) || (!image))
|
|
return;
|
|
|
|
i = 0;
|
|
while ((cs = parent->children[i]))
|
|
{
|
|
if (cs->dir)
|
|
{
|
|
iso_tree_add_new_dir(ISO_DIR(parent->node),
|
|
ecore_file_file_get(cs->dst), &cd);
|
|
cs->node = ISO_NODE(cd);
|
|
/* If the source has children, find the node from above
|
|
* (if applicable), and recursively add to it */
|
|
if (cs->num_children > 0)
|
|
{
|
|
ecdb_source_add_children_rec(cs, image);
|
|
}
|
|
|
|
}
|
|
|
|
/* file */
|
|
else
|
|
{
|
|
iso_tree_add_node(image, ISO_DIR(parent->node),
|
|
cs->dst, NULL);
|
|
}
|
|
i++;
|
|
}
|
|
}
|
|
|
|
/* proj->files should only have children */
|
|
BurnSource *
|
|
ecdb_image_project(Ecdb_Burn_Project *proj)
|
|
{
|
|
IsoImage *image;
|
|
Ecdb_Source *c;
|
|
IsoWriteOpts *opts;
|
|
BurnSource *data_src, *fifo_src;
|
|
|
|
if ((!proj->files) || (!proj->files->num_children))
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
/* To handle already-suplied image files */
|
|
if ((proj->files->num_children == 1) && (ECDB_PROJECT(proj)->type ==
|
|
ECDB_IMAGE_PROJECT))
|
|
{
|
|
efreet_mime_init();
|
|
c = proj->files->children[0];
|
|
if ((!ecore_file_is_dir(c->dst)) &&
|
|
(!strcmp(efreet_mime_type_get(c->dst), "application/x-cd-image")))
|
|
{
|
|
data_src = burn_file_source_new(c->dst, NULL);
|
|
goto FIFO_CREATE;
|
|
}
|
|
else if (ECDB_PROJECT(proj)->type == ECDB_IMAGE_PROJECT)
|
|
{
|
|
printf("Supplied file is not an image!\n");
|
|
efreet_mime_shutdown();
|
|
return NULL;
|
|
}
|
|
efreet_mime_shutdown();
|
|
}
|
|
|
|
/* Otherwise we have a bunch of files */
|
|
if (!iso_image_new(proj->volume_id, &image))
|
|
{
|
|
printf("Failed to create image!\n");
|
|
return NULL;
|
|
}
|
|
|
|
/* Set all ids */
|
|
if (proj->publisher_id)
|
|
{
|
|
iso_image_set_publisher_id(image, proj->publisher_id);
|
|
}
|
|
if (proj->data_preparer_id)
|
|
{
|
|
iso_image_set_data_preparer_id(image, proj->data_preparer_id);
|
|
}
|
|
if (proj->system_id)
|
|
{
|
|
iso_image_set_system_id(image, proj->system_id);
|
|
}
|
|
if (proj->application_id)
|
|
{
|
|
iso_image_set_application_id(image, proj->application_id);
|
|
}
|
|
if (proj->copywrite_id)
|
|
{
|
|
iso_image_set_copyright_file_id(image, proj->copywrite_id);
|
|
}
|
|
if (proj->abstract_id)
|
|
{
|
|
iso_image_set_abstract_file_id(image, proj->abstract_id);
|
|
}
|
|
if (proj->biblio_id)
|
|
{
|
|
iso_image_set_biblio_file_id(image, proj->biblio_id);
|
|
}
|
|
|
|
if (!iso_write_opts_new(&opts, 2))
|
|
{
|
|
printf("Failed to create writing options!\n");
|
|
iso_image_unref(image);
|
|
return NULL;
|
|
}
|
|
|
|
/* And some write opts stuff */
|
|
iso_write_opts_set_iso_level(opts, proj->iso_level);
|
|
iso_write_opts_set_joliet(opts, proj->use_joliet);
|
|
iso_write_opts_set_rockridge(opts, proj->use_rockridge);
|
|
iso_write_opts_set_iso1999(opts, proj->iso1990);
|
|
iso_write_opts_set_appendable(opts, proj->multi);
|
|
|
|
iso_tree_set_follow_symlinks(image, proj->follow_symlinks);
|
|
iso_tree_set_ignore_hidden(image, proj->ignore_hidden);
|
|
iso_tree_set_ignore_special(image, proj->ignore_special);
|
|
|
|
/* actually fill image with some files now */
|
|
proj->files->node = ISO_NODE(iso_image_get_root(image));
|
|
ecdb_source_add_children_rec(proj->files, image);
|
|
|
|
/* Make the burn source here */
|
|
iso_image_create_burn_source(image, opts, &data_src);
|
|
iso_write_opts_free(opts);
|
|
iso_image_unref(image);
|
|
|
|
/* And, convert to fifo */
|
|
FIFO_CREATE:
|
|
fifo_src = burn_fifo_source_new(data_src, proj->fifo_chunksize,
|
|
proj->fifo_chunks, 0);
|
|
burn_source_free(data_src);
|
|
return fifo_src;
|
|
}
|
|
|