#!/usr/bin/env python3

import os
import logging


class DataFace(object):
    """ Some data IFace bowtie can work with.
    """
    
    def size(self):
        raise NotImplementedError("size() needs to be implemented!")
 
 
 
class SamFile(DataFace):
    
    def __init__(self,sam_desc):
        self.file_desc = sam_desc
        
        
    def size(self):
        if hasattr(self,'no_frags'):
            return self.no_frags
        
        count = 0
        try:
            fh = open(self.file_desc,"r")
            for line in fh:
                if line[0] != "@":
                    count += 1
        except:
            logging.error("Exception reading sam file!")
            fh.close()
            raise
        else:
            fh.close()
    
        self.no_frags = count
        return count
            


class FastaQFile(DataFace):
    
    def __init__(self,fastq_desc):
        self.file_desc = fastq_desc
        
        
    def size(self):
        if hasattr(self,'no_seqs'):
            return self.no_seqs
        
        count = 0
        try:
            fh = open(self.file_desc,"r")
            lno = 0
            for line in fh:
                if lno == 0:
                    if line[0] != '@':
                        raise TypeError("This does not look like a fastq file!")
                    count += 1
                if lno == 2:
                    if line[0] != '+':
                        raise TypeError("This does not look like a fastq file!")
                lno = (lno + 1) % 4 
        except:
            logging.error("Exception reading fastq file!")
            fh.close()
            raise
        else:
            fh.close()
    
        self.no_seqs = count
        return self.no_seqs