/*  
    AVFS: A Virtual File System Library
    Copyright (C) 1998  Miklos Szeredi (mszeredi@inf.bme.hu)

    This program can be distributed under the terms of the GNU GPL.
    See the file COPYING.

    UBZ2 module
*/

#include "version.h"

#include "bzfile.h"
#include "filecache.h"
#include "oper.h"
#include "version.h"

struct bznode {
    struct avstat sig;
    struct bzcache *cache;
    avino_t ino;
};

struct bzipfile {
    struct bzfile *zfil;
    vfile *base;
    struct bznode *node;
};


static void bznode_destroy(struct bznode *nod)
{
    av_unref_obj(nod->cache);
}

static struct bznode *bz_new_node(ventry *ve, struct avstat *stbuf)
{
    struct bznode *nod;

    AV_NEW_OBJ(nod, bznode_destroy);
    nod->sig = *stbuf;
    nod->cache = av_bzcache_new();
    nod->ino = av_new_ino(ve->mnt->avfs);
    
    return nod;
}

static int bz_same(struct bznode *nod, struct avstat *stbuf)
{
    if(nod->sig.ino == stbuf->ino &&
       nod->sig.dev == stbuf->dev &&
       nod->sig.size == stbuf->size &&
       AV_TIME_EQ(nod->sig.mtime, stbuf->mtime))
        return 1;
    else
        return 0;
}

static struct bznode *bz_do_get_node(ventry *ve, const char *key,
                                     struct avstat *stbuf)
{
    static AV_LOCK_DECL(lock);
    struct bznode *nod;

    AV_LOCK(lock);
    nod = (struct bznode *) av_filecache_get(key);
    if(nod != NULL) {
        if(!bz_same(nod, stbuf)) {
            av_unref_obj(nod);
            nod = NULL;
        }
    }
    
    if(nod == NULL) {
        nod =  bz_new_node(ve, stbuf);
        av_filecache_set(key, nod);
    }
    AV_UNLOCK(lock);

    return nod;
}

static int bz_getnode(ventry *ve, vfile *base, struct bznode **resp)
{
    int res;
    struct avstat stbuf;
    const int attrmask = AVA_INO | AVA_DEV | AVA_SIZE | AVA_MTIME;
    struct bznode *nod;
    char *key;

    res = av_fgetattr(base, &stbuf, attrmask);
    if(res < 0)
        return res;

    res = av_filecache_getkey(ve, &key);
    if(res < 0)
        return res;

    nod = bz_do_get_node(ve, key, &stbuf);

    av_free(key);

    *resp = nod;
    return 0;
}

static int bz_lookup(ventry *ve, const char *name, void **newp)
{
    char *path = (char *) ve->data;
    
    if(path == NULL) {
        if(name[0] != '\0')
            return -ENOENT;
	if(ve->mnt->opts[0] != '\0')
            return -ENOENT;
        path = av_strdup(name);
    }
    else if(name == NULL) {
        av_free(path);
        path = NULL;
    }
    else 
        return -ENOENT;
    
    *newp = path;
    return 0;
}

static int bz_access(ventry *ve, int amode)
{
    return av_access(ve->mnt->base, amode);
}

static int bz_open(ventry *ve, int flags, avmode_t mode, void **resp)
{
    int res;
    vfile *base;
    struct bznode *nod;
    struct bzipfile *fil;

    if(flags & AVO_DIRECTORY)
        return -ENOTDIR;

    if(AV_ISWRITE(flags))
        return -EROFS;

    res = av_open(ve->mnt->base, AVO_RDONLY, 0, &base);
    if(res < 0)
        return res;

    res = bz_getnode(ve, base, &nod);
    if(res < 0) {
        av_close(base);
        return res;
    }

    AV_NEW(fil);
    if((flags & AVO_ACCMODE) != AVO_NOPERM)
        fil->zfil = av_bzfile_new(base);
    else
        fil->zfil = NULL;

    fil->base = base;
    fil->node = nod;
    
    *resp = fil;
    return 0;
}

static int bz_close(vfile *vf)
{
    struct bzipfile *fil = (struct bzipfile *) vf->data;

    av_unref_obj(fil->zfil);
    av_unref_obj(fil->node);
    av_close(fil->base);
    av_free(fil);

    return 0;
}

static avssize_t bz_read(vfile *vf, char *buf, avsize_t nbyte)
{
    avssize_t res;
    struct bzipfile *fil = (struct bzipfile *) vf->data;
 
    res = av_bzfile_pread(fil->zfil, fil->node->cache, buf, nbyte, vf->ptr);
    if(res > 0)
        vf->ptr += res;

    return res;
}

static int bz_getattr(vfile *vf, struct avstat *buf, int attrmask)
{
    int res;
    struct bzipfile *fil = (struct bzipfile *) vf->data;
    struct bznode *nod = fil->node;
    avoff_t size;
    const int basemask = AVA_MODE | AVA_UID | AVA_GID | AVA_MTIME | AVA_ATIME | AVA_CTIME;

    res = av_fgetattr(fil->base, buf, basemask);
    if(res < 0)
        return res;

    if((attrmask & (AVA_SIZE | AVA_BLKCNT)) != 0) {
        res = av_bzfile_size(fil->zfil, fil->node->cache, &size);
        if(res == 0 && size == -1) {
            fil->zfil = av_bzfile_new(fil->base);
            res = av_bzfile_size(fil->zfil, fil->node->cache, &size);
        }
        if(res < 0)
            return res;

        buf->size = size;
        buf->blocks = AV_BLOCKS(buf->size);
    }

    buf->mode &= ~(07000);
    buf->blksize = 4096;
    buf->dev = vf->mnt->avfs->dev;
    buf->ino = nod->ino;
    buf->nlink = 1;
    
    return 0;
}

extern int av_init_module_ubz2(struct vmodule *module);

int av_init_module_ubz2(struct vmodule *module)
{
    int res;
    struct avfs *avfs;
    struct ext_info ubz_exts[5];

    ubz_exts[0].from = ".bz2",  ubz_exts[0].to = NULL;
    ubz_exts[1].from = ".bz",   ubz_exts[1].to = NULL;
    ubz_exts[2].from = ".tbz2", ubz_exts[2].to = ".tar";
    ubz_exts[3].from = ".tbz",  ubz_exts[3].to = ".tar";
    ubz_exts[4].from = NULL;

    res = av_new_avfs("ubz2", ubz_exts, AV_VER, AVF_NOLOCK, module, &avfs);
    if(res < 0)
        return res;

    avfs->lookup   = bz_lookup;
    avfs->access   = bz_access;
    avfs->open     = bz_open;
    avfs->close    = bz_close; 
    avfs->read     = bz_read;
    avfs->getattr  = bz_getattr;

    av_add_avfs(avfs);

    return 0;
}