/*      $NetBSD: lvm_vg.c,v 1.1.1.1 2009/12/02 00:26:15 haad Exp $      */

/*
* Copyright (C) 2008,2009 Red Hat, Inc. All rights reserved.
*
* This file is part of LVM2.
*
* This copyrighted material is made available to anyone wishing to use,
* modify, copy, or redistribute it subject to the terms and conditions
* of the GNU Lesser General Public License v.2.1.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

#include "lib.h"
#include "lvm2app.h"
#include "toolcontext.h"
#include "metadata-exported.h"
#include "archiver.h"
#include "locking.h"
#include "lvm-string.h"
#include "lvmcache.h"
#include "metadata.h"

#include <errno.h>
#include <string.h>

vg_t lvm_vg_create(lvm_t libh, const char *vg_name)
{
       struct volume_group *vg;

       vg = vg_create((struct cmd_context *)libh, vg_name);
       /* FIXME: error handling is still TBD */
       if (vg_read_error(vg)) {
               vg_release(vg);
               return NULL;
       }
       vg->open_mode = 'w';
       return (vg_t) vg;
}

int lvm_vg_extend(vg_t vg, const char *device)
{
       struct pvcreate_params pp;

       if (vg_read_error(vg))
               return -1;

       if (!vg_check_write_mode(vg))
               return -1;

       if (!lock_vol(vg->cmd, VG_ORPHANS, LCK_VG_WRITE)) {
               log_error("Can't get lock for orphan PVs");
               return -1;
       }

       pvcreate_params_set_defaults(&pp);
       if (!vg_extend(vg, 1, (char **) &device, &pp)) {
               unlock_vg(vg->cmd, VG_ORPHANS);
               return -1;
       }
       /*
        * FIXME: Either commit to disk, or keep holding VG_ORPHANS and
        * release in lvm_vg_close().
        */
       unlock_vg(vg->cmd, VG_ORPHANS);
       return 0;
}

int lvm_vg_reduce(vg_t vg, const char *device)
{
       if (vg_read_error(vg))
               return -1;
       if (!vg_check_write_mode(vg))
               return -1;

       if (!vg_reduce(vg, (char *)device))
               return -1;
       return 0;
}

int lvm_vg_set_extent_size(vg_t vg, uint32_t new_size)
{
       if (vg_read_error(vg))
               return -1;
       if (!vg_check_write_mode(vg))
               return -1;

       if (!vg_set_extent_size(vg, new_size))
               return -1;
       return 0;
}

int lvm_vg_write(vg_t vg)
{
       struct pv_list *pvl;

       if (vg_read_error(vg))
               return -1;
       if (!vg_check_write_mode(vg))
               return -1;

       if (dm_list_empty(&vg->pvs)) {
               if (!vg_remove(vg))
                       return -1;
               return 0;
       }

       if (! dm_list_empty(&vg->removed_pvs)) {
               if (!lock_vol(vg->cmd, VG_ORPHANS, LCK_VG_WRITE)) {
                       log_error("Can't get lock for orphan PVs");
                       return 0;
               }
       }

       if (!archive(vg))
               return -1;

       /* Store VG on disk(s) */
       if (!vg_write(vg) || !vg_commit(vg))
               return -1;

       if (! dm_list_empty(&vg->removed_pvs)) {
               dm_list_iterate_items(pvl, &vg->removed_pvs) {
                       pv_write_orphan(vg->cmd, pvl->pv);
                       /* FIXME: do pvremove / label_remove()? */
               }
               dm_list_init(&vg->removed_pvs);
               unlock_vg(vg->cmd, VG_ORPHANS);
       }

       return 0;
}

int lvm_vg_close(vg_t vg)
{
       if (vg_read_error(vg) == FAILED_LOCKING)
               vg_release(vg);
       else
               unlock_and_release_vg(vg->cmd, vg, vg->name);
       return 0;
}

int lvm_vg_remove(vg_t vg)
{
       if (vg_read_error(vg))
               return -1;
       if (!vg_check_write_mode(vg))
               return -1;

       if (!vg_remove_check(vg))
               return -1;

       return 0;
}

vg_t lvm_vg_open(lvm_t libh, const char *vgname, const char *mode,
                 uint32_t flags)
{
       uint32_t internal_flags = 0;
       struct volume_group *vg;

       if (!strncmp(mode, "w", 1))
               internal_flags |= READ_FOR_UPDATE;
       else if (strncmp(mode, "r", 1)) {
               log_errno(EINVAL, "Invalid VG open mode");
               return NULL;
       }

       vg = vg_read((struct cmd_context *)libh, vgname, NULL, internal_flags);
       if (vg_read_error(vg)) {
               /* FIXME: use log_errno either here in inside vg_read */
               vg_release(vg);
               return NULL;
       }
       /* FIXME: combine this with locking ? */
       vg->open_mode = mode[0];

       return (vg_t) vg;
}

