/*
 *  SSHTools - Java SSH2 API
 *
 *  Copyright (C) 2002-2003 Lee David Painter and Contributors.
 *
 *  Contributions made by:
 *
 *  Brett Smith
 *  Richard Pernavas
 *  Erwin Bolwidt
 *
 *  This program is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU General Public License
 *  as published by the Free Software Foundation; either version 2
 *  of the License, or (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */
package com.sshtools.j2ssh.sftp;

import com.sshtools.j2ssh.io.*;

import java.io.*;

import java.text.*;

import java.util.*;


/**
 *
 *
 * @author $author$
 * @version $Revision: 1.22 $
 */
public class FileAttributes {
    // Version 4 types

    public static final int SSH_FILEXFER_TYPE_REGULAR      = 1;
    public static final int SSH_FILEXFER_TYPE_DIRECTORY    = 2;
    public static final int SSH_FILEXFER_TYPE_SYMLINK      = 3;
    public static final int SSH_FILEXFER_TYPE_SPECIAL      = 4;
    public static final int SSH_FILEXFER_TYPE_UNKNOWN      = 5;
    public static final int SSH_FILEXFER_TYPE_SOCKET       = 6;
    public static final int SSH_FILEXFER_TYPE_CHAR_DEVICE  = 7;
    public static final int SSH_FILEXFER_TYPE_BLOCK_DEVICE = 8;
    public static final int SSH_FILEXFER_TYPE_FIFO         = 9;
    
    private static final int SSH_FILEXFER_ATTR_SIZE        = 0x0000001;
    private static final int SSH_FILEXFER_ATTR_UIDGID      = 0x00000002;
    private static final int SSH_FILEXFER_ATTR_PERMISSIONS = 0x0000004;

    // Also as ACMODTIME for version 3
    private static final int SSH_FILEXFER_ATTR_ACCESSTIME = 0x0000008;
    private static final int SSH_FILEXFER_ATTR_CREATETIME = 0x0000010;
    private static final int SSH_FILEXFER_ATTR_MODIFYTIME = 0x0000020;
    private static final int SSH_FILEXFER_ATTR_ACL = 0x0000040;
    private static final int SSH_FILEXFER_ATTR_OWNERGROUP = 0x0000080;
    private static final int SSH_FILEXFER_ATTR_EXTENDED = 0x8000000;

    // Posix stats

    public static final int S_IFMT = 0xF000;
    public static final int S_IFSOCK = 0xC000;
    public static final int S_IFLNK = 0xA000;
    public static final int S_IFREG = 0x8000;
    public static final int S_IFBLK = 0x6000;
    public static final int S_IFDIR = 0x4000;
    public static final int S_IFCHR = 0x2000;
    public static final int S_IFIFO = 0x1000;
    public final static int S_ISUID = 0x800;
    public final static int S_ISGID = 0x400;
    public final static int S_IRUSR = 0x100;
    public final static int S_IWUSR = 0x80;
    public final static int S_IXUSR = 0x40;
    public final static int S_IRGRP = 0x20;
    public final static int S_IWGRP = 0x10;
    public final static int S_IXGRP = 0x08;
    public final static int S_IROTH = 0x04;
    public final static int S_IWOTH = 0x02;
    public final static int S_IXOTH = 0x01;
    int version = 3;
    long flags = 0x0000000; // Version 3 & 4
    int type; // Version 4 only
    UnsignedInteger64 size = null; // Version 3 & 4
    UnsignedInteger32 uid = null; // Version 3 only
    UnsignedInteger32 gid = null; // Version 3 only
    String owner = null; // Version 4 only
    String group = null; // Version 4 only
    UnsignedInteger32 permissions = null; // Version 3 & 4
    UnsignedInteger32 atime = null; // Version 3 & 4
    UnsignedInteger32 createtime = null; // Version 4 only
    UnsignedInteger32 mtime = null; // Version 3 & 4
    List acl = new Vector(); // Version 4 only
    Map extended = new HashMap(); // Version 3 & 4
    char[] types = { 'p', 'c', 'd', 'b', '-', 'l', 's', };

