HORIZON PROJECT

/*

    TODO

    * 스키마 정의 함수 제작

    *

*/

var mongoose = require('mongoose');

var fs = require('fs');

var mongod = require('mongod');

// Client 스키마 정의

var clientSchema = mongoose.Schema({

    _ip: String,

    _mac: String,

    _registerd: String

});

var ClientModel;

var dbCenter = {};

var MongoClient = require('mongodb').MongoClient;

var urlgogo = "mongodb://localhost:27017/mydb";

// ▼ DB 접속 관련 함수 ▼//////////////////////////////////////////////////////////////////////////////////////////////////////////

// DB 를 Create 해준다.

dbCenter.createDB = function (url, callback){

    MongoClient.connect(urlgogo, function(err, db){

       if( err ){

           callback(err);

           throw err;

       } else {

           console.log("Database Created");

           db.close();

       }

    });

}

// cmd 에서 mongod 설정해주지 않는 역할을 한다.

// db 에 연결한다.

dbCenter.db_Open = function(callback){

    console.log('[dbCenter] db_Open 호출');

    var server = new mongod({

    dbpath:'/Users/spboy/CMS/clients',

    port:27017});

    server.open((err) => {

       if(err) {

           console.log('[dbCenter][db_Open] err : ' + err.toString());

           callback(err);

       } else {

           console.log('[dbCenter][db_Open] dbServer opened');

           callback();

       }

    });

}

// Server 와 DB 를 연결해 준다.

dbCenter.db_Connect = function(callback){

    console.log('[dbCenter] db_Connect 호출');

    // DB 서버를 오픈한다.

    dbCenter.db_Open(function(err){

        if(err){

            callback(err);

            return;

        } else {

            var dbUrl = 'mongodb://localhost:27017/local';

            mongoose.Promise = global.Promise;

            mongoose.connect(dbUrl);

            database = mongoose.connection;

            database.on('open', function(){

                console.log('[dbCenter][db_Connect] 데이터베이스에 연결됨 : ' + dbUrl);

                UserSchema = clientSchema;

                console.log('[dbCenter][db_Connect] Client Schema 정의함.');

                ClientModel = mongoose.model('clients', UserSchema);

                saveData = new ClientModel();

                console.log('[dbCenter][db_Connect] ClientModel 정의함');

            });

            database.on('disconnected', function(){

                console.log('[dbCenter][db_Connect] 데이터베이스 연결 끊어짐.');

            });

            database.on('error', console.error.bind(console, 'mongoose connection error.'));

            callback(null);

        }

    });

}

// DB 에 연결이 되어 있는지 확인하고 되어 있지 않으면 연결해준다.

dbCenter.checkConnectDB = function(callback){

    console.log('[dbCenter] checkConnectDB 호출');

    if(UserModel){

        console.log('서버 연결이 이미 되어 있습니다.');

        callback('success');

    } else {

        dbCenter.connect(function(result){

            if(result == 'success'){

                console.log('서버 연결 성공');

                callback('success');

            } else {

                console.log('서버 연결 실패');

                callback('fail');

            }

        });

    }

};

// ▲ DB 접속 관련 함수 ▲//////////////////////////////////////////////////////////////////////////////////////////////////////////

// ▼ 데이터 관리 함수 ▼//////////////////////////////////////////////////////////////////////////////////////////////////////////

/*

    * 데이터 저장 순서 *

    1. data_set 으로 데이터를 셋팅해준다.

    2. data_Add 으로 데이터를 저장해준다.

*/

var saveData; // 정보를 모아서 저장할 객체

// 데이터 셋팅

// data : 저장하고자 하는 데이터의 json 형태

dbCenter.data_Set = function (data, callback){

    console.log('[dbCenter] data_Set 호출');

    // TODO : 클라이언트 모델만으로 정의 되어 있는데 다른 것으로 바뀌었을 경우에는 어떻게 할것인가?

    saveData = new ClientModel();

    // json 데이터의 위치에 맞게 넣어준다.

    try {

        saveData._ip = data.body._ip;

        saveData._mac = data.body._mac;

        saveData._registerd = data.body._registerd;

    } catch(err) {

        console.log('[dbCenter][data_Set] err : ' + err.toString());

        callback(err, null);

        return;

    }

    // 데이터 셋팅이 끝나면 중복검사를 해준다.

    dbCenter.data_Find(saveData,function(err, docs){

        if(err != null && docs == null){

            // 에러만이 존재한다.

            callback(err, null);

        } else if ( err == null && docs != null ) {

            // 찾고자 하는 데이터가 있다.

            callback(err, null);

        } else if ( err == null && docs == null ){

            // 찾고자하는 데이터가 없다.

            // 데이터 셋팅이 끝나게 되면 데이터를 저장해준다.

            dbCenter.data_Add(saveData, function(err){

                if(err){

                } else {

                    callback(null);

                }

            });

        }

    });

};

// 데이터 추가

// data : 저장하고자 하는 데이터 / mongoose.model 형태

dbCenter.data_Add = function(data, callback){

    console.log('[dbCenter] data_Add 호출');

    data.save(function(err){

        if(err){

            console.log('[dbCenter][data_Add] err : ' + err.toString());

            callback(err);

            return;

        } else {

            console.log('[dbCenter][data_Add] 사용자 데이터 추가함.');

            callback(null);

        }

    });

};

