restaurant-search/server.go

/*
 * Copyright (c) 2015 Alex Yatskov <alex@foosoft.net>
 * Author: Alex Yatskov <alex@foosoft.net>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of
 * this software and associated documentation files (the "Software"), to deal in
 * the Software without restriction, including without limitation the rights to
 * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
 * the Software, and to permit persons to whom the Software is furnished to do so,
 * subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
 * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
 * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
 * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
 * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

package main

import (
	"database/sql"
	"encoding/json"
	_ "github.com/go-sql-driver/mysql"
	"log"
	"net/http"
	"os"
	"path/filepath"
	"strings"
)

var db *sql.DB

func executeQuery(rw http.ResponseWriter, req *http.Request) {
	type JsonFeatureMap map[string]float64

	type JsonRange struct {
		Max float64 `json:"max"`
		Min float64 `json:"min"`
	}

	type JsonGeo struct {
		Latitude  float64 `json:"latitude"`
		Longitude float64 `json:"longitude"`
		Valid     bool    `json:"valid"`
	}

	type JsonRequest struct {
		Features    JsonFeatureMap `json:"features"`
		Geo         *JsonGeo       `json:"geo"`
		HintSteps   int            `json:"hintSteps"`
		MaxResults  int            `json:"maxResults"`
		MinScore    float64        `json:"minScore"`
		Profile     JsonFeatureMap `json:"profile"`
		Range       JsonRange      `json:"range"`
		WalkingDist float64        `json:"walkingDist"`
	}

	var request JsonRequest
	if err := json.NewDecoder(req.Body).Decode(&request); err != nil {
		http.Error(rw, err.Error(), http.StatusInternalServerError)
		return
	}

	log.Print("Ready")

	// function runQuery(query, callback) {
	//     query.profile  = fixupProfile(query.profile);
	//     query.features = fixupFeatures(query.features);

	//     var context = {
	//         geo:         query.geo,
	//         profile:     query.profile,
	//         walkingDist: query.walkingDist * 1000.0
	//     };

	//     getRecords(context, function(data) {
	//         var searchResults = findRecords(
	//             data,
	//             query.features,
	//             query.minScore
	//         );

	//         var graphColumns = {};
	//         for (var feature in query.features) {
	//             var searchHints = buildHints(
	//                 data,
	//                 query.features,
	//                 feature,
	//                 query.minScore,
	//                 query.range,
	//                 query.hintSteps
	//             );

	//             graphColumns[feature] = {
	//                 value: query.features[feature],
	//                 hints: searchHints,
	//                 steps: query.hintSteps
	//             };
	//         }

	//         callback({
	//             columns: graphColumns,
	//             items:   searchResults.slice(0, query.maxResults),
	//             count:   searchResults.length
	//         });
	//     });
	// }
}

func getCategories(rw http.ResponseWriter, req *http.Request) {
	type JsonCategory struct {
		Description string `json:"description"`
		Id          int    `json:"id"`
	}

	rows, err := db.Query("SELECT * FROM categories")
	if err != nil {
		log.Fatal(err)
	}
	defer rows.Close()

	var categories []JsonCategory
	for rows.Next() {
		var (
			description string
			id          int
		)

		if err := rows.Scan(&description, &id); err != nil {
			log.Fatal(err)
		}

		categories = append(categories, JsonCategory{description, id})
	}

	if err := rows.Err(); err != nil {
		log.Fatal(err)
	}

	js, err := json.Marshal(categories)
	if err != nil {
		log.Fatal(err)
	}

	rw.Header().Set("Content-Type", "application/json")
	rw.Write(js)
}

func addCategory(rw http.ResponseWriter, req *http.Request) {
	type JsonRequest struct {
		Description string `json:"description"`
	}

	type JsonResponse struct {
		Description string `json:"description"`
		Id          int    `json:"id"`
		Success     bool   `json:"success"`
	}

	var request JsonRequest
	if err := json.NewDecoder(req.Body).Decode(&request); err != nil {
		http.Error(rw, err.Error(), http.StatusInternalServerError)
		return
	}

	response := JsonResponse{Description: strings.TrimSpace(request.Description)}

	if len(request.Description) > 0 {
		result, err := db.Exec("INSERT INTO categories(description) VALUES(?)", request.Description)
		if err != nil {
			log.Fatal(err)
		}

		insertId, err := result.LastInsertId()
		if err != nil {
			log.Fatal(err)
		}

		affectedRows, err := result.RowsAffected()
		if err != nil {
			log.Fatal(err)
		}

		response.Success = affectedRows > 0
		response.Id = int(insertId)
	}

	js, err := json.Marshal(response)
	if err != nil {
		log.Fatal(err)
	}

	rw.Header().Set("Content-Type", "application/json")
	rw.Write(js)
}

func removeCategory(rw http.ResponseWriter, req *http.Request) {
	type JsonRequest struct {
		Id int `json:"id"`
	}

	type JsonResponse struct {
		Success bool `json:"success"`
	}

	var request JsonRequest
	if err := json.NewDecoder(req.Body).Decode(&request); err != nil {
		http.Error(rw, err.Error(), http.StatusInternalServerError)
		return
	}

	result, err := db.Exec("DELETE FROM categories WHERE id = (?)", request.Id)
	if err != nil {
		log.Fatal(err)
	}

	affectedRows, err := result.RowsAffected()
	if err != nil {
		log.Fatal(err)
	}

	js, err := json.Marshal(JsonResponse{affectedRows > 0})
	if err != nil {
		log.Fatal(err)
	}

	rw.Header().Set("Content-Type", "application/json")
	rw.Write(js)
}

func accessReview(rw http.ResponseWriter, req *http.Request) {
	// function accessReview(query, callback) {
	//     query.profile = fixupProfile(query.profile);

	//     pool.query('SELECT url FROM reviews WHERE id = (?) LIMIT 1', [query.id], function(err, rows) {
	//         if (err) {
	//             throw err;
	//         }

	//         var results = {
	//             success: rows.length > 0
	//         };

	//         if (results.success) {
	//             results.url = 'http://www.tripadvisor.com' + rows[0].url;

	//             pool.query('UPDATE reviews SET accessCount = accessCount + 1 WHERE id = (?)', [query.id], function(err, info) {
	//                 if (_.keys(query.profile).length > 0) {
	//                     pool.query('INSERT INTO history(date, reviewId) VALUES(NOW(), ?)', [query.id], function(err, info) {
	//                         if (err) {
	//                             throw err;
	//                         }

	//                         for (var categoryId in query.profile) {
	//                             pool.query(
	//                                 'INSERT INTO historyGroups(categoryId, categoryValue, historyId) VALUES(?, ?, ?)',
	//                                 [categoryId, query.profile[categoryId], info.insertId]
	//                             );
	//                         }
	//                     });
	//                 }
	//             });
	//         }

	//         callback(results);
	//     });
	// }
}

func staticPath() (string, error) {
	if len(os.Args) > 1 {
		return os.Args[1], nil
	}

	return filepath.Abs(filepath.Join(filepath.Dir(os.Args[0]), "static"))
}

func main() {
	dir, err := staticPath()
	if err != nil {
		log.Fatal(err)
	}

	db, err = sql.Open("mysql", "hscd@/hscd")
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	http.HandleFunc("/query", executeQuery)
	http.HandleFunc("/categories", getCategories)
	http.HandleFunc("/learn", addCategory)
	http.HandleFunc("/forget", removeCategory)
	http.HandleFunc("/access", accessReview)
	http.Handle("/", http.FileServer(http.Dir(dir)))

	log.Fatal(http.ListenAndServe(":3000", nil))
}