    /**
     * Creates a new FileAttributes object.
     */
    public FileAttributes() {
    }

    /*
        public FileAttributes(int type, int version) {
       if(type >= 1 && type <= 5) {
         this.type = type;
         this.version = version;
       }
       else
         throw new IllegalArgumentException("The type must be a valid FileAttribute type");
        }
     */
    public FileAttributes(ByteArrayReader bar) throws IOException {
        flags = bar.readInt();

        /*
              type = bar.readInt();
         */
        if (isFlagSet(SSH_FILEXFER_ATTR_SIZE)) {
            size = bar.readUINT64();
        }

        if (isFlagSet(SSH_FILEXFER_ATTR_UIDGID)) {
            uid = bar.readUINT32();
            gid = bar.readUINT32();
        }

        /*if(isFlagSet(SSH_FILEXFER_ATTR_OWNERGROUP)) {
             owner = bar.readString();
             group = bar.readString();
              }*/
        if (isFlagSet(SSH_FILEXFER_ATTR_PERMISSIONS)) {
            permissions = bar.readUINT32();
        }

        if (isFlagSet(SSH_FILEXFER_ATTR_ACCESSTIME)) {
            atime = bar.readUINT32();
            mtime = bar.readUINT32();
        }

        /*
              if(isFlagSet(SSH_FILEXFER_ATTR_CREATETIME))
           createtime = bar.readUINT32();
              if(isFlagSet(SSH_FILEXFER_ATTR_MODIFYTIME))
           mtime = bar.readUINT32();
         */
        /*
              if(isFlagSet(SSH_FILEXFER_ATTR_ACL)) {
           int count = (int)bar.readInt();
           for(int i=0;i<count;i++) {
             acl.add(new ACL(bar));
           }
              }*/
        if (isFlagSet(SSH_FILEXFER_ATTR_EXTENDED)) {
            int count = (int) bar.readInt();
            String type;
            String data;

            for (int i = 0; i < count; i++) {
                type = bar.readString();
                data = bar.readString();
                extended.put(type, data);
            }
        }
    }

    /**
     *
     *
     * @return
     */
    public UnsignedInteger32 getUID() {
        if (uid != null) {
            return uid;
        } else {
            return new UnsignedInteger32(0);
        }
    }

    /**
     *
     *
     * @param uid
     */
    public void setUID(UnsignedInteger32 uid) {
        flags |= SSH_FILEXFER_ATTR_UIDGID;
        this.uid = uid;
    }

    /**
     *
     *
     * @param gid
     */
    public void setGID(UnsignedInteger32 gid) {
        flags |= SSH_FILEXFER_ATTR_UIDGID;
        this.gid = gid;
    }

    /**
     *
     *
     * @return
     */
    public UnsignedInteger32 getGID() {
        if (gid != null) {
            return gid;
        } else {
            return new UnsignedInteger32(0);
        }
    }

    /**
     *
     *
     * @param size
     */
    public void setSize(UnsignedInteger64 size) {
        this.size = size;

        // Set the flag
        if (size != null) {
            flags |= SSH_FILEXFER_ATTR_SIZE;
        } else {
            flags ^= SSH_FILEXFER_ATTR_SIZE;
        }
    }

    /**
     *
     *
     * @return
     */
    public UnsignedInteger64 getSize() {
        if (size != null) {
            return size;
        } else {
            return new UnsignedInteger64("0");
        }
    }

    /*
        public void setOwner(String owner) {
       this.owner = owner;
       // Set the flag
       if(group!=null || owner!=null)
         flags |= SSH_FILEXFER_ATTR_OWNERGROUP;
       else
         flags ^= SSH_FILEXFER_ATTR_OWNERGROUP;
        }
        public String getOwner() {
       return owner;
        }*/
    /*
        public void setGroup(String group) {
       this.group = group;
       // Set the flag
       if(group!=null || owner!=null)
         flags |= SSH_FILEXFER_ATTR_OWNERGROUP;
       else
         flags ^= SSH_FILEXFER_ATTR_OWNERGROUP;
        }
        public String getGroup() {
       return group;
        }*/
    public void setPermissions(UnsignedInteger32 permissions) {
        this.permissions = permissions;

        // Set the flag
        if (permissions != null) {
            flags |= SSH_FILEXFER_ATTR_PERMISSIONS;
        } else {
            flags ^= SSH_FILEXFER_ATTR_PERMISSIONS;
        }
    }

