60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"github.com/coopernurse/gorp"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
var dbmap *gorp.DbMap
|
|
var dbName = "docSet.dsidx"
|
|
|
|
// InitDb will initialize a new instance of a sqlite db for indexing
|
|
func InitDb(buildDir string) *gorp.DbMap {
|
|
dbPath := filepath.Join(buildDir, dbName)
|
|
err := os.MkdirAll(filepath.Dir(dbPath), 0755)
|
|
ExitIfError(err)
|
|
|
|
db, err := sql.Open("sqlite3", dbPath)
|
|
ExitIfError(err)
|
|
|
|
dbmap := &gorp.DbMap{Db: db, Dialect: gorp.SqliteDialect{}}
|
|
|
|
dbmap.AddTableWithName(SearchIndex{}, "searchIndex").SetKeys(true, "ID")
|
|
|
|
err = dbmap.CreateTablesIfNotExists()
|
|
ExitIfError(err)
|
|
|
|
err = dbmap.TruncateTables()
|
|
ExitIfError(err)
|
|
|
|
return dbmap
|
|
}
|
|
|
|
// SaveSearchIndex will index a particular entry into the sqlite3 database
|
|
func SaveSearchIndex(dbmap *gorp.DbMap, entry TOCEntry, entryType SupportedType, toc *AtlasTOC) {
|
|
if entry.LinkAttr.Href == "" || !entryType.IsValidType() {
|
|
return
|
|
}
|
|
|
|
relLink := entry.GetContentFilepath(toc, false)
|
|
name := entry.CleanTitle(entryType)
|
|
if entryType.ShowNamespace && len(entryHierarchy) > 0 {
|
|
// Show namespace for methods
|
|
name = entryHierarchy[len(entryHierarchy)-1] + "." + name
|
|
}
|
|
|
|
si := SearchIndex{
|
|
Name: name,
|
|
Type: entryType.TypeName,
|
|
Path: relLink,
|
|
}
|
|
|
|
err := dbmap.Insert(&si)
|
|
ExitIfError(err)
|
|
|
|
LogDebug("%s is indexed as a %s", entry.Text, entryType.TypeName)
|
|
}
|