frontend
This commit is contained in:
@@ -0,0 +1,52 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type newsItem struct {
|
||||
ID int `json:"id"`
|
||||
Ticker string `json:"ticker"`
|
||||
Headline string `json:"headline"`
|
||||
Source string `json:"source"`
|
||||
URL string `json:"url"`
|
||||
Sentiment string `json:"sentiment"`
|
||||
PublishedAt string `json:"published_at"`
|
||||
}
|
||||
|
||||
func (s *Server) handleGetNews(w http.ResponseWriter, r *http.Request) {
|
||||
ticker := r.URL.Query().Get("ticker")
|
||||
|
||||
var rows *sql.Rows
|
||||
var err error
|
||||
|
||||
if ticker != "" {
|
||||
rows, err = s.db.Query(`
|
||||
SELECT id, COALESCE(ticker,''), headline, COALESCE(source,''), COALESCE(url,''), COALESCE(sentiment,''), COALESCE(published_at,'')
|
||||
FROM news WHERE ticker = ? ORDER BY published_at DESC LIMIT 100`, ticker)
|
||||
} else {
|
||||
rows, err = s.db.Query(`
|
||||
SELECT id, COALESCE(ticker,''), headline, COALESCE(source,''), COALESCE(url,''), COALESCE(sentiment,''), COALESCE(published_at,'')
|
||||
FROM news ORDER BY published_at DESC LIMIT 100`)
|
||||
}
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
items := []newsItem{}
|
||||
for rows.Next() {
|
||||
var it newsItem
|
||||
if err := rows.Scan(&it.ID, &it.Ticker, &it.Headline, &it.Source, &it.URL, &it.Sentiment, &it.PublishedAt); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
items = append(items, it)
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(items)
|
||||
}
|
||||
@@ -0,0 +1,71 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
|
||||
"github.com/gorilla/mux"
|
||||
)
|
||||
|
||||
type watchlistItem struct {
|
||||
ID int `json:"id"`
|
||||
Ticker string `json:"ticker"`
|
||||
Name string `json:"name"`
|
||||
Sector string `json:"sector"`
|
||||
Exchange string `json:"exchange"`
|
||||
Active int `json:"active"`
|
||||
}
|
||||
|
||||
func (s *Server) handleGetWatchlist(w http.ResponseWriter, r *http.Request) {
|
||||
rows, err := s.db.Query(`SELECT id, ticker, COALESCE(name,''), COALESCE(sector,''), COALESCE(exchange,''), active FROM watchlist WHERE active=1 ORDER BY ticker`)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
items := []watchlistItem{}
|
||||
for rows.Next() {
|
||||
var it watchlistItem
|
||||
if err := rows.Scan(&it.ID, &it.Ticker, &it.Name, &it.Sector, &it.Exchange, &it.Active); err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
items = append(items, it)
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(items)
|
||||
}
|
||||
|
||||
func (s *Server) handleAddWatchlist(w http.ResponseWriter, r *http.Request) {
|
||||
var body struct {
|
||||
Ticker string `json:"ticker"`
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&body); err != nil || body.Ticker == "" {
|
||||
http.Error(w, "ticker required", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
|
||||
_, err := s.db.Exec(`INSERT OR IGNORE INTO watchlist (ticker) VALUES (?)`, body.Ticker)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusCreated)
|
||||
w.Write([]byte(`{"status":"added"}`))
|
||||
}
|
||||
|
||||
func (s *Server) handleRemoveWatchlist(w http.ResponseWriter, r *http.Request) {
|
||||
ticker := mux.Vars(r)["ticker"]
|
||||
_, err := s.db.Exec(`DELETE FROM watchlist WHERE ticker = ?`, ticker)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.Write([]byte(`{"status":"removed"}`))
|
||||
}
|
||||
@@ -34,19 +34,42 @@ func New(database *db.DB, port string) (*Server, error) {
|
||||
}
|
||||
|
||||
func (s *Server) setupRoutes() {
|
||||
s.router.Use(corsMiddleware)
|
||||
|
||||
api := s.router.PathPrefix("/api").Subrouter()
|
||||
api.HandleFunc("/health", s.handleHealth).Methods("GET")
|
||||
api.HandleFunc("/health", s.handleHealth).Methods("GET", "OPTIONS")
|
||||
|
||||
// Settings
|
||||
api.HandleFunc("/settings", s.handleGetSettings).Methods("GET")
|
||||
api.HandleFunc("/settings", s.handleSaveSettings).Methods("POST")
|
||||
api.HandleFunc("/settings/test/{provider}", s.handleTestKey).Methods("GET")
|
||||
api.HandleFunc("/settings", s.handleGetSettings).Methods("GET", "OPTIONS")
|
||||
api.HandleFunc("/settings", s.handleSaveSettings).Methods("POST", "OPTIONS")
|
||||
api.HandleFunc("/settings/test/{provider}", s.handleTestKey).Methods("GET", "OPTIONS")
|
||||
|
||||
// Watchlist
|
||||
api.HandleFunc("/watchlist", s.handleGetWatchlist).Methods("GET", "OPTIONS")
|
||||
api.HandleFunc("/watchlist", s.handleAddWatchlist).Methods("POST", "OPTIONS")
|
||||
api.HandleFunc("/watchlist/{ticker}", s.handleRemoveWatchlist).Methods("DELETE", "OPTIONS")
|
||||
|
||||
// News
|
||||
api.HandleFunc("/news", s.handleGetNews).Methods("GET", "OPTIONS")
|
||||
|
||||
s.router.PathPrefix("/").HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
fmt.Fprintf(w, "StockRadar API running")
|
||||
})
|
||||
}
|
||||
|
||||
func corsMiddleware(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
w.Header().Set("Access-Control-Allow-Methods", "GET, POST, DELETE, OPTIONS")
|
||||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
|
||||
if r.Method == "OPTIONS" {
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
return
|
||||
}
|
||||
next.ServeHTTP(w, r)
|
||||
})
|
||||
}
|
||||
|
||||
func (s *Server) handleHealth(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
fmt.Fprintf(w, `{"status":"ok","port":"%s"}`, s.port)
|
||||
|
||||
Reference in New Issue
Block a user