    /**
     * Set permissions given a UNIX style mask
     *
     * @param mask mask
     *
     * @throws IllegalArgumentException if badly formatted string
     */
    public void setPermissionsFromMaskString(String mask) {
        if (mask.length() != 4) {
            throw new IllegalArgumentException("Mask length must be 4");
        }

        try {
            setPermissions(new UnsignedInteger32(String.valueOf(
                        Integer.parseInt(mask, 8))));
        } catch (NumberFormatException nfe) {
            throw new IllegalArgumentException(
                "Mask must be 4 digit octal number.");
        }
    }

    /**
     *
     *
     * @param newPermissions
     */
    public void setPermissions(String newPermissions) {
        int cp = 0;

        if (permissions != null) {
            cp = cp |
                (((permissions.intValue() & S_IFMT) == S_IFMT) ? S_IFMT : 0);
            cp = cp |
                (((permissions.intValue() & S_IFSOCK) == S_IFSOCK) ? S_IFSOCK : 0);
            cp = cp |
                (((permissions.intValue() & S_IFLNK) == S_IFLNK) ? S_IFLNK : 0);
            cp = cp |
                (((permissions.intValue() & S_IFREG) == S_IFREG) ? S_IFREG : 0);
            cp = cp |
                (((permissions.intValue() & S_IFBLK) == S_IFBLK) ? S_IFBLK : 0);
            cp = cp |
                (((permissions.intValue() & S_IFDIR) == S_IFDIR) ? S_IFDIR : 0);
            cp = cp |
                (((permissions.intValue() & S_IFCHR) == S_IFCHR) ? S_IFCHR : 0);
            cp = cp |
                (((permissions.intValue() & S_IFIFO) == S_IFIFO) ? S_IFIFO : 0);
            cp = cp |
                (((permissions.intValue() & S_ISUID) == S_ISUID) ? S_ISUID : 0);
            cp = cp |
                (((permissions.intValue() & S_ISGID) == S_ISGID) ? S_ISGID : 0);
        }

        int len = newPermissions.length();

        if (len >= 1) {
            cp = cp |
                ((newPermissions.charAt(0) == 'r') ? FileAttributes.S_IRUSR : 0);
        }

        if (len >= 2) {
            cp = cp |
                ((newPermissions.charAt(1) == 'w') ? FileAttributes.S_IWUSR : 0);
        }

        if (len >= 3) {
            cp = cp |
                ((newPermissions.charAt(2) == 'x') ? FileAttributes.S_IXUSR : 0);
        }

        if (len >= 4) {
            cp = cp |
                ((newPermissions.charAt(3) == 'r') ? FileAttributes.S_IRGRP : 0);
        }

        if (len >= 5) {
            cp = cp |
                ((newPermissions.charAt(4) == 'w') ? FileAttributes.S_IWGRP : 0);
        }

        if (len >= 6) {
            cp = cp |
                ((newPermissions.charAt(5) == 'x') ? FileAttributes.S_IXGRP : 0);
        }

        if (len >= 7) {
            cp = cp |
                ((newPermissions.charAt(6) == 'r') ? FileAttributes.S_IROTH : 0);
        }

        if (len >= 8) {
            cp = cp |
                ((newPermissions.charAt(7) == 'w') ? FileAttributes.S_IWOTH : 0);
        }

        if (len >= 9) {
            cp = cp |
                ((newPermissions.charAt(8) == 'x') ? FileAttributes.S_IXOTH : 0);
        }

        setPermissions(new UnsignedInteger32(cp));
    }

    /**
     *
     *
     * @return
     */
    public UnsignedInteger32 getPermissions() {
        return permissions;
    }

