commit 287f04905896c662647f78c46b8f225673b92c18 Author: IllusionMan1212 Date: Tue Mar 16 22:04:12 2021 +0200 initial commit diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..5f23e48 --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +.env +kotapi-go \ No newline at end of file diff --git a/db/db.go b/db/db.go new file mode 100644 index 0000000..6bf7a3a --- /dev/null +++ b/db/db.go @@ -0,0 +1,56 @@ +package db + +import ( + "context" + "fmt" + "log" + "os" + + "go.mongodb.org/mongo-driver/bson" + "go.mongodb.org/mongo-driver/mongo" + "go.mongodb.org/mongo-driver/mongo/options" +) + +// package-wide variables for the collection and the context +var Kots *mongo.Collection +var Ctx = context.TODO() + +func InitializeDB() { + clientOptions := options.Client().ApplyURI(os.Getenv("DB_URI")) + client, err := mongo.Connect(Ctx, clientOptions) + if err != nil { + log.Fatal(err) + } + + err = client.Ping(Ctx, nil) + if err != nil { + log.Fatal(err) + } + fmt.Println("Connected to mongo") + + Kots = client.Database("kotapi").Collection("kots") + + // creates a unique index on the _id field, which also auto generates it + Kots.Indexes().CreateOne(Ctx, mongo.IndexModel{ + Keys: bson.D{{Key: "_id", Value: 1}}, + Options: options.Index().SetUnique(true), + }) + + // creates a unique index on the id field + Kots.Indexes().CreateOne(Ctx, mongo.IndexModel{ + Keys: bson.D{{Key: "id", Value: 1}}, + Options: options.Index().SetUnique(true), + }) + + // creates a unique index on the id field + Kots.Indexes().CreateOne(Ctx, mongo.IndexModel{ + Keys: bson.D{{Key: "url", Value: 1}}, + Options: options.Index().SetUnique(true), + }) + + // creates a unique index on the id field + Kots.Indexes().CreateOne(Ctx, mongo.IndexModel{ + Keys: bson.D{{Key: "compressed_url", Value: 1}}, + Options: options.Index().SetUnique(true), + }) +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..044922c --- /dev/null +++ b/go.mod @@ -0,0 +1,13 @@ +module illusionman1212/kotapi-go + +go 1.16 + +require ( + github.com/disintegration/imaging v1.6.2 // indirect + github.com/edwvee/exiffix v0.0.0-20190810152521-16aac9658f23 // indirect + github.com/gorilla/mux v1.8.0 // indirect + github.com/joho/godotenv v1.3.0 // indirect + github.com/rwcarlsen/goexif v0.0.0-20190401172101-9e8deecbddbd // indirect + go.mongodb.org/mongo-driver v1.5.0 // indirect + golang.org/x/image v0.0.0-20210220032944-ac19c3e999fb // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..478970c --- /dev/null +++ b/go.sum @@ -0,0 +1,126 @@ +github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= +github.com/aws/aws-sdk-go v1.34.28 h1:sscPpn/Ns3i0F4HPEWAVcwdIRaZZCuL7llJ2/60yPIk= +github.com/aws/aws-sdk-go v1.34.28/go.mod h1:H7NKnBqNVzoTJpGfLrQkkD+ytBA93eiDYi/+8rV9s48= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/disintegration/imaging v1.6.2 h1:w1LecBlG2Lnp8B3jk5zSuNqd7b4DXhcjwek1ei82L+c= +github.com/disintegration/imaging v1.6.2/go.mod h1:44/5580QXChDfwIclfc/PCwrr44amcmDAg8hxG0Ewe4= +github.com/edwvee/exiffix v0.0.0-20190810152521-16aac9658f23 h1:cHT1oZQVzPQzq3Iz3CCISUA4X94kHdoOFJ/xcbwEtqs= +github.com/edwvee/exiffix v0.0.0-20190810152521-16aac9658f23/go.mod h1:KoE3Ti1qbQXCb3s/XGj0yApHnbnNnn1bXTtB5Auq/Vc= +github.com/go-sql-driver/mysql v1.5.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg= +github.com/go-stack/stack v1.8.0 h1:5SgMzNM5HxrEjV0ww2lTmX6E2Izsfxas4+YHWRs3Lsk= +github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY= +github.com/gobuffalo/attrs v0.0.0-20190224210810-a9411de4debd/go.mod h1:4duuawTqi2wkkpB4ePgWMaai6/Kc6WEz83bhFwpHzj0= +github.com/gobuffalo/depgen v0.0.0-20190329151759-d478694a28d3/go.mod h1:3STtPUQYuzV0gBVOY3vy6CfMm/ljR4pABfrTeHNLHUY= +github.com/gobuffalo/depgen v0.1.0/go.mod h1:+ifsuy7fhi15RWncXQQKjWS9JPkdah5sZvtHc2RXGlg= +github.com/gobuffalo/envy v1.6.15/go.mod h1:n7DRkBerg/aorDM8kbduw5dN3oXGswK5liaSCx4T5NI= +github.com/gobuffalo/envy v1.7.0/go.mod h1:n7DRkBerg/aorDM8kbduw5dN3oXGswK5liaSCx4T5NI= +github.com/gobuffalo/flect v0.1.0/go.mod h1:d2ehjJqGOH/Kjqcoz+F7jHTBbmDb38yXA598Hb50EGs= +github.com/gobuffalo/flect v0.1.1/go.mod h1:8JCgGVbRjJhVgD6399mQr4fx5rRfGKVzFjbj6RE/9UI= +github.com/gobuffalo/flect v0.1.3/go.mod h1:8JCgGVbRjJhVgD6399mQr4fx5rRfGKVzFjbj6RE/9UI= +github.com/gobuffalo/genny v0.0.0-20190329151137-27723ad26ef9/go.mod h1:rWs4Z12d1Zbf19rlsn0nurr75KqhYp52EAGGxTbBhNk= +github.com/gobuffalo/genny v0.0.0-20190403191548-3ca520ef0d9e/go.mod h1:80lIj3kVJWwOrXWWMRzzdhW3DsrdjILVil/SFKBzF28= +github.com/gobuffalo/genny v0.1.0/go.mod h1:XidbUqzak3lHdS//TPu2OgiFB+51Ur5f7CSnXZ/JDvo= +github.com/gobuffalo/genny v0.1.1/go.mod h1:5TExbEyY48pfunL4QSXxlDOmdsD44RRq4mVZ0Ex28Xk= +github.com/gobuffalo/gitgen v0.0.0-20190315122116-cc086187d211/go.mod h1:vEHJk/E9DmhejeLeNt7UVvlSGv3ziL+djtTr3yyzcOw= +github.com/gobuffalo/gogen v0.0.0-20190315121717-8f38393713f5/go.mod h1:V9QVDIxsgKNZs6L2IYiGR8datgMhB577vzTDqypH360= +github.com/gobuffalo/gogen v0.1.0/go.mod h1:8NTelM5qd8RZ15VjQTFkAW6qOMx5wBbW4dSCS3BY8gg= +github.com/gobuffalo/gogen v0.1.1/go.mod h1:y8iBtmHmGc4qa3urIyo1shvOD8JftTtfcKi+71xfDNE= +github.com/gobuffalo/logger v0.0.0-20190315122211-86e12af44bc2/go.mod h1:QdxcLw541hSGtBnhUc4gaNIXRjiDppFGaDqzbrBd3v8= +github.com/gobuffalo/mapi v1.0.1/go.mod h1:4VAGh89y6rVOvm5A8fKFxYG+wIW6LO1FMTG9hnKStFc= +github.com/gobuffalo/mapi v1.0.2/go.mod h1:4VAGh89y6rVOvm5A8fKFxYG+wIW6LO1FMTG9hnKStFc= +github.com/gobuffalo/packd v0.0.0-20190315124812-a385830c7fc0/go.mod h1:M2Juc+hhDXf/PnmBANFCqx4DM3wRbgDvnVWeG2RIxq4= +github.com/gobuffalo/packd v0.1.0/go.mod h1:M2Juc+hhDXf/PnmBANFCqx4DM3wRbgDvnVWeG2RIxq4= +github.com/gobuffalo/packr/v2 v2.0.9/go.mod h1:emmyGweYTm6Kdper+iywB6YK5YzuKchGtJQZ0Odn4pQ= +github.com/gobuffalo/packr/v2 v2.2.0/go.mod h1:CaAwI0GPIAv+5wKLtv8Afwl+Cm78K/I/VCm/3ptBN+0= +github.com/gobuffalo/syncx v0.0.0-20190224160051-33c29581e754/go.mod h1:HhnNqWY95UYwwW3uSASeV7vtgYkT2t16hJgV3AEPUpw= +github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4= +github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q= +github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/gorilla/mux v1.8.0 h1:i40aqfkR1h2SlN9hojwV5ZA91wcXFOvkdNIeFDP5koI= +github.com/gorilla/mux v1.8.0/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB71So= +github.com/inconshreveable/mousetrap v1.0.0/go.mod h1:PxqpIevigyE2G7u3NXJIT2ANytuPF1OarO4DADm73n8= +github.com/jmespath/go-jmespath v0.4.0 h1:BEgLn5cpjn8UN1mAw4NjwDrS35OdebyEtFe+9YPoQUg= +github.com/jmespath/go-jmespath v0.4.0/go.mod h1:T8mJZnbsbmF+m6zOOFylbeCJqk5+pHWvzYPziyZiYoo= +github.com/jmespath/go-jmespath/internal/testify v1.5.1/go.mod h1:L3OGu8Wl2/fWfCI6z80xFu9LTZmf1ZRjMHUOPmWr69U= +github.com/joho/godotenv v1.3.0 h1:Zjp+RcGpHhGlrMbJzXTrZZPrWj+1vfm90La1wgB6Bhc= +github.com/joho/godotenv v1.3.0/go.mod h1:7hK45KPybAkOC6peb+G5yklZfMxEjkZhHbwpqxOKXbg= +github.com/karrick/godirwalk v1.8.0/go.mod h1:H5KPZjojv4lE+QYImBI8xVtrBRgYrIVsaRPx4tDPEn4= +github.com/karrick/godirwalk v1.10.3/go.mod h1:RoGL9dQei4vP9ilrpETWE8CLOZ1kiN0LhBygSwrAsHA= +github.com/klauspost/compress v1.9.5 h1:U+CaK85mrNNb4k8BNOfgJtJ/gr6kswUCFj6miSzVC6M= +github.com/klauspost/compress v1.9.5/go.mod h1:RyIbtBH6LamlWaDj8nUwkbUhJ87Yi3uG0guNDohfE1A= +github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= +github.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ= +github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/markbates/oncer v0.0.0-20181203154359-bf2de49a0be2/go.mod h1:Ld9puTsIW75CHf65OeIOkyKbteujpZVXDpWK6YGZbxE= +github.com/markbates/safe v1.0.1/go.mod h1:nAqgmRi7cY2nqMc92/bSEeQA+R4OheNU2T1kNSCBdG0= +github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc= +github.com/pelletier/go-toml v1.7.0/go.mod h1:vwGMzjaWMwyfHwgIBhI2YUM4fB6nL6lVAvS1LBMMhTE= +github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4= +github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/rogpeppe/go-internal v1.1.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4= +github.com/rogpeppe/go-internal v1.2.2/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4= +github.com/rogpeppe/go-internal v1.3.0/go.mod h1:M8bDsm7K2OlrFYOpmOWEs/qY81heoFRclV5y23lUDJ4= +github.com/rwcarlsen/goexif v0.0.0-20190401172101-9e8deecbddbd h1:CmH9+J6ZSsIjUK3dcGsnCnO41eRBOnY12zwkn5qVwgc= +github.com/rwcarlsen/goexif v0.0.0-20190401172101-9e8deecbddbd/go.mod h1:hPqNNc0+uJM6H+SuU8sEs5K5IQeKccPqeSjfgcKGgPk= +github.com/sirupsen/logrus v1.4.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo= +github.com/sirupsen/logrus v1.4.1/go.mod h1:ni0Sbl8bgC9z8RoU9G6nDWqqs/fq4eDPysMBDgk/93Q= +github.com/sirupsen/logrus v1.4.2/go.mod h1:tLMulIdttU9McNUspp0xgXVQah82FyeX6MwdIuYE2rE= +github.com/spf13/cobra v0.0.3/go.mod h1:1l0Ry5zgKvJasoi3XT1TypsSe7PqH0Sj9dhYf7v3XqQ= +github.com/spf13/pflag v1.0.3/go.mod h1:DYY7MBk1bdzusC3SYhjObp+wFpr4gzcvqqNjLnInEg4= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/tidwall/pretty v1.0.0/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk= +github.com/xdg/scram v0.0.0-20180814205039-7eeb5667e42c h1:u40Z8hqBAAQyv+vATcGgV0YCnDjqSL7/q/JyPhhJSPk= +github.com/xdg/scram v0.0.0-20180814205039-7eeb5667e42c/go.mod h1:lB8K/P019DLNhemzwFU4jHLhdvlE6uDZjXFejJXr49I= +github.com/xdg/stringprep v0.0.0-20180714160509-73f8eece6fdc h1:n+nNi93yXLkJvKwXNP9d55HC7lGK4H/SRcwB5IaUZLo= +github.com/xdg/stringprep v0.0.0-20180714160509-73f8eece6fdc/go.mod h1:Jhud4/sHMO4oL310DaZAKk9ZaJ08SJfe+sJh0HrGL1Y= +github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d h1:splanxYIlg+5LfHAM6xpdFEAYOk8iySO56hMFq6uLyA= +github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d/go.mod h1:rHwXgn7JulP+udvsHwJoVG1YGAP6VLg4y9I5dyZdqmA= +go.mongodb.org/mongo-driver v1.5.0 h1:REddm85e1Nl0JPXGGhgZkgJdG/yOe6xvpXUcYK5WLt0= +go.mongodb.org/mongo-driver v1.5.0/go.mod h1:boiGPFqyBs5R0R5qf2ErokGRekMfwn+MqKaUyHs7wy0= +golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20190422162423-af44ce270edf/go.mod h1:WFFai1msRO1wXaEeE5yQxYXgSfI8pQAWXbQop6sCtWE= +golang.org/x/crypto v0.0.0-20200302210943-78000ba7a073 h1:xMPOj6Pz6UipU1wXLkrtqpHbR0AVFnyPEQq/wRWz9lM= +golang.org/x/crypto v0.0.0-20200302210943-78000ba7a073/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/image v0.0.0-20191009234506-e7c1f5e7dbb8/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0= +golang.org/x/image v0.0.0-20210220032944-ac19c3e999fb h1:fqpd0EBDzlHRCjiphRR5Zo/RSWWQlWv34418dnEixWk= +golang.org/x/image v0.0.0-20210220032944-ac19c3e999fb/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0= +golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20200202094626-16171245cfb2/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/sync v0.0.0-20190227155943-e225da77a7e6/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190412183630-56d357773e84/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e h1:vcxGaoTs7kV8m5Np9uUNQin4BrLOthgV7252N8V+FwY= +golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20180905080454-ebe1bf3edb33/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190403152447-81d4e9dc473e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190419153524-e8e3143a4f4a/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190422165155-953cdadca894/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20190531175056-4c3a928424d2/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.3 h1:cokOdA+Jmi5PJGXLlLllQSgYigAEfHXJAERHVMaCc2k= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190329151228-23e29df326fe/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190416151739-9c9e1878f421/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190420181800-aa740d480789/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190531172133-b3315ee88b7d/go.mod h1:/rFqwRUd4F7ZHNgwSSTFct+R/Kf4OFW1sUzUTQQTgfc= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI= +gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= diff --git a/models/models.go b/models/models.go new file mode 100644 index 0000000..87494e2 --- /dev/null +++ b/models/models.go @@ -0,0 +1,10 @@ +package models + +import "go.mongodb.org/mongo-driver/bson/primitive" + +type Kot struct { + ID primitive.ObjectID `json:"_id" bson:"_id"` + MyID int32 `json:"id" bson:"id"` + Url string `json:"url" bson:"url"` + CompressedUrl string `json:"compressed_url" bson:"compressed_url"` +} diff --git a/routes/routes.go b/routes/routes.go new file mode 100644 index 0000000..e9f09b2 --- /dev/null +++ b/routes/routes.go @@ -0,0 +1,242 @@ +package routes + +import ( + "bytes" + "fmt" + "illusionman1212/kotapi-go/db" + "illusionman1212/kotapi-go/models" + "image/jpeg" + "image/png" + "math/rand" + "net/http" + "os" + "strconv" + "strings" + "time" + + "github.com/edwvee/exiffix" + "github.com/gorilla/mux" + "go.mongodb.org/mongo-driver/bson" + "go.mongodb.org/mongo-driver/bson/primitive" + "go.mongodb.org/mongo-driver/mongo" +) + +func RandomHandler(w http.ResponseWriter, req *http.Request) { + var kot models.Kot + + w.Header().Set("Content-Type", "application/json") + w.Header().Set("Access-Control-Allow-Methods", "GET") + + max, err := db.Kots.EstimatedDocumentCount(db.Ctx) + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + w.Write([]byte(`{ "error": "An error has occurred while querying the database", "status": 500, "failed": true }`)) + return + } + + // seed the default source using the current time in unix nanosecond format + rand.Seed(time.Now().UnixNano()) + // generate random number between 0 and `max` + num := rand.Intn(int(max + 1)) + + // find the random kot document and store it in the `kot` variable + err = db.Kots.FindOne(db.Ctx, bson.M{"id": num}).Decode(&kot) + if err != nil { + if err == mongo.ErrNoDocuments { + w.WriteHeader(http.StatusNotFound) + w.Write([]byte(`{ "error": "Not found","status": 404, "failed": true }`)) + } else { + w.WriteHeader(http.StatusInternalServerError) + w.Write([]byte(`{ "error": "An unknown error has occurred","status": 500, "failed": true }`)) + } + return + } + + // create a json from the kot struct variable and then write it as a response + w.WriteHeader(http.StatusOK) + w.Write([]byte(`{ + "id": "` + strconv.Itoa(int(kot.MyID)) + `", + "url": "` + kot.Url + `", + "compressed_url": "` + kot.CompressedUrl + `", + "status": 200, + "failed": false + }`)) +} + +func IdHandler(w http.ResponseWriter, req *http.Request) { + params := mux.Vars(req) + var kot models.Kot + + w.Header().Set("Content-Type", "application/json") + w.Header().Set("Access-Control-Allow-Methods", "GET") + + id, err := strconv.Atoi(params["id"]) + if err != nil { + w.WriteHeader(http.StatusBadRequest) + w.Write([]byte(`{ "error": "Invalid or incomplete request","status": 400, "failed": true }`)) + return + } + + err = db.Kots.FindOne(db.Ctx, bson.M{"id": id}).Decode(&kot) + if err != nil { + if err == mongo.ErrNoDocuments { + w.WriteHeader(http.StatusNotFound) + w.Write([]byte(`{ "error": "Not found","status": 404, "failed": true }`)) + } else { + w.WriteHeader(http.StatusInternalServerError) + w.Write([]byte(`{ "error": "An unknown error has occurred","status": 500, "failed": true }`)) + } + return + } + + w.WriteHeader(http.StatusOK) + w.Write([]byte(`{ + "id": "` + strconv.Itoa(int(kot.MyID)) + `", + "url": "` + kot.Url + `", + "compressed_url": "` + kot.CompressedUrl + `", + "status": 200, + "failed": false + }`)) +} + +func AddKotHandler(w http.ResponseWriter, req *http.Request) { + var kot models.Kot + + w.Header().Set("Content-Type", "application/json") + w.Header().Set("Access-Control-Allow-Methods", "POST") + + // 10 MB file size limit + req.ParseMultipartForm(10 << 20) + + file, handler, err := req.FormFile("image") + if err != nil { + w.WriteHeader(http.StatusBadRequest) + w.Write([]byte(`{ "error": "Invalid or incomplete request","status": 400, "failed": true }`)) + return + } + + password := req.Form.Get("password") + if password == "" { + w.WriteHeader(http.StatusBadRequest) + w.Write([]byte(`{ "error": "Invalid or incomplete request","status": 400, "failed": true }`)) + return + } + if password != os.Getenv("PASSWORD") { + w.WriteHeader(http.StatusUnauthorized) + w.Write([]byte(`{ "error": "Not authorized","status": 401, "failed": true }`)) + return + } + + filename := make([]byte, 16) + rand.Seed(time.Now().UnixNano()) + _, err = rand.Read(filename) + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + w.Write([]byte(`{ "error": "An unknown error has occurred","status": 500, "failed": true }`)) + return + } + + extensions := strings.Split(handler.Filename, ".") + extension := strings.ToLower(extensions[len(extensions)-1]) + + defer file.Close() + + data := make([]byte, handler.Size) + file.Read(data) + compressedData := data + + if extension == "png" { + enc := &png.Encoder{ + CompressionLevel: png.BestCompression, + } + buf := &bytes.Buffer{} + decodedImage, err := png.Decode(bytes.NewReader(data)) + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + w.Write([]byte(`{ "error": "An error occurred while compressing the image","status": 500, "failed": true }`)) + return + } + enc.Encode(buf, decodedImage) + compressedData = buf.Bytes() + } else if extension == "jpeg" || extension == "jpg" { + decodedImage, _, err := exiffix.Decode(bytes.NewReader(data)) + buf := &bytes.Buffer{} + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + w.Write([]byte(`{ "error": "An error occurred while compressed the image","status": 500, "failed": true }`)) + return + } + + jpeg.Encode(buf, decodedImage, &jpeg.Options{Quality: 75}) + compressedData = buf.Bytes() + } + + os.Mkdir("./kots", 0755) + os.Mkdir("./kots/compressed", 0755) + newFile, err := os.Create(fmt.Sprintf("./kots/%x.%s", filename, extension)) + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + w.Write([]byte(`{ "error": "An error occurred while writing the file","status": 500, "failed": true }`)) + return + } + + defer newFile.Close() + newFile.Write(data) + + compressedNewFile, err := os.Create(fmt.Sprintf("./kots/compressed/%x.%s", filename, extension)) + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + w.Write([]byte(`{ "error": "An error occurred while writing compressed file","status": 500, "failed": true }`)) + return + } + + defer compressedNewFile.Close() + compressedNewFile.Write(compressedData) + + count, err := db.Kots.EstimatedDocumentCount(db.Ctx) + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + w.Write([]byte(`{ "error": "An unknown error has occurred","status": 500, "failed": true }`)) + return + } + + kot.ID = primitive.NewObjectIDFromTimestamp(time.Now()) + kot.MyID = int32(count + 1) + kot.Url = fmt.Sprintf("%s/kots/%x.%s", os.Getenv("BASE_URL"), filename, extension) + kot.CompressedUrl = fmt.Sprintf("%s/kots/compressed/%x.%s", os.Getenv("BASE_URL"), filename, extension) + + _, err = db.Kots.InsertOne(db.Ctx, kot) + if err != nil { + w.WriteHeader(http.StatusInternalServerError) + w.Write([]byte(`{ "error": "An unknown error has occurred","status": 500, "failed": true }`)) + return + } + w.WriteHeader(http.StatusOK) + w.Write([]byte(`{ "message": "Successfully welcomed a new kot :)" "status": 200, "failed": false }`)) +} + +func KotsHandler(w http.ResponseWriter, req *http.Request) { + filename := mux.Vars(req) + filepath := fmt.Sprintf("./kots/%s", filename["filename"]) + _, err := os.Stat(filepath) + if err != nil { + w.Header().Set("Content-Type", "application/json") + w.WriteHeader(http.StatusNotFound) + w.Write([]byte(`{ "error": "Not found","status": 404, "failed": true }`)) + return + } + http.ServeFile(w, req, filepath) +} + +func KotsCompressedHandler(w http.ResponseWriter, req *http.Request) { + filename := mux.Vars(req) + filepath := fmt.Sprintf("./kots/compressed/%s", filename["filename"]) + _, err := os.Stat(filepath) + if err != nil { + w.Header().Set("Content-Type", "application/json") + w.WriteHeader(http.StatusNotFound) + w.Write([]byte(`{ "error": "Not found","status": 404, "failed": true }`)) + return + } + http.ServeFile(w, req, filepath) +} diff --git a/server.go b/server.go new file mode 100644 index 0000000..f5fb724 --- /dev/null +++ b/server.go @@ -0,0 +1,31 @@ +package main + +import ( + "fmt" + "illusionman1212/kotapi-go/db" + "illusionman1212/kotapi-go/routes" + "log" + "net/http" + + "github.com/gorilla/mux" + "github.com/joho/godotenv" +) + +func main() { + err := godotenv.Load() + if err != nil { + log.Fatal(err) + } + db.InitializeDB() + + router := mux.NewRouter() + router.HandleFunc("/", routes.RandomHandler).Methods("GET") + router.HandleFunc("/addkot", routes.AddKotHandler).Methods("POST") + router.HandleFunc("/{id}", routes.IdHandler).Methods("GET") + + router.HandleFunc("/kots/compressed/{filename}", routes.KotsCompressedHandler).Methods("GET") + router.HandleFunc("/kots/{filename}", routes.KotsHandler).Methods("GET") + + fmt.Println("Listening on 8080") + http.ListenAndServe(":8080", router) +}