struct dm_list *lvm_vg_list_pvs(vg_t vg)
{
       struct dm_list *list;
       pv_list_t *pvs;
       struct pv_list *pvl;

       if (dm_list_empty(&vg->pvs))
               return NULL;

       if (!(list = dm_pool_zalloc(vg->vgmem, sizeof(*list)))) {
               log_errno(ENOMEM, "Memory allocation fail for dm_list.");
               return NULL;
       }
       dm_list_init(list);

       dm_list_iterate_items(pvl, &vg->pvs) {
               if (!(pvs = dm_pool_zalloc(vg->vgmem, sizeof(*pvs)))) {
                       log_errno(ENOMEM,
                               "Memory allocation fail for lvm_pv_list.");
                       return NULL;
               }
               pvs->pv = pvl->pv;
               dm_list_add(list, &pvs->list);
       }
       return list;
}

struct dm_list *lvm_vg_list_lvs(vg_t vg)
{
       struct dm_list *list;
       lv_list_t *lvs;
       struct lv_list *lvl;

       if (dm_list_empty(&vg->lvs))
               return NULL;

       if (!(list = dm_pool_zalloc(vg->vgmem, sizeof(*list)))) {
               log_errno(ENOMEM, "Memory allocation fail for dm_list.");
               return NULL;
       }
       dm_list_init(list);

       dm_list_iterate_items(lvl, &vg->lvs) {
               if (!(lvs = dm_pool_zalloc(vg->vgmem, sizeof(*lvs)))) {
                       log_errno(ENOMEM,
                               "Memory allocation fail for lvm_lv_list.");
                       return NULL;
               }
               lvs->lv = lvl->lv;
               dm_list_add(list, &lvs->list);
       }
       return list;
}

uint64_t lvm_vg_get_seqno(const vg_t vg)
{
       return vg_seqno(vg);
}

uint64_t lvm_vg_is_clustered(const vg_t vg)
{
       return vg_is_clustered(vg);
}

uint64_t lvm_vg_is_exported(const vg_t vg)
{
       return vg_is_exported(vg);
}

uint64_t lvm_vg_is_partial(const vg_t vg)
{
       return (vg_missing_pv_count(vg) != 0);
}

/* FIXME: invalid handle? return INTMAX? */
uint64_t lvm_vg_get_size(const vg_t vg)
{
       return vg_size(vg);
}

uint64_t lvm_vg_get_free_size(const vg_t vg)
{
       return vg_free(vg);
}

uint64_t lvm_vg_get_extent_size(const vg_t vg)
{
       return vg_extent_size(vg);
}

uint64_t lvm_vg_get_extent_count(const vg_t vg)
{
       return vg_extent_count(vg);
}

uint64_t lvm_vg_get_free_extent_count(const vg_t vg)
{
       return vg_free_count(vg);
}

uint64_t lvm_vg_get_pv_count(const vg_t vg)
{
       return vg_pv_count(vg);
}

uint64_t lvm_vg_get_max_pv(const vg_t vg)
{
       return vg_max_pv(vg);
}

uint64_t lvm_vg_get_max_lv(const vg_t vg)
{
       return vg_max_lv(vg);
}

char *lvm_vg_get_uuid(const vg_t vg)
{
       char uuid[64] __attribute((aligned(8)));

       if (!id_write_format(&vg->id, uuid, sizeof(uuid))) {
               log_error("Internal error converting uuid");
               return NULL;
       }
       return strndup((const char *)uuid, 64);
}

char *lvm_vg_get_name(const vg_t vg)
{
       char *name;

       name = dm_malloc(NAME_LEN + 1);
       strncpy(name, (const char *)vg->name, NAME_LEN);
       name[NAME_LEN] = '\0';
       return name;
}

/*
* FIXME: These functions currently return hidden VGs.  We should either filter
* these out and not return them in the list, or export something like
* is_orphan_vg and tell the caller to filter.
*/
struct dm_list *lvm_list_vg_names(lvm_t libh)
{
       return get_vgnames((struct cmd_context *)libh, 0);
}

struct dm_list *lvm_list_vg_uuids(lvm_t libh)
{
       return get_vgids((struct cmd_context *)libh, 0);
}

/*
* FIXME: Elaborate on when to use, side-effects, .cache file, etc
*/
int lvm_scan(lvm_t libh)
{
       if (!lvmcache_label_scan((struct cmd_context *)libh, 2))
               return -1;
       return 0;
}