    /**
     *
     *
     * @param atime
     * @param mtime
     */
    public void setTimes(UnsignedInteger32 atime, UnsignedInteger32 mtime) {
        this.atime = atime;
        this.mtime = mtime;

        // Set the flag
        if (atime != null) {
            flags |= SSH_FILEXFER_ATTR_ACCESSTIME;
        } else {
            flags ^= SSH_FILEXFER_ATTR_ACCESSTIME;
        }
    }

    /**
     *
     *
     * @return
     */
    public UnsignedInteger32 getAccessedTime() {
        return atime;
    }

    /**
     *
     *
     * @return
     */
    public UnsignedInteger32 getModifiedTime() {
        if (mtime != null) {
            return mtime;
        } else {
            return new UnsignedInteger32(0);
        }
    }

    /*
        public void setCreatedTime(UnsignedInteger32 atime) {
       this.createtime = createtime;
       // Set the flag
       if(createtime!=null)
         flags |= SSH_FILEXFER_ATTR_CREATETIME;
       else
         flags ^= SSH_FILEXFER_ATTR_CREATETIME;
        }
        public UnsignedInteger32 getCreatedTime() {
       return createtime;
        }
        public void setModifiedTime(UnsignedInteger32 atime) {
       this.mtime = mtime;
       // Set the flag
       if(mtime!=null)
         flags |= SSH_FILEXFER_ATTR_MODIFYTIME;
       else
         flags ^= SSH_FILEXFER_ATTR_MODIFYTIME;
        }
        public UnsignedInteger32 getModifiedTime() {
       return mtime;
        }*/
    /*
        public List getACL() {
       return acl;
        }*/
    public Map getExtendedAttributes() {
        return extended;
    }

    /**
     *
     *
     * @param flag
     *
     * @return
     */
    public boolean isFlagSet(int flag) {
        return ((flags & flag) == flag);
    }

    /**
     *
     *
     * @return
     *
     * @throws IOException
     */
    public byte[] toByteArray() throws IOException {
        ByteArrayWriter baw = new ByteArrayWriter();

        if (extended.size() > 0) {
            flags |= SSH_FILEXFER_ATTR_EXTENDED;

            /*
                  if(acl.size() > 0)
               flags |= SSH_FILEXFER_ATTR_ACL;
             */
        }
        
        baw.writeInt(flags);

        /*
              baw.write(type);
         */
        
        if (isFlagSet(SSH_FILEXFER_ATTR_SIZE)) {
            baw.writeUINT64(size);
        }

        if (isFlagSet(SSH_FILEXFER_ATTR_UIDGID)) {
            if (uid != null) {
                baw.writeUINT32(uid);
            } else {
                baw.writeInt(0);
            }

            if (gid != null) {
                baw.writeUINT32(gid);
            } else {
                baw.writeInt(0);
            }
        }

        /*
              if(isFlagSet(SSH_FILEXFER_ATTR_OWNERGROUP)) {
            baw.writeString(owner);
            baw.writeString(group);
              }
         */
        if (isFlagSet(SSH_FILEXFER_ATTR_PERMISSIONS)) {
            baw.writeUINT32(permissions);
        }

        if (isFlagSet(SSH_FILEXFER_ATTR_ACCESSTIME)) {
            baw.writeUINT32(atime);
            baw.writeUINT32(mtime);
        }

        /*
              if(isFlagSet(SSH_FILEXFER_ATTR_CREATETIME))
           baw.writeUINT32(createtime);
              if(isFlagSet(SSH_FILEXFER_ATTR_MODIFYTIME))
           baw.writeUINT32(mtime);
              if(isFlagSet(SSH_FILEXFER_ATTR_ACL)) {
           ByteArrayWriter acls = new ByteArrayWriter();
           acls.writeInt(acl.size());
           Iterator it = acl.iterator();
           while(it.hasNext()) {
             acls.write(((ACL)it.next()).toByteArray());
           }
           baw.writeBinaryString(acls.toByteArray());
           acls = null;
              }*/
        if (isFlagSet(SSH_FILEXFER_ATTR_EXTENDED)) {
            baw.writeInt(extended.size());

            Iterator it = extended.entrySet().iterator();
            Set set;

            while (it.hasNext()) {
                Map.Entry entry = (Map.Entry) it.next();
                baw.writeString((String) entry.getKey());
                baw.writeString((String) entry.getValue());
            }
        }

        return baw.toByteArray();
    }

