package com.lumaserv.bgp.protocol.attribute;

import lombok.NoArgsConstructor;
import lombok.Getter;
import lombok.Setter;

import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.List;

@NoArgsConstructor
@Getter
@Setter
public class ASPathAttribute implements PathAttribute {

    List<Segment> segments = new ArrayList<>();

    public ASPathAttribute(byte typeCode, byte[] data) {
        int offset = 0;
        while (offset < data.length) {
            Segment segment = new Segment().setType(data[offset]);
            byte length = data[offset + 1];
            for(int i=0; i<length; i++)
                segment.asns.add(((data[offset + 2 + (i * 2)] & 0xFF) << 8) | (data[offset + 3 + (i * 2)] & 0xFF));
            offset += 2 + (length * 2);
            segments.add(segment);
        }
    }

    public byte getTypeCode() {
        return 2;
    }

    public void build(ByteBuffer b) {
        for (Segment segment: segments) {
            b.put(segment.getType());
            b.put((byte)segment.getAsns().size());
            for (Integer asn: segment.getAsns()) {
                b.putShort(asn.shortValue());
            }
        }
    }

    @Getter
    @Setter
    public static class Segment {
        byte type;
        List<Integer> asns = new ArrayList<>();

        public String toString() {
            StringBuffer buf = new StringBuffer();

            buf.append("(");
            for(Integer asn : asns) {
                if (buf.length() > 1)
                    buf.append(" ");
                buf.append(asn);
            }
            buf.append(")");
            return buf.toString();
        }
    }

    public String toString() {
        StringBuffer buf = new StringBuffer();

        buf.append("[");
        for(Segment segment : segments) {
            if (buf.length() > 1)
                buf.append(", ");
            buf.append(segment);
        }
        buf.append("]");
        return buf.toString();
    }
}