// 검색 기능

dbCenter.data_Find = function(req, callback){

    console.log('[dbCenter] data_Find 호출');

    // _id 를 기준으로 찾아옵니다. _id는 몽고디비에서 설정해 주는 유니크한 값입니다.

    ClientModel.find({"_mac": req._mac}, function(err, docs){

        if(err){

            console.log("[dbCenter][data_Find] err : " + err.toString());

            callback(err, null);

            return;

        }

        if(docs.length > 0){

            console.log('[dbCenter][data_Find] 찾았습니다. : \n' + docs);

            callback(null, docs);

        }else{

            console.log('[dbCenter][data_Find] 찾지 못했습니다');

            callback(null, null);

        }

    });

};

// 모든 데이터 가져오기

dbCenter.data_All = function(callback){

    console.log('[dbCenter] data_All 호출');

    // ClientModel 에서 모든걸 다 가져온다.

    ClientModel.find({}, function(err, users){

        if(err){

            console.log('[dbCenter][data_All] err : ' + err.toString());

            callback(err, null);

        } else {

            console.log('[dbCenter][data_All] 데이터 전송 완료');

            callback(null, users);

        }

    });

};

// ▲ 데이터 관리 함수 ▲//////////////////////////////////////////////////////////////////////////////////////////////////////////

// 메인 이미지 url 설정

dbCenter.setMainImageUrl = function(data, callback){

    console.log('[connectDB] setMainImageUrl 호출');

    try{

        saveData._mainimg = data;

        callback(null);

    } catch (err){

        callback(err);

    }

};

// 콘탠츠 이미지 rul 설정

dbCenter.setContentsUrl = function(type, url,  callback){

    console.log('[connectDB] setContentsUrl 호출');

    try{

        var content = {"_type":type, "_url":url}

        console.log(JSON.stringify(content));

        saveData._contents.push(content);

        callback(null);

    } catch (err){

        callback(err);

    }

};

// 삭제

dbCenter.deleteData = function(data, callback){

    console.log('[connectDB] deleteData 호출');

    for( var cnt = 0 ; cnt < data.length; cnt++)

    {

    // 1. 메인 이미지를 삭제해준다.

        var mainImageUrl = data[cnt]._mainimg;

        fs.unlink(mainImageUrl, function(err){

            if(err){

                console.log('mainImage 삭제 실패 : ' + err.toString());

            } else {

                console.log('mainImage 삭제 성공');

            }

        });

        // 2. 콘탠츠 이미지를 삭제해준다.

        for(var i = 0; i< data[cnt]._contents.length; i++)

        {

            var url = data[cnt]._contents[i]._url;

            fs.unlink(url, function(err){

                if(err){

                    console.log('contents 삭제 실패 : ' + err.toString());

                } else {

                    console.log('contents 삭제 성공');

                }

            });

        }

        // 데이터에서 삭제해준다.

        UserModel.remove({"_id": data[cnt]._id}, function(err, docs){

            if(err){

                callback(err, null);

            }else{

                callback(null, docs);

            }

        });

    }

};

var tmpModifyData;

var orgModifyData;

// 데이터 수정하기

dbCenter.modifyData = function(org, dst, callback){

    console.log('[connectDB] modifyData 호출');

    tmpModifyData = org[0];

    orgModifyData = org[0];

    tmpModifyData._contents = [];

    //org[0]._title_kr  = dst.body._title_kr;

    callback(null, null);

};

// 메인 이미지 url 설정

dbCenter.setModifyMainImageUrl = function(data, callback){

    console.log('[connectDB] setModifyMainImageUrl 호출');

    try{

        tmpModifyData._mainimg = data;

        callback(null, tmpModifyData);

    } catch (err){

        callback(err, null);

    }

};

// 콘탠츠 이미지 rul 설정

dbCenter.seModifytContentsUrl = function(type, url,  callback){

    console.log('[connectDB] seModifytContentsUrl 호출');

    try{

        var content = {"_type":type, "_url":url}

        console.log(JSON.stringify(content));

        tmpModifyData._contents.push(content);

        callback(null);

    } catch (err){

        callback(err);

    }

};

dbCenter.dataModifyInDB = function(callback){

      UserModel.where({_id : tmpModifyData._id}).update(tmpModifyData, function(err, docs){

        if(err){

            callback(err, null);

        }else{

            // 1. 메인 이미지를 삭제해준다.

            var mainImageUrl = orgModifyData._mainimg;

            fs.unlink(mainImageUrl, function(err){

                if(err){

                    console.log('mainImage 삭제 실패 : ' + err.toString());

                } else {

                    console.log('mainImage 삭제 성공');

                }

            });

            // 2. 콘탠츠 이미지를 삭제해준다.

            for(var i = 0; i< orgModifyData._contents.length; i++)

            {

                var url = orgModifyData._contents[i]._url;

                fs.unlink(url, function(err){

                    if(err){

                        console.log('contents 삭제 실패 : ' + err.toString());

                    } else {

                        console.log('contents 삭제 성공');

                    }

                });

            }

            callback(null, docs);

        }

    });

}

module.exports = dbCenter;