package com.company.iptest;

import com.mongodb.BasicDBObject;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.Mongo;
import com.mongodb.MongoURI;
import com.mongodb.ServerAddress;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.net.UnknownHostException;

public class IpTest {

    private static final Logger log = LoggerFactory.getLogger(IpTest.class);

    public static final String _mongoUriString = "mongodb://username:password@localhost:27017/database";

    public static MongoURI _mongoUri;
    public static Mongo.Holder _mongoHolder;
    public static Mongo _mongo;
    public static DB _db;
    public static DBCollection _collection;

    static {
        try {
            _mongoUri = new MongoURI(_mongoUriString);
            _mongoHolder = new Mongo.Holder();
            _mongo = _mongoHolder.connect(_mongoUri);
            _mongo.getDB(_mongoUri.getDatabase()).authenticate(_mongoUri.getUsername(), _mongoUri.getPassword());
            _db = _mongo.getDB("database");
            _collection = _db.getCollection("stuff");
        } catch ( Exception e ) {
            log.error("Unexpected exception", e);
        }
    }

    public static DBCollection getStuffCollection ( String mongoUriString )
            throws UnknownHostException {
        return getStuffCollection(new MongoURI(mongoUriString));
    }

    public static DBCollection getStuffCollection ( MongoURI mongoUri )
            throws UnknownHostException {
        DBCollection c = getStuffCollection(new Mongo(mongoUri));
        DB db = c.getDB();
        if ( !db.isAuthenticated() ) {
            db = db.getMongo().getDB(mongoUri.getDatabase());
            db.authenticate(mongoUri.getUsername(), mongoUri.getPassword());
        }
        return c;
    }

    public static DBCollection getStuffCollection ( Mongo.Holder mongoHolder )
            throws UnknownHostException {
        return getStuffCollection(mongoHolder.connect(new MongoURI(_mongoUriString)));
    }

    public static DBCollection getStuffCollection ( Mongo mongo ) {
        return getStuffCollection(mongo.getDB("database"));
    }

    public static DBCollection getStuffCollection ( DB db ) {
        return db.getCollection("stuff");
    }

    public static DBObject findStuff ( DBCollection stuffCollection ) {
        return stuffCollection.findOne();
    }

}