/* SPDX-License-Identifier: ISC */
#ifndef PKG2SQFS_H
#define PKG2SQFS_H

#include <sys/types.h>
#include <stdbool.h>
#include <unistd.h>
#include <stdlib.h>
#include <getopt.h>
#include <stdint.h>
#include <assert.h>
#include <stdio.h>
#include <fcntl.h>
#include <errno.h>

#ifdef HAVE_LIBBSD
#include <bsd/bsd.h>
#endif

#include "util/util.h"

#include "filelist/image_entry.h"
#include "pkg/pkgreader.h"
#include "pkg/pkgio.h"
#include "squashfs.h"

#include "config.h"

typedef struct {
	uint8_t data[SQFS_META_BLOCK_SIZE + 2];
	size_t offset;
	size_t written;
	int outfd;
} meta_writer_t;

typedef struct file_info_t {
	struct file_info_t *frag_next;
	uint64_t size;
	uint64_t startblock;
	uint32_t id;
	uint32_t fragment;
	uint32_t fragment_offset;
	uint32_t blocksizes[];
} file_info_t;

typedef struct {
	struct node_t *children;
	uint64_t dref;
	size_t size;
} dir_info_t;

typedef struct node_t {
	struct node_t *parent;
	struct node_t *next;
	const char *name;

	union {
		dir_info_t *dir;
		const char *symlink;
		dev_t device;
		file_info_t *file;
	} data;

	uint64_t inode_ref;
	uint32_t inode_num;
	uint16_t type;
	uint16_t mode;
	uint16_t uid_idx;
	uint16_t gid_idx;

	uint8_t extra[];
} node_t;

typedef struct {
	uint32_t default_uid;
	uint32_t default_gid;
	uint32_t default_mode;

	node_t **node_tbl;
	size_t num_inodes;

	uint32_t *id_tbl;
	size_t num_ids;
	size_t max_ids;

	node_t *root;
} vfs_t;

typedef struct {
	int outfd;
	sqfs_super_t super;
	vfs_t fs;
	pkg_reader_t *rd;
	void *block;
	void *scratch;
	void *fragment;

	sqfs_fragment_t *fragments;
	size_t num_fragments;
	size_t max_fragments;

	int file_block_count;
	file_info_t *frag_list;
	size_t frag_offset;
} sqfs_info_t;

int pkg_data_to_sqfs(sqfs_info_t *info);

int sqfs_super_init(sqfs_super_t *s, int64_t timestamp,
		    uint32_t blocksize, E_SQFS_COMPRESSOR comp);

int sqfs_super_write(sqfs_info_t *info);

int create_vfs_tree(sqfs_info_t *info);

void destroy_vfs_tree(vfs_t *fs);

node_t *vfs_node_from_file_id(vfs_t *fs, uint32_t id);

int sqfs_write_inodes(sqfs_info_t *info);

int sqfs_write_ids(sqfs_info_t *info);

int sqfs_write_fragment_table(sqfs_info_t *info);

meta_writer_t *meta_writer_create(int fd);

void meta_writer_destroy(meta_writer_t *m);

int meta_writer_flush(meta_writer_t *m);

int meta_writer_append(meta_writer_t *m, const void *data, size_t size);

int sqfs_write_table(sqfs_info_t *info, const void *data, size_t entsize,
		     size_t count, uint64_t *start);

#endif /* PKG2SQFS_H */