    private int octal(int v, int r) {
        v >>>= r;

        return (((v & 0x04) != 0) ? 4 : 0) + (((v & 0x02) != 0) ? 2 : 0) +
        +(((v & 0x01) != 0) ? 1 : 0);
    }

    private String rwxString(int v, int r) {
        v >>>= r;

        String rwx = ((((v & 0x04) != 0) ? "r" : "-") +
            (((v & 0x02) != 0) ? "w" : "-"));

        if (((r == 6) && ((permissions.intValue() & S_ISUID) == S_ISUID)) ||
                ((r == 3) && ((permissions.intValue() & S_ISGID) == S_ISGID))) {
            rwx += (((v & 0x01) != 0) ? "s" : "S");
        } else {
            rwx += (((v & 0x01) != 0) ? "x" : "-");
        }

        return rwx;
    }

    /**
     *
     *
     * @return
     */
    public String getPermissionsString() {
        if (permissions != null) {
            StringBuffer str = new StringBuffer();
            str.append(types[(permissions.intValue() & S_IFMT) >>> 13]);
            str.append(rwxString(permissions.intValue(), 6));
            str.append(rwxString(permissions.intValue(), 3));
            str.append(rwxString(permissions.intValue(), 0));

            return str.toString();
        } else {
            return "";
        }
    }

    /**
     * Return the UNIX style mode mask
     *
     * @return mask
     */
    public String getMaskString() {
        StringBuffer buf = new StringBuffer();
        buf.append('0');

        int i = permissions.intValue();
        buf.append(octal(i, 6));
        buf.append(octal(i, 3));
        buf.append(octal(i, 0));

        return buf.toString();
    }

    /**
     *
     *
     * @return
     */
    public String getModTimeString() {
        if (mtime == null) {
            return "";
        }

        SimpleDateFormat df;
        long mt = (mtime.longValue() * 1000L);
        long now = System.currentTimeMillis();

        if ((now - mt) > (6 * 30 * 24 * 60 * 60 * 1000L)) {
            df = new SimpleDateFormat("MMM dd  yyyy");
        } else {
            df = new SimpleDateFormat("MMM dd hh:mm");
        }

        return df.format(new Date(mt));
    }

    /**
     *
     *
     * @return
     */
    public boolean isDirectory() {
      if (permissions == null) {
        return false;
      } 
      else {
        return ( ( permissions.intValue() &  S_IFMT ) == S_IFDIR );
      }
    }

    /**
     *
     *
     * @return
     */
    public boolean isFile() {
      if (permissions == null) {
        return false;
      } else {
        return (permissions.intValue() & S_IFMT ) == S_IFREG;
      }
    }

    /**
     *
     *
     * @return
     */
    public boolean isLink() {
      if (permissions == null) {
        return false;
      } else {
        return (permissions.intValue() & S_IFMT) == S_IFLNK;
      }
    }

    /**
     *
     *
     * @return
     */
    public boolean isFifo() {
      if (permissions == null) {
        return false;
      } else {
        return (permissions.intValue() & S_IFMT) == S_IFIFO;
      }
    }

    /**
     *
     *
     * @return
     */
    public boolean isBlock() {
      if (permissions == null) {
        return false;
      } else {
        return (permissions.intValue() & S_IFMT) == S_IFBLK;
      }
    }

    /**
     *
     *
     * @return
     */
    public boolean isCharacter() {
      if (permissions == null) {
        return false;
      } else {
        return (permissions.intValue() & S_IFMT) == S_IFCHR;
      }
    }

    /**
     *
     *
     * @return
     */
    public boolean isSocket() {
      if (permissions == null) {
        return false;
      } else {
        return (permissions.intValue() & S_IFMT) == S_IFSOCK;
      }
    }
}